如何编写支持动态变量WHERE子句的GORM通用函数？

当然可以！GORM本身就提供了足够灵活的方式来处理动态多条件查询，我们可以轻松写出通用函数来满足你的需求。下面给你几种不同场景的实现方案：

方案1：使用Map传递动态条件（最简单） #

GORM会自动忽略Map中值为零值（比如0、""、nil）的键，非常适合快速构建动态查询：

// 假设你的User模型定义如下
type User struct {
    ID     int
    Name   string
    Gender string
    Age    int
}

// GetUsers 根据动态条件查询用户列表
func GetUsers(db *gorm.DB, conditions map[string]interface{}) ([]User, error) {
    var users []User
    err := db.Where(conditions).Find(&users).Error
    return users, err
}

// 使用示例：
// 1. 查询 id=10 AND name="Chetan"
users, err := GetUsers(dbDriver, map[string]interface{}{"id": 10, "name": "Chetan"})
// 2. 只查询 id=10
users, err := GetUsers(dbDriver, map[string]interface{}{"id": 10})
// 3. 查询 gender="male" AND name="Chetan" AND age=30
users, err := GetUsers(dbDriver, map[string]interface{}{"gender": "male", "name": "Chetan", "age": 30})

注意：如果你的业务场景中需要保留零值条件（比如查询age=0的用户），这种方法会自动忽略该条件，此时可以选择方案2或方案3。

方案2：使用Struct传递类型安全的条件 #

如果需要更严格的类型检查，可以定义一个查询结构体，GORM会基于结构体的非零值字段生成查询条件：

// UserQuery 定义用户查询的结构体
type UserQuery struct {
    ID     int    // 对应users表的id字段
    Name   string // 对应name字段
    Gender string // 对应gender字段
    Age    int    // 对应age字段
}

// GetUsersByStruct 根据结构体条件查询用户
func GetUsersByStruct(db *gorm.DB, query UserQuery) ([]User, error) {
    var users []User
    err := db.Where(query).Find(&users).Error
    return users, err
}

// 使用示例：
users, err := GetUsersByStruct(dbDriver, UserQuery{ID:10, Name:"Chetan"})
users, err := GetUsersByStruct(dbDriver, UserQuery{ID:10})
users, err := GetUsersByStruct(dbDriver, UserQuery{Gender:"male", Name:"Chetan", Age:30})

如果需要忽略结构体中的某些字段，可以添加gorm:"-"标签；如果需要保留零值条件，可以在结构体字段上添加gorm:"default:null"或者改用Map方案。

方案3：自定义条件结构体支持复杂操作符 #

如果你的查询需要支持大于、小于、模糊匹配（LIKE）等复杂操作符，可以自定义一个条件结构体来构建灵活的查询链：

import "fmt"

// Condition 定义单个查询条件
type Condition struct {
    Column string // 数据库列名
    Op     string // 操作符，比如 "=", ">", "<", "LIKE"
    Value  interface{} // 条件值
}

// GetUsersWithCustomConditions 支持复杂操作符的动态查询
func GetUsersWithCustomConditions(db *gorm.DB, conditions []Condition) ([]User, error) {
    var users []User
    tx := db.Model(&User{})
    
    // 遍历所有条件，链式调用Where
    for _, cond := range conditions {
        tx = tx.Where(fmt.Sprintf("%s %s ?", cond.Column, cond.Op), cond.Value)
    }
    
    err := tx.Find(&users).Error
    return users, err
}

// 使用示例：
// 1. 查询 id=10 AND name="Chetan"
conditions := []Condition{
    {"id", "=", 10},
    {"name", "=", "Chetan"},
}
users, err := GetUsersWithCustomConditions(dbDriver, conditions)

// 2. 查询 age > 25 AND gender="male"
conditions := []Condition{
    {"age", ">", 25},
    {"gender", "=", "male"},
}
users, err := GetUsersWithCustomConditions(dbDriver, conditions)

// 3. 查询 name LIKE "%Che%"
conditions := []Condition{
    {"name", "LIKE", "%Che%"},
}
users, err := GetUsersWithCustomConditions(dbDriver, conditions)

这种方式完全自定义查询逻辑，同时通过?占位符避免了SQL注入风险，非常适合复杂的业务场景。

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