golang实现mysql where in查询

素履coder

发布于 2022-02-17 06:20:26

2.2K00

代码可运行

文章被收录于专栏：素履coder素履coder

运行总次数：0

代码可运行

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