如何向Hot Chocolate解析器传递ID列表以减少数据库查询次数

核心思路 #

Hot Chocolate提供的DataLoader机制专门用于解决GraphQL中的N+1查询问题，它会自动收集同一批次里所有需要加载的用户ID，一次性调用数据库查询，再将结果分配给对应的实体，从根源减少数据库请求次数。

实现步骤 #

1. 创建Email批量加载器（DataLoader）

继承BatchDataLoader<int, Email>，实现批量加载逻辑：

public class EmailDataLoader : BatchDataLoader<int, Email>
{
    private readonly UserService _userService;

    public EmailDataLoader(IBatchScheduler batchScheduler, UserService userService)
        : base(batchScheduler)
    {
        _userService = userService;
    }

    protected override async Task<IReadOnlyDictionary<int, Email>> LoadBatchAsync(
        IReadOnlyList<int> userIds,
        CancellationToken cancellationToken)
    {
        // 批量获取目标用户的邮箱列表
        var emails = await _userService.GetUsersEmails(userIds);
        // 转换为以UserId为键的字典，便于快速匹配到对应用户
        return emails.ToDictionary(e => e.UserId);
    }
}

2. 修改User类的邮箱解析方法

移除原有的直接调用UserService的逻辑，改用DataLoader来加载邮箱：

public class User
{
    public int Id { get; set; }
    public string Name { get; set; }

    // 通过[DataLoader]特性注入批量加载器
    public async Task<Email> GetEmailAsync([DataLoader] EmailDataLoader emailLoader)
    {
        // 加载当前用户对应的邮箱，DataLoader会自动批量处理多个请求
        return await emailLoader.LoadAsync(Id);
    }
}

3. 注册DataLoader到GraphQL服务

在服务配置中添加DataLoader的注册：

// 在Program.cs或Startup.cs的服务配置代码段中
builder.Services.AddGraphQLServer()
    .AddQueryType<Query>() // 替换为你项目中的实际查询类型
    .AddType<User>()
    .AddDataLoader<EmailDataLoader>(); // 注册邮箱批量加载器

工作原理 #

当GraphQL查询需要返回多个User对象的Email字段时：

1. DataLoader自动收集所有待加载的User ID；
2. 一次性调用GetUsersEmails获取全部邮箱数据；
3. 将返回的邮箱列表按UserId映射为字典，再分配给对应的User对象；
4. 彻底避免了每个User单独触发一次数据库查询的N+1问题。

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