如何在Gorm+GraphQL中跳过不必要的Players预加载以降低MySQL负载

可以通过gqlgen提供的上下文工具，判断客户端是否请求了players字段，再动态决定是否执行GORM的Preload操作，具体实现如下：

1. 首先导入gqlgen的graphql包：

import "github.com/99designs/gqlgen/graphql"

2. 修改Teams resolver函数，添加字段判断逻辑：

func (r *queryResolver) Teams(ctx context.Context) ([]*model.Team, error) {
    var teams []*model.Team
    // 初始化基础查询
    dbQuery := r.DB.Model(&model.Team{})

    // 获取当前查询中客户端请求的teams子字段
    opCtx := graphql.GetOperationContext(ctx)
    fieldCtx := graphql.GetFieldContext(ctx)
    selectedFields := graphql.CollectFields(opCtx, fieldCtx.Field.Selections, nil)

    // 检查是否包含players字段
    needPreloadPlayers := false
    for _, field := range selectedFields {
        if field.Name == "players" {
            needPreloadPlayers = true
            break
        }
    }

    // 按需添加预加载
    if needPreloadPlayers {
        dbQuery = dbQuery.Preload("Players")
    }

    // 执行查询
    if err := dbQuery.Find(&teams).Error; err != nil {
        return nil, err
    }

    return teams, nil
}

原理说明 #

gqlgen会将客户端的查询结构信息存入请求上下文中，通过graphql.CollectFields可以提取出当前teams字段下被请求的所有子字段。我们只需要判断其中是否包含players，就能决定是否触发GORM的关联预加载，从而避免不必要的MySQL查询，降低服务器负载。

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