如何在非ID字段上使用GraphQL Dataloader？Node.js(TypeScript)场景

Dataloader并非只能依赖单个字段，完全可以通过复合键实现多条件（比如你的user_id+follow_user_id）的批量加载，核心思路是把多个查询条件组合成唯一缓存键，再在批量处理函数中统一解析并查询。

实现步骤 #

1. 创建基于复合键的Dataloader

将两个ID用分隔符拼接成唯一字符串（或用JSON序列化对象作为键），作为Dataloader的缓存标识：

import DataLoader from 'dataloader';
import { Followers } from './your-models-path'; // 替换为你的Follower模型路径

// 定义复合键类型
type FollowCompositeKey = `${string}|${string}`;

// 创建Dataloader实例
export const createFollowDataloader = () => {
  return new DataLoader<FollowCompositeKey, typeof Followers | null>(async (keys) => {
    // 解析所有复合键，提取对应的user_id和follow_user_id
    const queryConditions = keys.map(key => {
      const [userId, followUserId] = key.split('|');
      return { user_id: userId, follow_user_id: followUserId };
    });

    // 批量查询所有匹配的Follower记录
    const followers = await Followers.find({
      $or: queryConditions
    });

    // 将查询结果映射回对应的复合键
    return keys.map(key => {
      const [userId, followUserId] = key.split('|');
      return followers.find(f => 
        f.user_id.toString() === userId && f.follow_user_id.toString() === followUserId
      ) || null;
    });
  });
};

2. 将Dataloader注入GraphQL上下文

在上下文初始化逻辑中添加这个Dataloader：

// AppContext类型定义
export type AppContext = {
  user: { _id: string };
  followDataloader: ReturnType<typeof createFollowDataloader>;
};

// 创建上下文的函数
export const createContext = ({ req }): AppContext => {
  return {
    user: req.user, // 假设已通过认证获取当前用户
    followDataloader: createFollowDataloader(),
  };
};

3. 修改字段解析器使用Dataloader

替换原有的直接查询逻辑，改用Dataloader加载数据：

@FieldResolver(returns => Follower, { nullable: true })
async follows_me(
    @Root() user: DocumentType<User>,
    @Ctx() context: AppContext
) {
    // 生成复合键
    const compositeKey = `${user._id}|${context.user._id}`;
    // 调用Dataloader获取数据
    return context.followDataloader.load(compositeKey);
}

关键说明 #

• 复合键兼容性：如果ID可能包含|这类分隔符，建议用JSON.stringify({ userId: user._id, followUserId: context.user._id })作为键，避免冲突。
• 性能优化：$or批量查询能一次性获取所有匹配记录，相比多次findOne大幅减少数据库请求次数，尤其在批量查询多个User的follows_me字段时效果明显。
• 自动缓存：Dataloader会自动缓存相同复合键的请求，重复请求同一user_id+follow_user_id组合时直接返回缓存结果，无需重复查询。

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