typescript - 将 pg-promise 任务与 graphql 和 dataloader 一起使用
问题描述
首先要说的是我已经看到了这个问题在 http 请求的各个部分共享一个 pg-promise 任务 并阅读了答案。我想重新审视这个问题,增加一些复杂性。
我正在apollo-server-express
使用 postgres 数据库作为后端和pg-promise
数据库接口库的 typescript graphql 服务器。我有使用数据加载器的查询解析器,使用此处描述的模式:https ://github.com/graphql/dataloader#creating-a-new-dataloader-per-request 。
所有数据库访问都通过推荐的单个实例进行pg-promise
,例如https://github.com/vitaly-t/pg-promise-demo/blob/master/TypeScript/db/index.ts。
与最初的问题一样,我正在尝试找到一种创建任务的好方法,该任务可以合并到数据加载器中(或在调用数据加载器时作为参数传递),以便查询解析器在单个数据库连接中运行。我不知道将调用哪些解析器或调用它们的顺序,所以我不能说某个解析器应该创建任务。
我一直在尝试寻找一种方法来使用 graphqlContext
来共享任务,因为我目前正在使用Context
来共享 dataloaders 对象,但正如所指出的,任务中的所有调用都发生在回调函数中,因此通过Context
is共享任务出去。以下是如何设置上下文的示例:
/*
createLoaders() returns a new object of DataLoaders e.g.
{
users: new DataLoader(ids => getUsers(ids)),
tags: new DataLoader(ids => getTags(ids))
}
*/
import { createLoaders } from './dataloaders'
const server = new ApolloServer({
schema,
context: (): Promise<Context> => {
// create a connection here to inject into createLoaders?
const loaders = createLoaders()
return { loaders }
}
})
以及在上下文中创建数据库连接的 Apollo 文档中的示例:https ://www.apollographql.com/docs/apollo-server/data/resolvers/#the-context-argument (请注意,它们没有描述在任何地方关闭连接)。
context: async () => ({
db: await client.connect(),
})
// Resolver
(parent, args, context, info) => {
return context.db.query('SELECT * FROM table_name');
}
如果不使用该方法,这可能是不可能的connect
,但是库的作者不鼓励将其用于这种类型的操作。并且出现了同样的问题,即在将结果发送回客户端之前最终解决查询时需要将连接返回到池中。
任何帮助表示赞赏!(感谢@vitaly-t 的出色工作pg-promise
)
编辑 1 @Daniel-Rearden 您的建议和链接的要点最终起作用了。如果总是打开任务成为问题,我可能会检查来自插件内的传入请求。标记你的答案正确!
解决方案
您在这里使用 DataLoader 的事实在很大程度上是无关紧要的。就像您已经建议的那样,您只需要将用于调用该query
方法的任何实例传递给createLoaders
. 如果您没有使用 DataLoader,您仍然会遇到相同的问题,即我们如何创建任务并使其可供所有解析器使用。
据我所知,使用 Apollo Server 做到这一点的唯一方法是创建一个自定义插件。你可以查看这个要点,它展示了除了交易之外的相同原理。实际上,您会执行以下操作:
import { ApolloServerPlugin, GraphQLRequestContext } from 'apollo-server-plugin-base'
export const createTaskPlugin = (db: Database): ApolloServerPlugin => ({
requestDidStart(requestContext: GraphQLRequestContext) {
let taskResult: Promise<{ ok?: boolean, error?: any }> | void = undefined
return {
executionDidStart() {
let ok: (value?: unknown) => void
let fail: (reason?: any) => void
const didFinish = new Promise((resolve, reject) => {
ok = resolve
fail = reject
})
const task = new Promise((resolve) =>
taskResult = db.task(t => {
resolve(t)
return didFinish
}).then(() => ({ ok: true }), error => ({ error }))
)
requestContext.context.runWithTask = async (cb) => cb(await task)
return (err) => {
if (err) fail(err)
ok()
}
},
willSendResponse() {
if (!taskResult) return
return taskResult
.then(({ error }) => {
if (error) {
throw error
}
})
}
}
}
})
安装您的插件,如文档中所示:
const server = new ApolloServer({
typeDefs,
resolvers,
plugins: [
createTaskPlugin(db),
],
})
然后在您的解析器中,您将执行以下操作:
resolve (parent, args, context) {
return runWithTask(t => t.query('SELECT * FROM table_name'))
}
推荐阅读
- sql - 如何查询表中的传递匹配?
- c# - 如何在异步数据库连接期间避免“访问冲突”?
- linux - 从数据文件的每一行中提取 2 个字段并插入到外部程序中
- typescript - 将自引用作为抽象基类静态函数中的参数传递给子类
- swift - 为什么 Xcode 会为选择器添加自动完成方法?
- java - 来自 nginx 的请求失败,状态码为 504
- tensorflow - 我得到“启用急切执行时,Tensor.op 毫无意义。” 在我的简单编码器模型中。(TF 2.0)
- android - 无法在 Android Studio 中解析符号“URL”
- javascript - 如何使 ag-grid 列与列中最大的文本一样宽
- database-administration - Greenplum 数据库中的时间点恢复