Apollo Gateway自定义解析器：GraphQL转REST请求实现问题咨询

你的核心问题是选错了数据源基类：RemoteGraphQLDataSource是专门用于和下游GraphQL服务通信的，它默认会把请求封装成GraphQL格式转发，完全不适合转成REST请求。下面提供两种可行的实现方案：

方案一：使用官方RESTDataSource（推荐） #

Apollo提供了RESTDataSource专门用于对接REST服务，内置请求封装、缓存等能力，适配性更强。

1. 定义自定义REST数据源 #

import { RESTDataSource } from '@apollo/datasource-rest';

class CustomRestDataSource extends RESTDataSource {
  constructor(private serviceConfig: any) {
    super();
    // 设置REST服务的基础URL
    this.baseURL = serviceConfig.baseUrl;
  }

  // 对应GraphQL中的login mutation，直接封装REST请求
  async login(username: string, password: string) {
    // 发送POST请求到REST接口
    const response = await this.post('/login', {
      username,
      password
    });
    // 返回GraphQL期望的字段格式（这里假设REST返回{ token: "xxx" }）
    return response.token;
  }

  // 其他需要对接的REST接口同理，比如获取用户信息
  async getUser(userId: string) {
    const user = await this.get(`/users/${userId}`);
    return {
      id: user.id,
      name: user.fullName,
      email: user.emailAddress
    };
  }
}

2. 修改Gateway的buildService逻辑 #

确保REST服务对应的数据源使用上面的自定义类：

new ApolloGateway({
  supergraphSdl: createSuperGraphSdl(serviceSourceMap),
  buildService: ({ url, name }) => {
    const serviceConfig = backingSchemas.get(name);
    const isRestService = name.includes("rest");
    if (!serviceConfig) {
      throw new Error(`Missing config for service: ${name}`);
    }
    
    if (isRestService) {
      return new CustomRestDataSource(serviceConfig);
    } else {
      return new LocalGraphQLDataSource(serviceConfig.schema);
    }
  },
});

3. 关联Supergraph字段与数据源方法 #

需要确保你的自定义Supergraph SDL中，REST服务对应的字段 resolver 指向数据源的方法。例如，Supergraph中login mutation的定义要绑定到CustomRestDataSource.login：

type Mutation {
  login(username: String!, password: String!): String! @unsecured
}

（如果你的Supergraph生成逻辑自动关联服务字段与数据源方法，这一步可以忽略；如果没有，需要手动配置 resolver 映射）

方案二：重写fetch方法完全自定义请求流程 #

如果需要更灵活的控制（比如处理复杂的请求转换逻辑），可以重写RemoteGraphQLDataSource的fetch方法，完全接管请求的发送和响应转换。

1. 自定义数据源实现 #

class CustomRestDataSource extends RemoteGraphQLDataSource {
  constructor(private serviceConfig: any) {
    super({ schema: serviceConfig.schema });
    this.baseURL = serviceConfig.baseUrl;
  }

  async fetch(request: RequestInit, options: any) {
    // 解析原始GraphQL请求的内容
    const graphqlReq = JSON.parse(request.body as string);
    const { operationName, variables } = graphqlReq;

    // 根据操作名处理不同的REST请求
    switch (operationName) {
      case 'login': {
        const { username, password } = variables;
        // 构造并发送REST POST请求
        const restResponse = await fetch(`${this.baseURL}/login`, {
          method: 'POST',
          headers: {
            'Content-Type': 'application/json',
            // 从context获取授权信息（如果需要）
            'Authorization': `Bearer ${options.context.authToken}`
          },
          body: JSON.stringify({ username, password })
        });

        const restData = await restResponse.json();
        // 将REST响应转换为GraphQL规范格式
        return new Response(JSON.stringify({
          data: { login: restData.token }
        }), {
          status: restResponse.status,
          headers: restResponse.headers
        });
      }
      // 其他操作可以在这里扩展
      default:
        throw new Error(`Unsupported operation: ${operationName}`);
    }
  }
}

2. 替换Gateway中的数据源 #

同方案一，在buildService中为REST服务返回这个自定义数据源即可。

关键注意事项 #

• 上下文传递：如果需要在REST请求中携带授权信息、用户ID等，可以通过options.context获取网关的上下文数据，添加到REST请求的headers或参数中。
• 响应格式转换：必须确保返回的响应符合GraphQL规范（包含data或errors字段），否则网关无法正确解析并返回给客户端。
• Supergraph兼容性：你的自定义Supergraph需要正确定义REST服务对应的类型和字段，确保网关能将请求路由到对应的数据源。

内容的提问来源于stack exchange，提问作者Mr.P