Gorm 多个 Select 查询动态地拼 Sql

  • Z时代
  • 2024-01-10
  • 分类：综合

在代码里，如果我想 select 的数量是不确定的，是根据一个传入的列表动态决定的，有可能是6个数有可能是8个数，这时有可能会想到动态地拼 gorm 的 select 语句：

var db *gorm.DB
var conditionList [][]int32 // 省略初始化
for _, val := range conditionList {
    db = db.Select("count(product_type in (?), orders.id, null))", val)
}

也就是，拼出来之后得到一个类似这样的 gorm 语句：

db = db.Select("count XXX").Select("count XXX").Select(......)

但是 ，这样运行之后会发现实际的 sql 语句只会有一个 select，其他的 select 都没有了，相当于只剩下：

db = db.Select("count XXX")

这是因为在 gorm 里 select 只能指定（assign）一次。所以剩下的方法就是，自己拼一个大的字符串形式的 select 语句：

var db *gorm.DB
var conditionList [][]int32 // 省略初始化
selectSql := ""
for _, val := range conditionList {
    if len(selectSql) == 0 {
        selectSql += fmt.Sprintf("count(product_type in (%v), orders.id, null))", val)
    } else {
        selectSql += fmt.Sprintf(",count(product_type in (%v), orders.id, null))", val)
    }
}
db = db.Select(selectSql)

当然，这样直接拼字符串会有 sql 注入的风险，更好的方法是使用 gorm 自带的拼接，实现方式是把动态条件统一放到一个 interface{} 的列表中，比如：

var db *gorm.DB
var conditionList [][]int32 // 省略初始化
selectSql := ""
selectCondition := make([]interface{}, 0)
for _, val := range conditionList {
    if len(selectSql) == 0 {
        selectSql += "count(product_type in (?), orders.id, null))"
    } else {
        selectSql += ",count(product_type in (?), orders.id, null))"
    }
    selectCondition = append(selectCondition, val)
}
db = db.Select(selectSql, selectCondition...)

这样就能动态地拼接多个 select 语句了

以上是 Gorm 多个 Select 查询动态地拼 Sql 的全部内容， 来源链接： utcz.com/z/264528.html

回到顶部