Apollo服务器：如何从REST API数据源访问Dataloader中解析器之外的“上下文”

如何解决Apollo服务器：如何从REST API数据源访问Dataloader中解析器之外的“上下文”

希望有人可以帮助我解决这个小问题，我现在无法解决。

问题陈述：

为了进行身份验证，我想在DataLoader中访问“上下文”。此DataLoader是在单独的路径/loaders中定义的。在我的resolvers.js文件中，我可以使用dataSources.userAPI.getAllUsers()很好地访问上下文。 但是如何在服务器端应用程序中的其他任何地方访问它，例如f.e.在我的/loaders文件夹中？ 我只是无法了解如何访问上下文对象，然后将令牌传递给DataLoader，然后从我的API加载数据，然后将这些数据传递给我的resolvers.js文件。 非常感谢您的帮助，我不知道该如何解决这个简单的问题..谢谢！

以下是代码：

index.js

const express = require('express');
const connectDB = require('./config/db');
const path = require('path');
var app = express();
const cors = require('cors')
const axios = require('axios')

// apollo graphql
const { ApolloServer } = require('apollo-server-express');
const DataLoader = require('dataloader')
const { userDataLoader } = require('./loaders/index')

// Connect Database
connectDB();

// gql import
const typeDefs = require('./schema');
const resolvers = require('./resolvers')

// apis
const UserAPI = require('./datasources/user')


// datasources
const dataSources = () => ({
    userAPI: new UserAPI(),});

// context
const context = ({ req,res }) => ({

    token: req.headers.authorization || null,loaders: {
        userLoader: userDataLoader,},res
})


// init server
const server = new ApolloServer({
    typeDefs,resolvers,dataSources,context
});

// middleware
app.use(express.json());


// cors
var corsOptions = {
    credentials: true
}
app.use(cors(corsOptions))


// serve middleware
server.applyMiddleware({
    app
});


// run server
app.listen({ port: 4000 },() =>
    console.log(`Server ready at http://localhost:4000${server.graphqlPath}`)
);

module.exports = {
    dataSources,context,typeDefs,loaders,ApolloServer,UserAPI,server,};

loaders / index.js

   const userDataLoader = require('./user')

module.exports = {
    userDataLoader
}

loaders / user.js

const UserAPI = require('../datasources/users')
// init loader
const userDataLoader = new DataLoader(keys => batchUser(keys))

// batch
const batchUsers = async (keys) => {

   // this part is not working!
   // How to access the UserAPI methods in my DataLoader?
   // Or lets say: How to access context from here,// so I can add auth for the server I am requesting data from?

    const userAPI = new UserAPI()
    const users = userAPI.getAllUsers()
        .then(res => {
            return res.data
        })


    return keys.map(userId => users.find(user=> user._id === userId))
}

module.exports = userDataLoader

resolvers.js

// here is just my api call to get the data from my
// dataloader with userLoader.load() and this works perfectly
// if I just make API calls with axios in my loaders/user
// here just a little snippet from the resolver file

....
users: async (parent,args,{ loaders }) => {
            const { userLoader } = loaders
            if (!parent.users) {
                return null;
            }
            return await userLoader.load(parent.user)
        },....

数据源/user.js

const { RESTDataSource } = require('apollo-datasource-rest');

class UserAPI extends RESTDataSource {
    constructor() {
        super()
        this.baseURL = 'http://mybaseurl.com/api'
    }


    willSendRequest(request) {
        request.headers.set('Authorization',this.context.token
        );
    }

    async getUserById(id) {
        return this.get(`/users/${id}`)
    }

    async getAllUsers() {
        const data = await this.get('/users');
        return data;
    }
}

module.exports = UserAPI;

解决方法

我建议创建一个函数，该函数创建数据加载器并在闭包内部提供所需的状态，而完全不使用数据源：

module.exports.createDataloaders = function createDataLoaders(options) {
  const batchUsers = ids => {
    const users = await fetch('/users/',{ headers: { Authorization: options.auth } });
    // ...
  }

  return {
    userLoader: new Dataloader(batchUsers);
  };
}

// now in index.js
// context
const context = ({ req,res }) => ({
    token: req.headers.authorization || null,loaders: createDataloaders({ auth: req.headers.authorization || null }),res
})


// init server
const server = new ApolloServer({
    typeDefs,resolvers,context
});

或考虑将您的图层翻转：

const { RESTDataSource } = require('apollo-datasource-rest');

class UserAPI extends RESTDataSource {
    constructor() {
        super()
        this.baseURL = 'http://mybaseurl.com/api'
        this.dataloader = new Dataloader(ids => {
          // use this.get here
        });
    }


    willSendRequest(request) {
        request.headers.set('Authorization',this.context.token
        );
    }

    async getUserById(id) {
        return this.dataloader.load(id);
    }

    async getAllUsers() {
        const data = await this.get('/users');
        return data;
    }
}

module.exports = UserAPI;

但是，数据源并未设计为与Dataloader配合使用，如Apollo Docs本身的here中所述。因此，如果您想保留源代码，可以一起摆脱装载程序。

基本上，他们所说的是，包装其余API的GraphQL API从（全局）请求缓存中受益的比从预请求数据加载器中获得的更多。跨请求缓存可能会导致授权问题，因此请小心。