在go-gorm/gen中使用原生SQL进行复杂查询

前言

在数据库操作中，有时会遇到一些复杂的查询场景，比如需要处理日期函数、自定义聚合或者特殊的字段转换。这些情况下，ORM自动生成的查询可能无法满足需求。本文将详细介绍如何在go-gorm/gen框架中使用原生SQL进行复杂查询。

基本概念

go-gorm/gen是基于GORM的代码生成工具，它提供了类型安全的查询构建方式。但在某些特殊场景下，我们仍然需要直接使用原生SQL来实现特定功能。

使用原生SQL的场景

1. 需要使用数据库特定的函数（如YEAR、DATE_FORMAT等）
2. 需要进行复杂的字段转换或计算
3. 需要执行数据库特有的聚合操作
4. 需要优化特定查询性能时

实现方式

1. 通过UnderlyingDB方法获取原生GORM实例

gen生成的DAO对象提供了UnderlyingDB方法，可以获取底层的*gorm.DB实例，从而使用GORM的所有原生功能。

type Result struct {
    Id        int
    Year      int
    FirstName string
}

var result Result
author.WithContext(ctx).Where(author.ID.Eq(1)).UnderlyingDB().
    Select([]string{
        string(author.ID.ColumnName()),
        "YEAR(CAST(birth_date as DATE)) as year",
        "first_name",
    }).Debug().Scan(&result)

2. 自定义Select字段

在Select方法中，可以混合使用gen生成的字段名和原生SQL表达式：

• 使用string(author.ID.ColumnName())获取gen生成的字段名
• 直接写入原生SQL表达式，如"YEAR(CAST(birth_date as DATE)) as year"
• 也可以直接使用数据库字段名，如"first_name"

3. 复杂查询示例

假设我们需要查询每年发布的文章数量：

type YearCount struct {
    Year  int
    Count int
}

var results []YearCount
article.WithContext(ctx).UnderlyingDB().
    Select("YEAR(CAST(release_time as DATE)) as year, COUNT(*) as count").
    Group("year").
    Scan(&results)

注意事项

1. 使用原生SQL时要注意SQL注入风险，避免直接拼接用户输入
2. 跨数据库兼容性问题：某些数据库函数可能在其他数据库中不可用
3. 结果映射需要确保结构体字段与查询结果的列名匹配
4. 对于复杂查询，建议添加适当的索引以提高性能

最佳实践

1. 尽量将复杂SQL封装到模型方法中
2. 为复杂查询编写单元测试
3. 添加适当的注释说明查询目的
4. 考虑使用数据库视图(View)来简化复杂查询

通过合理使用原生SQL，我们可以在保持go-gorm/gen类型安全优势的同时，处理各种复杂的查询需求。