GORM Gen 中 OR 条件查询的正确使用方式

在使用 GORM Gen 进行数据库查询时，开发者经常会遇到需要构建复杂查询条件的情况，特别是需要组合 AND 和 OR 条件的场景。本文将通过一个典型示例，深入分析如何正确使用 GORM Gen 构建包含 OR 条件的查询。

问题背景

假设我们有一个数据结构表示某个资源的生命周期：

type MyStruct struct {
  StartTime time.Time
  DecommissionTime *time.Time
}

业务需求是查询在指定时间点（如2023-01-01 00:00:00）有效的所有记录。有效的定义是：

1. 记录的开始时间早于查询时间点
2. 记录要么未被停用（DecommissionTime为NULL），要么停用时间晚于查询时间点

对应的SQL应该是：

SELECT count(*) FROM "my_structs" WHERE 
  ("decommission_time" IS NULL OR "decommission_time" > '2023-01-01 00:00:00') 
  AND "start_time" < '2023-01-01 00:00:00'

常见错误实现

许多开发者会尝试以下方式构建查询：

var conditions = make([]gen.Condition, 2)
conditions = append(conditions, query.MyStruct.Or(
  query.MyStruct.DecommissionTime.IsNull(),
  query.MyStruct.DecommissionTime.Gt(period.StartTime)
))
conditions = append(conditions, query.MyStruct.DeployTime.Lt(*period.EndTime))
return query.MyStruct.Debug().Where(conditions...).Count()

但实际生成的SQL却是：

SELECT count(*) FROM "my_structs" WHERE 
  "my_structs"."decommission_time" IS NULL 
  AND "my_structs"."decommission_time" > '2023-01-01 00:00:00' 
  AND "my_structs"."deploy_time" < '2023-01-01 00:00:00'

这显然不符合预期，所有条件都被AND连接起来了。

问题根源分析

通过查看GORM Gen的源代码，可以发现Or方法的实现方式导致了这个问题。在底层，它使用了clause.Or包裹clause.And，这种嵌套方式实际上无法实现真正的OR条件组合。

正确实现方式

GORM Gen提供了更直观的条件组合方式，可以更清晰地表达查询意图：

query.MyStruct.Where(
  // 第一个条件组，内部使用OR连接
  query.MyStruct.
    Where(query.MyStruct.DecommissionTime.IsNull()).
    Or(query.MyStruct.DecommissionTime.Gt(period.StartTime)),
  
  // 第二个条件，与前一个条件组用AND连接
  query.MyStruct.DeployTime.Lt(*period.EndTime)
).Count()

这种方式生成的SQL正是我们期望的：

SELECT count(*) FROM "my_structs" WHERE 
  ("decommission_time" IS NULL OR "decommission_time" > '2023-01-01 00:00:00') 
  AND "start_time" < '2023-01-01 00:00:00'

最佳实践建议

1. 条件分组：使用Where方法创建条件组，组内条件默认使用AND连接
2. 显式OR：在需要OR连接的条件上明确使用Or方法
3. 链式调用：利用GORM Gen的链式调用特性，使查询构建更加清晰
4. 调试输出：在复杂查询中使用Debug()方法查看生成的SQL，验证是否符合预期

通过理解GORM Gen的条件组合机制，开发者可以构建出各种复杂的查询条件，满足业务需求的同时保持代码的可读性和可维护性。