在GraphQL Apollo Server中调用REST API时如何正确添加自定义请求头？

你的代码核心问题大概率是数据源未正确注册到Apollo Server，或者异常处理掩盖了真实错误，以下是具体修复步骤：

1. 确保数据源正确注册 #

Apollo Server不会自动初始化你的CountriesAPI，必须在Server配置中通过dataSources函数注册：

const { ApolloServer } = require('@apollo/server');
const { startStandaloneServer } = require('@apollo/server/standalone');

// 假设你有对应的类型定义和解析器
const typeDefs = `#graphql
  type Country {
    # 你的类型字段定义
  }
  type Query {
    getAllCountries: [Country]
  }
`;

const resolvers = {
  Query: {
    getAllCountries: async (_, __, { dataSources }) => {
      // 从上下文的dataSources中调用API方法
      return dataSources.countriesAPI.getAllCountries();
    },
  },
};

const server = new ApolloServer({
  typeDefs,
  resolvers,
  // 关键：注册数据源实例
  dataSources: () => ({
    countriesAPI: new CountriesAPI(),
  }),
});

// 启动Apollo服务
startStandaloneServer(server, { listen: { port: 4000 } });

2. 修正请求头设置与异常处理 #

• 简化请求头设置语法，同时不要掩盖请求错误：

class CountriesAPI extends RESTDataSource {
  constructor() {
    super();
    this.baseURL = "http://localhost:9321";
  }

  willSendRequest(request) {
    // 直接设置请求头，避免不必要的换行
    request.headers.set("name", "Tat");
    // 打印请求信息确认头是否被正确添加
    console.log('待发送请求:', request);
  }

  async getAllCountries() {
    try {
      return this.get("/getData");
    } catch (error) {
      console.error('REST API请求失败:', error);
      // 重新抛出错误，让Apollo Server返回给客户端
      throw new Error(`获取国家数据失败：${error.message}`);
    }
  }
}

3. 额外检查项 #

• 确认http://localhost:9321服务正常运行，/getData接口可直接访问
• 检查后端是否允许name这个自定义请求头（部分后端会过滤非标准HTTP头）
• 确认@apollo/datasource-rest版本在v3及以上（旧版本API逻辑可能不同）

内容的提问来源于stack exchange，提问作者Tat