最近工作遇到一个小问题,即如何使用原生的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)

上面其实犯了两个错误,

  1. 即原生sql并不能识别(?),这在gorm可以是因为已经封装好了,所以经过多次试错和思考之后,反应过来,其实in的范围要写成字符串的形式,毕竟是把整条sql当成原生的查询语句,所以不能出现变量之类的东西,
  2. 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循环直接拼接,所以整体的性能是更优的。