在Apollo Server中,可以使用dataSources和context选项来与数据库一起使用。dataSources是一个用于管理数据源的集合,而context是一个用于在请求之间共享数据的对象。
下面是一个示例,演示了如何使用dataSources和context选项与数据库一起使用:
npm install apollo-server apollo-datasource-mongodb mongodb
const { DataSource } = require('apollo-datasource');
class MongoDBDataSource extends DataSource {
constructor(collection) {
super();
this.collection = collection;
}
initialize(config) {
this.context = config.context;
}
async getUsers() {
const users = await this.collection.find({}).toArray();
return users;
}
async getUserById(id) {
const user = await this.collection.findOne({ _id: id });
return user;
}
async addUser(user) {
const result = await this.collection.insertOne(user);
return result.ops[0];
}
}
module.exports = MongoDBDataSource;
const { ApolloServer, gql } = require('apollo-server');
const { MongoClient } = require('mongodb');
const MongoDBDataSource = require('./MongoDBDataSource');
const typeDefs = gql`
type User {
id: ID!
name: String!
email: String!
}
type Query {
users: [User]!
user(id: ID!): User
}
type Mutation {
addUser(name: String!, email: String!): User
}
`;
const resolvers = {
Query: {
users: (_, __, { dataSources }) => dataSources.mongoDB.getUsers(),
user: (_, { id }, { dataSources }) => dataSources.mongoDB.getUserById(id),
},
Mutation: {
addUser: (_, { name, email }, { dataSources }) => {
const user = { name, email };
return dataSources.mongoDB.addUser(user);
},
},
};
async function startServer() {
const client = await MongoClient.connect('');
const db = client.db('');
const collection = db.collection('');
const server = new ApolloServer({
typeDefs,
resolvers,
dataSources: () => ({
mongoDB: new MongoDBDataSource(collection),
}),
context: ({ req }) => ({
// 在context中可以添加其他自定义数据
token: req.headers.authorization,
}),
});
server.listen().then(({ url }) => {
console.log(`Server ready at ${url}`);
});
}
startServer();
在上面的示例中,MongoDBDataSource类扩展了Apollo Server的DataSource类,并实现了用于获取用户、根据ID获取用户和添加用户的方法。在initialize方法中,我们可以访问context对象,以便在数据源中共享任何自定义数据。
在Apollo Server配置中,我们将数据源传递给dataSources选项,并通过context选项将自定义数据添加到context对象中。
请注意,在MongoDB连接字符串、数据库名称和集合名称之间填写适当的值。
这是一个简单的示例,演示了如何使用dataSources和context选项与数据库一起使用。根据你的具体需求,可以根据这个示例进行修改和扩展。