将数组传递给 go-pg 查询

Question

我正在使用Go-pg，当我使用以下方式执行sql查询时：

db.Query(&result, sqlQuery, params)

其中params 是如下结构：

type MyParams struct {
    Codes []int
}

而sqlQuery 是

SELECT id FROM bar WHERE code_id IN (?codes)

在实际的 SQL 查询中，我得到这样的查询：

SELECT id FROM bar WHERE code_id IN ('[7,45]')

是否可以正确传递 int 数组占位符以进行查询：

SELECT id FROM bar WHERE code_id IN (7,45)

Answer 1

你可以做几件事：

1. 继续使用in (...) 并使用pg.In。
2. 在您的 SQL 中使用 = any(array) 而不是在 Go 中使用 in (list) 和 pg.Array 将正确的数组发送到 PostgreSQL。

第一个看起来像：

db.Query(&result, `SELECT id FROM bar WHERE code_id IN (?)`, pg.In(params.Codes))

第二个看起来像：

db.Query(&result, `SELECT id FROM bar WHERE code_id = ANY (?)`, pg.Array(params.Codes))

Answer 2

你可以使用 go-pg ORM 来代替相同的结果：

ids := []int{1, 2, 3}
err := db.Model((*Book)(nil)).
    Where("id in (?)", pg.In(ids)).
    Select()

// SELECT * FROM books WHERE id IN (1, 2, 3)

Answer 3

一种方法是创建一个带有分隔符的字符串以创建一个字符串，然后将其传递给db.query 以获取结果。

package main

import (
    "fmt"
    "strconv"
    "strings"
)

func main() {
    a := []int{7,45,32}
    str := ConvertToString(a, ",")
    query := `Select * from table1 where ID IN(`+ str +`)`
    fmt.Println(query)
}

func ConvertToString(a []int, delim string) string{
    var b string
    for _, v := range a{
        b += strconv.Itoa(v)
        b += ","
    }
    return strings.Trim(b,",")
}

Playground 上的工作代码。

已编辑：-

您可以使用Golang sqlx 包来满足您的要求。

database/sql 包不检查您的查询，它通过了您的 直接向驱动程序提供参数，它处理查询 IN 子句难：

SELECT * FROM users WHERE level IN (?);

当 this 被准备为后端的语句时， bindvar ? 只会对应一个参数，但通常是需要的 是因为它是可变数量的参数，具体取决于 一些切片的长度，例如：

var levels = []int{4, 6, 7}
rows, err := db.Query("SELECT * FROM users WHERE level IN (?);", levels)