最近工作遇到一个小问题,即如何使用原生的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)
上面其实犯了两个错误,
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循环直接拼接,所以整体的性能是更优的。