【Golang】使用GORM实现高效联表查询

原创已于 2025-05-01 15:59:16 修改 · 535 阅读

10 ·

CC 4.0 BY-SA版权

文章标签：

#golang #windows #开发语言

于 2024-12-03 13:35:59 首次发布

Golang 专栏收录该内容

31 篇文章

订阅专栏

场景模拟

假设我们有一个在线教育平台，需要展示大学及其扩展信息，涉及以下实体：

大学基本信息表（university）
大学详情表（university_details）

模型定义

// 大学基本信息模型
type University struct {
    UUID     string    `gorm:"primaryKey"`
    Name     string    `gorm:"size:100"`
    Focus    int       // 关注指数
    Status   *int      // 启用状态（0停用 1启用）
    CreateAt time.Time
    // ...其他基础字段
}

// 大学详情模型
type UniversityDetail struct {
    CollegeID   string `gorm:"column:college_id"` // 关联大学UUID
    Remark      string
    FacultyNum  int    // 教师数量
    StudentNum  int    // 学生数量
    ProgramNum  int    // 项目数量
}

查询实现

1. 基础查询构建

func (s *EduService) GetUniversityList(filter models.UniversityFilter) ([]UniversityDTO, int64, error) {
    var result []struct {
        University
        UniversityDetail
    }

    // 分页参数处理
    pageSize := filter.PageSize
    offset := filter.PageSize * (filter.Page - 1)

    // 联表查询语句
    joinClause := "LEFT JOIN university_details ON university.uuid = university_details.college_id"
    selectFields := `
        university.*,
        COALESCE(university.status, 0) as status,
        university_details.remark,
        university_details.faculty_num,
        university_details.student_num,
        university_details.program_num
    `

    query := global.DB.Model(&University{}).
        Joins(joinClause).
        Select(selectFields)

2. 条件过滤处理

    // 时间范围过滤
    if filter.StartTime != nil && filter.EndTime != nil {
        query = query.Where("created_at BETWEEN ? AND ?", 
            filter.StartTime, filter.EndTime)
    }

    // 状态过滤
    if filter.ActiveOnly {
        query = query.Where("status IS NULL OR status = ?", 1)
    }

    // 模糊搜索
    if filter.Name != "" {
        query = query.Where("name LIKE ?", "%"+filter.Name+"%")
    }

    // 合作类型过滤
    if filter.PartnerType != nil {
        query = query.Where("partner_type = ?", *filter.PartnerType)
    }

3. 分页与排序

    // 总数统计
    var total int64
    if err := query.Count(&total).Error; err != nil {
        return nil, 0, fmt.Errorf("查询总数失败: %w", err)
    }

    // 动态排序处理
    if filter.SortBy != "" && filter.SortOrder != "" {
        // 驼峰转蛇形命名
        re := regexp.MustCompile("([a-z0-9])([A-Z])")
        fieldName := re.ReplaceAllString(filter.SortBy, "${1}_${2}")
        fieldName = strings.ToLower(fieldName)
        
        query = query.Order(fmt.Sprintf("%s %s", fieldName, filter.SortOrder))
    } else {
        // 默认排序：启用状态优先，关注度降序
        query = query.Order("status DESC").Order("focus DESC")
    }

    // 执行查询
    err := query.
        Limit(pageSize).
        Offset(offset).
        Find(&result).
        Error

    return result, total, err