如何高效实现MongoDB嵌套GraphQL查询的Resolver？

问题背景 #

我正在编写一个GraphQL Resolver，用于获取客户名下的账户。parent.accounts是一个account_id数组，当前代码能正常返回预期结果，但性能很差——全量请求时，Resolver会针对每个客户的account_id数组单独发起数据库查询，查询次数随客户数量增加而暴涨，请求速度越来越慢。

当前实现代码：

import Account from "../models/accounts.js";
import Customer from "../models/customers.js";
import { AccountInterface } from "../types/accounts.js";
import { CustomerInterface } from "../types/customers.js";

const resolvers = {
    Query: {
        // 其他查询逻辑
        customers: async () => await Customer.find(),
    },
    Customer: {
        accounts: async (parent: CustomerInterface) => {
            const response = await Account.find()
                .where("account_id")
                .in(parent.accounts);
            return response;
        },
    },
};
export default resolvers;

优化方案：用DataLoader实现批量查询 #

你的问题本质是GraphQL典型的N+1查询问题——先查询所有客户，再为每个客户单独发起账户查询，导致查询次数随客户数量线性增长。解决核心是通过DataLoader批量收集所有需要查询的账户ID，一次性从数据库获取数据后再分配给对应的客户。

步骤1：安装DataLoader #

npm install dataloader

步骤2：创建账户批量加载器 #

单独创建一个加载器文件（比如loaders/accountLoader.js），实现批量查询逻辑：

import DataLoader from 'dataloader';
import Account from "../models/accounts.js";

// 批量加载函数：接收一组account_id，返回对应账户数组
const accountLoader = new DataLoader(async (accountIds) => {
    // 一次性查询所有匹配的账户
    const accounts = await Account.find({ account_id: { $in: accountIds } });
    
    // 将账户按ID分组，方便后续快速匹配
    const accountMap = new Map();
    accounts.forEach(account => {
        accountMap.set(account.account_id, account);
    });
    
    // 保持返回数组顺序与传入的accountIds一致，保证数据对应正确
    return accountIds.map(id => accountMap.get(id));
});

export default accountLoader;

步骤3：修改Resolver使用DataLoader #

在Resolver中导入并使用刚才创建的加载器：

import Account from "../models/accounts.js";
import Customer from "../models/customers.js";
import { AccountInterface } from "../types/accounts.js";
import { CustomerInterface } from "../types/customers.js";
import accountLoader from './loaders/accountLoader.js';

const resolvers = {
    Query: {
        customers: async () => await Customer.find(),
    },
    Customer: {
        accounts: async (parent: CustomerInterface) => {
            // 通过loader批量加载账户，自动合并同一请求内的所有查询
            return Promise.all(parent.accounts.map(id => accountLoader.load(id)));
        },
    },
};
export default resolvers;

优化效果说明 #

• 数据库查询次数从N次（N为客户数量）直接减少到1次，性能大幅提升
• DataLoader会自动去重同一请求内的重复ID，避免无效查询
• 严格保证返回结果的顺序与请求ID的顺序一致，完全符合GraphQL的数据返回要求

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