您当前的位置:首页 > 计算机 > 软件应用 > 数据库 > 其它

Gorm 多个 Select 查询动态地拼 Sql

时间:12-14来源:作者:点击数:

在代码里,如果我想 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 语句了

方便获取更多学习、工作、生活信息请关注本站微信公众号城东书院 微信服务号城东书院 微信订阅号
推荐内容
相关内容
栏目更新
栏目热门
本栏推荐