如何在 jackc/pgx 中使用“where id in”子句？

学习Golang要努力，但是不要急！今天的这篇文章《如何在 jackc/pgx 中使用“where id in”子句？》将会介绍到等等知识点，如果你想深入学习Golang，可以关注我！我会持续更新相关文章的，希望对大家都能有所帮助！

pgx 是否提供对“where in”子句的支持？我在另一个 stackoverflow 线程中发现应该使用字符串连接来手动构建查询。 imo 这有点容易出错，因为你必须自己处理转义/sql 注入等问题。

我也尝试自己解决：

const updatepurgedrecordingsstmt = "update recordings set status = 'deleted', deleted = now() where status <> 'deleted' and id in ($1);"

func (r *repository) deleterecordings() error {
    pool, err := r.connpool()
    if err != nil {
        return errors.wrap(err, "cannot establish connection")
    }
    pgrecids := &pgtype.int4array{}

    if err := pgrecids.set([]int32{int32(1), int32(2)}); err != nil {
        return errors.wrap(err, "id conversion failed")
    }
    if _, err = pool.exec(updatepurgedrecordingsstmt, pgrecids); err != nil {
        return errors.wrap(err, "update stmt failed")
    }
    return nil
}

当我执行此代码时，我收到以下错误：

错误：绑定参数 1 中的二进制数据格式不正确 (sqlstate 22p03)

我使用的版本：

postgres：

db=> select version();
                                              version
-----------------------------------------------------------------------------------------------------------
 postgresql 9.6.11 on x86_64-pc-linux-gnu, compiled by gcc (gcc) 4.8.2 20140120 (red hat 4.8.2-16), 64-bit
(1 row)

pgx：

github.com/jackc/fake v0.0.0-20150926172116-812a484cc733 h1:vr3AYkKovP8uR8AvSGGUK1IDqRa5lAAvEkZG1LKaCRc=
github.com/jackc/fake v0.0.0-20150926172116-812a484cc733/go.mod h1:WrMFNQdiFJ80sQsxDoMokWK1W5TQtxBFNpzWTD84ibQ=
github.com/jackc/pgx v3.3.0+incompatible h1:Wa90/+qsITBAPkAZjiByeIGHFcj3Ztu+VzrrIpHjL90=
github.com/jackc/pgx v3.3.0+incompatible/go.mod h1:0ZGrqGqkRlliWnWB4zKnWtjbSWbGkVEFm4TeybAXq+I=
github.com/lib/pq v1.0.0 h1:X5PMW56eZitiTeO7tKzZxFCSpbFZJtkMMooicw2us9A=
github.com/lib/pq v1.0.0/go.mod h1:5WUZQaWbwv1U+lTReE5YruASi9Al49XbQIvNi/34Woo=

解决方案

正如您所知，in 需要一个标量表达式列表，而不是一个数组，但是 pgtype.int4array 表示一个数组，而不是标量表达式列表。 p>

“在我看来，这有点容易出错，因为你必须自己处理转义/sql注入等问题。”

不一定，您可以循环数组，构造参数引用字符串，将其连接到查询，然后通过 ... 传入数组来执行它。

var paramrefs string
ids := []interface{}{1,2,3,4}
for i, _ := range ids {
    paramrefs += `$` + strconv.itoa(i+1) + `,`
}
paramrefs = paramrefs[:len(paramrefs)-1] // remove last ","

query := `update ... where id in (` + paramrefs + `)`
pool.exec(query, ids...)

或者，您可以使用 any 代替 in。

ids := &pgtype.int4array{}
ids.set([]int{1,2,3,4})
query := `update ... where id = any ($1)`
pool.exec(query, ids)

（这里你可能需要将参数引用强制转换为适当的数组类型，我不确定，请尝试不强制转换，如果不行，请尝试强制转换）

func prepareWhereINString(count int) string {
    var paramrefs string
    for i := 0; i < count; i++ {
        paramrefs += `$` + strconv.Itoa(i+1) + `,`
    }
    paramrefs = paramrefs[:len(paramrefs)-1] // remove last ","
    return paramrefs
}

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《如何在 jackc/pgx 中使用“where id in”子句？》文章吧，也可关注golang学习网公众号了解相关技术文章。