如何在NestJS GraphQL网关中注入Service构建上下文？

完全可以用NestJS服务替代普通函数的方式来构建GraphQL网关的上下文，还能充分利用NestJS的依赖注入能力实现复杂业务逻辑。具体实现步骤如下：

1. 创建带依赖注入的上下文服务 #

定义一个@Injectable()装饰的上下文服务，把原getContext的逻辑迁移到服务方法中，同时可以注入其他NestJS服务：

import { Injectable } from '@nestjs/common';
import { Request } from 'express';
// 示例：注入鉴权服务
import { AuthService } from './auth/auth.service';

@Injectable()
export class ContextService {
  constructor(private readonly authService: AuthService) {}

  async createContext({ req }: { req: Request }) {
    // 示例：解析请求头token并验证用户
    const authToken = req.headers.authorization?.split(' ')[1];
    const user = authToken ? await this.authService.validateToken(authToken) : null;
    
    return { user };
  }
}

2. 改用forRootAsync配置GraphQL网关 #

原同步的forRoot无法访问NestJS依赖容器，需要换成forRootAsync，通过注入上下文服务来构建网关上下文：

import { Module } from '@nestjs/common';
import { GraphQLModule } from '@nestjs/graphql';
import { ApolloGatewayDriver, ApolloGatewayDriverConfig } from '@nestjs/apollo';
import { IntrospectAndCompose } from '@apollo/gateway';
import { ContextService } from './context.service';
import { AuthService } from './auth/auth.service';

const API_URL_IAM = 'http://your-iam-subgraph-url';

@Module({
  imports: [
    // ...其他模块
    GraphQLModule.forRootAsync<ApolloGatewayDriverConfig>({
      driver: ApolloGatewayDriver,
      // 注入上下文服务
      inject: [ContextService],
      useFactory: (contextService: ContextService) => ({
        server: {
          // 调用服务的createContext方法生成上下文
          context: ({ req }) => contextService.createContext({ req }),
        },
        gateway: {
          buildService: ({ name, url }) => {
            return new RemoteGraphQLDataSource({
              url,
              willSendRequest({ request, context }: any) {
                // 示例：将上下文用户ID传递给子图
                if (context.user) {
                  request.http.headers.set('x-user-id', context.user.id);
                }
              },
            });
          },
          supergraphSdl: new IntrospectAndCompose({
            subgraphs: [
              { name: 'iam', url: API_URL_IAM },
            ],
          }),
        },
      }),
    }),
  ],
  // 注册上下文服务及依赖服务
  providers: [ContextService, AuthService],
})
export class AppModule {}

关键说明 #

• forRootAsync的作用：它允许在GraphQL模块配置阶段访问NestJS依赖注入容器，从而获取到ContextService实例。
• 避免this指向问题：用箭头函数包裹服务方法调用（如({ req }) => contextService.createContext({ req })），无需额外绑定this，代码更简洁。
• 复用NestJS生态：上下文服务可以注入任何NestJS服务，比如数据库服务、缓存服务等，完全复用框架的依赖注入能力。

内容的提问来源于stack exchange，提问作者MD. Jahidul Islam