Go-PG包JOIN查询优化求助：重构批量学生团队校验逻辑

问题背景 #

当前实现通过循环逐个发起数据库查询，判断每个指定ID的学生是否在目标团队中，这种方案会产生与输入ID数量等同的数据库请求，在ID数量较多时会严重影响性能，需要优化为单次数据库查询完成批量判断。

先修正结构体的拼写错误 #

注意到Team结构体中StudentID字段的JSON标签拼写错误（strudent_id应为student_id），这会导致ORM映射异常，先修正：

type Team struct {
   ID int `json:"id"`
   StudentID int `json:"student_id"` // 修正拼写错误
   TeamName string `json:"team_name"`
   CreateDate time.Time `json:"create_date"`
}

优化后的实现方案 #

利用单次批量查询结合JOIN，一次性获取所有指定UserID的学生是否在目标团队中的状态，再将结果转换为目标map结构：

func (p *storage) FindStudentsInTeam(ids []int, teamID int) (map[int]bool, error) {
    // 初始化结果map，默认所有ID都不在团队中
    studentsInTeam := make(map[int]bool, len(ids))
    for _, id := range ids {
        studentsInTeam[id] = false
    }

    // 定义接收查询结果的结构体切片
    var results []struct {
        UserID int `pg:"user_id"`
    }

    // 单次查询：找出所有在目标团队中的学生UserID
    err := p.Model((*Student)(nil)).
        Column("students.user_id").
        Join("JOIN teams ON teams.student_id = students.id").
        Where("teams.id = ?", teamID). // 若团队ID字段是`team_id`而非`id`，请改为`teams.team_id = ?`
        Where("teams.create_date <= ?", time.Now().UTC()).
        Where("students.user_id IN (?)", pg.In(ids)).
        Select(&results)
    if err != nil {
        return nil, err
    }

    // 将查询到的UserID标记为存在于团队中
    for _, res := range results {
        studentsInTeam[res.UserID] = true
    }

    return studentsInTeam, nil
}

方案优势 #

• 减少数据库请求次数：从N次请求变为1次，大幅降低数据库连接开销和网络往返延迟
• 利用数据库批量处理能力：数据库对批量查询的优化远优于应用层循环查询
• 逻辑更简洁：避免循环中的重复查询逻辑，代码可维护性更强

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