最近工作遇到一个小问题,即如何使用原生的sql查询where in语句,因为之前使用gorm习惯了,gorm已经封装好了,突然写原生的反而有点不熟悉,同时还要考虑到性能和代码是否繁琐,所以写这个笔记记录一下当时的几种解决方法。
场景
我传入in的范围是一个数组,里面值的类型为int64型,例如如下:
idSlice := []int{1, 2, 3, 4, 5, 6, 7}
正常的sql语句是这样写的:
select * from table where id in (1, 2, 3, 4, 5, 6, 7);
于是我想当然的也在代码这样写:
idSlice := []int{1, 2, 3, 4, 5, 6, 7}
query := fmt.Sprintf("select * from table where id in (?)", idSlice)
result, err := db.Query(query)
上面其实犯了两个错误,
- 即原生sql并不能识别(?),这在gorm可以是因为已经封装好了,所以经过多次试错和思考之后,反应过来,其实in的范围要写成字符串的形式,毕竟是把整条sql当成原生的查询语句,所以不能出现变量之类的东西,
- idSlice为数组类型,如果把(?)改成(%v),就会变成
in [1, 2, 3, 4, 5, 6, 7]
,同样不可以
所以我总结了几种方法去解决这些问题:
方法一
idSlice := []int{1, 2, 3, 4, 5, 6, 7}
var s []string
for i := 0; i < len(idSlice); i++ {
s = append(s, strconv.Itoa(idSlice[i])) //把每个元素都变成string类型
}
var ss string
ss = strings.Join(s, "','") //这里填入的s必须要为string类型的数组,所以前面要转换成string类型
//此时的ss为:1','2','3','4','5','6','7
query := fmt.Sprintf("select * from table where id in ('%s')", ss)
//组合之后:('1','2','3','4','5','6','7')
result, err := db.Query(query)
这种方法的问题在于使用了strings.Joins函数,而函数的第一个值必须为string类型的数值,所以需要用for循环进行一次转换,同时strings.Joins函数自己内部也有for循环,这里加起来就有两个for循环了,这样对整体的性能就有影响了。
方法二
idSlice := []int{1, 2, 3, 4, 5, 6, 7}
var ss string
for i := 0; i < len(idSlice); i++ {
if i == 0 {
s := fmt.Sprintf("'%d'", idSlice[i])
ss += s
} else {
s := fmt.Sprintf(",'%d'", idSlice[i])
ss += s
}
}
query := fmt.Sprintf("select * from table where id in (%s)", ss)
result, err := db.Query(query)
这个方法的优点是没有使用strings.Joins函数,只用了一个for循环直接拼接,所以整体的性能是更优的。
...