Создание защищенных подписок с помощью makeRemoteExecutableSchema

У меня есть одно рабочее решение: идея состоит в том, чтобы не создавать один экземпляр SubscriptionClient для всего приложения. Вместо этого я создаю клиентов для каждого подключения к прокси-серверу:

server.start({
    port: 4000,
    subscriptions: {
      onConnect: (connectionParams, websocket, context) => {
        return {
          subscriptionClients: {
            messageService: new SubscriptionClient(process.env.MESSAGE_SERVICE_SUBSCRIPTION_URL, {
              connectionParams,
              reconnect: true,
            }, ws)
          }
        };
      },
      onDisconnect: async (websocket, context) => {
        const params = await context.initPromise;
        const { subscriptionClients } = params;
        for (const key in subscriptionClients) {
          subscriptionClients[key].close();
        }
      }
    }
  }, (options) => console.info('Server is running on http://localhost:4000'))

если бы у вас было больше удаленных схем, вы бы просто создали больше экземпляров SubscriptionClient в карте subscriptionClients.

Чтобы использовать этих клиентов в удаленной схеме, вам нужно сделать две вещи:

1. выставить их в контексте:

   const server = new GraphQLServer({
     schema,
     context: ({ connection }) => {
       if (connection && connection.context) {
         return connection.context;
       }
     }
   });
   

2. использовать реализацию настраиваемой ссылки вместо WsLink

   (operation, forward) => {
       const context = operation.getContext();
       const { graphqlContext: { subscriptionClients } } = context;
       return subscriptionClients && subscriptionClients[clientName] && subscriptionClients[clientName].request(operation);
   };
   

Таким образом, все параметры подключения будут переданы на удаленный сервер.

Полный пример можно найти здесь: https://gist.github.com/josephktcheung/cd1b65b321736a520ae9d822ae5a951b

Заявление об ограничении ответственности:

Код не мой, поскольку @josephktcheung опередил меня, предоставив пример. Я просто немного помог с этим. Вот исходное обсуждение: https://github.com/apollographql/graphql-tools/issues/864

Это рабочий пример удаленной схемы с подпиской через webscoket и запросом и изменением через http. Его можно защитить с помощью настраиваемых заголовков (параметров) и показать в этом примере.

Поток

Запрос клиента -> context создается путем чтения req или connection (jwt декодируется и создает пользовательский объект в контексте)
-> удаленная схема выполняется -> link называется -> link разбивается по операции (wsLink для подписки, httpLink для запросов и мутаций) -> wsLink или httpLink доступ к context, созданному выше (= graphqlContext) -> wsLink или httpLink используют context для создания заголовков (заголовок авторизации с подписанным jwt в этом примере) для удаленной схемы. -> «подписка» или «запрос или изменение» пересылаются на удаленный сервер.

Примечание

1. В настоящее время ContextLink не влияет на WebsocketLink. Итак, вместо concat мы должны создать необработанный ApolloLink.
2. При создании контекста проверяйте connection, а не только req. Первый будет доступен, если запрос является веб-сокетом и содержит метаинформацию, отправляемую пользователем, например токен аутентификации.
3. HttpLink ожидает глобальную выборку со стандартной спецификацией. Таким образом, не используйте node-fetch, спецификация которого несовместима (особенно с машинописным текстом). Вместо этого используйте cross-fetch.

const wsLink = new ApolloLink(operation => {
    // This is your context!
    const context = operation.getContext().graphqlContext

    // Create a new websocket link per request
    return new WebSocketLink({
      uri: "<YOUR_URI>",
      options: {
        reconnect: true,
        connectionParams: { // give custom params to your websocket backend (e.g. to handle auth) 
          headers: {
            authorization: jwt.sign(context.user, process.env.SUPER_SECRET),
            foo: 'bar'
          }
        },
      },
      webSocketImpl: ws,
    }).request(operation)
    // Instead of using `forward()` of Apollo link, we directly use websocketLink's request method
  })

const httpLink = setContext((_graphqlRequest, { graphqlContext }) => {
  return {
    headers: {
      authorization: jwt.sign(graphqlContext.user, process.env.SUPER_SECRET),
    },
  }
}).concat(new HttpLink({
  uri,
  fetch,
}))

const link = split(
  operation => {
    const definition = getMainDefinition(operation.query)
    return (
      definition.kind === 'OperationDefinition' &&
      definition.operation === 'subscription'
    )
  },
  wsLink, // <-- Executed if above function returns true
  httpLink, // <-- Executed if above function returns false
)

const schema = await introspectSchema(link)

const executableSchema = makeRemoteExecutableSchema({
    schema,
    link,
  })

const server = new ApolloServer({
  schema: mergeSchemas([ executableSchema, /* ...anotherschemas */]),
  context: ({ req, connection }) => {
    let authorization;
    if (req) { // when query or mutation is requested by http
      authorization = req.headers.authorization
    } else if (connection) { // when subscription is requested by websocket
      authorization = connection.context.authorization
    }
    const token = authorization.replace('Bearer ', '')
    return {
      user: getUserFromToken(token),
    }
  },
})