在GraphQL中实现实时数据更新之Pub/Sub

原创

種法龍

发布于 2023-12-31 14:40:43

1770

发布于 2023-12-31 14:40:43

文章被收录于专栏：未知未知

Pub/Sub 模式是一种发布-订阅模式，其中一个组件（发布者）发布消息，而其他组件（订阅者）监听并接收这些消息。在 GraphQL 中，可以使用 Pub/Sub 模式来实现实时数据更新，使服务器能够向客户端推送数据变更。

在下面的示例中，将使用 Redis 作为 Pub/Sub 的中间件。请确保你已经安装了 graphql-yoga（一个用于构建 GraphQL 服务器的库）和 redis（用于创建 Redis 客户端的库）。可以使用以下命令安装这两个库：

npm install graphql-yoga redis

然后，可以使用以下代码实现 GraphQL 服务器，使用 Redis Pub/Sub 模式实现实时数据更新：

const { GraphQLServer, PubSub } = require('graphql-yoga');
const Redis = require('ioredis');

// 创建 Redis 客户端
const redis = new Redis();

// 创建 Pub/Sub 实例
const pubsub = new PubSub();

// 数据库模拟
const db = {
  messages: [],
};

// GraphQL 类型定义
const typeDefs = `
  type Message {
    id: ID!
    content: String!
  }

  type Query {
    messages: [Message!]!
  }

  type Mutation {
    postMessage(content: String!): ID!
  }

  type Subscription {
    messageAdded: Message!
  }
`;

// GraphQL 解析器
const resolvers = {
  Query: {
    messages: () => db.messages,
  },
  Mutation: {
    postMessage: (parent, { content }) => {
      const id = String(db.messages.length + 1);
      const message = { id, content };
      db.messages.push(message);

      // 发布消息到 Redis
      pubsub.publish('messageAdded', { messageAdded: message });

      return id;
    },
  },
  Subscription: {
    messageAdded: {
      subscribe: () => pubsub.asyncIterator('messageAdded'),
    },
  },
};

// 创建 GraphQL 服务器
const server = new GraphQLServer({ typeDefs, resolvers });

// 启动服务器
server.start({ port: 4000 }, () => {
  console.log('Server is running on http://localhost:4000');
});

// Redis 订阅消息
redis.subscribe('messageAdded', (err, count) => {
  console.log(`Subscribed to ${count} channel(s)`);
});

// 处理接收到的消息
redis.on('message', (channel, message) => {
  console.log(`Received message on channel ${channel}: ${message}`);
  const parsedMessage = JSON.parse(message);
  pubsub.publish(channel, parsedMessage);
});

// 处理错误
redis.on('error', (err) => {
  console.error(`Redis error: ${err}`);
});

上述代码创建了一个简单的 GraphQL 服务器，其中包含查询（Query）、变更（Mutation）和订阅（Subscription）。当使用 postMessage 变更时，服务器会发布消息到 Redis 的 messageAdded 频道，而订阅者将通过订阅 messageAdded 频道来获取实时更新。

请注意，这只是一个简单的示例，实际项目中可能需要处理更复杂的逻辑和错误情况。确保已经按照项目需求进行了适当的配置和错误处理。

原创声明：本文系作者授权腾讯云开发者社区发表，未经许可，不得转载。

如有侵权，请联系 cloudcommunity@tencent.com 删除。

publish-subscribe