如何在 Apollo 中关闭 GraphQL 订阅的套接字连接

Question

我的 Apollo 服务器上有 GraphQL 订阅，我想在用户注销后关闭。最初的问题是我们应该在客户端还是在后端关闭这个（套接字）连接。

在前端，我将 Angular 与 Apollo 客户端一起使用，并通过从 apollo-angular 扩展 Subscription 类来处理 GraphQL 订阅。我可以使用典型的takeUntil rxjs 实现来关闭订阅频道：

this.userSubscription
  .subscribe()
  .pipe(takeUntil(this.subscriptionDestroyed$))
  .subscribe(
    ({ data }) => {
      // logic goes here
    },
    (error) => {
      // error handling
   }
);
  

但是，这不会关闭服务器上的 websocket，如果我是对的，这将导致订阅内存泄漏。

Apollo 服务器（和 express）的订阅设置方式如下：

const server = new ApolloServer({
  typeDefs,
  resolvers,
  subscriptions: {
    onConnect: (connectionParams, webSocket, context) => {
      console.log('on connect');
      const payload = getAuthPayload(connectionParams.accessToken);
      if (payload instanceof Error) {
        webSocket.close();
      }
      return { user: payload };
    },
    onDisconnect: (webSocket, context) => {
      console.log('on Disconnect');
    }
  },
  context: ({ req, res, connection }) => {
    if (connection) {
      // set up context for subscriptions...
    } else {
      // set up context for Queries, Mutations...
    }

当客户端注册新的 GraphQL 订阅时，我总是会在服务器日志上看到 console.log('on connect');，但除非我关闭前端应用程序，否则我永远不会看到 console.log('on Disconnect');。

我还没有看到任何关于如何关闭 websocket 以使用 Apollo 订阅的示例。我主要是想完成一个 Logout 的实现。

我在这里遗漏了什么吗？提前致谢！

Answer 1

我在 Apollo Client GitHub 网站上的相关讨论中发布了 my solution。