Postgres 中的 Go 和 IN 子句

来到golang学习网的大家，相信都是编程学习爱好者，希望在这里学习Golang相关编程知识。下面本篇文章就来带大家聊聊《Postgres 中的 Go 和 IN 子句》，介绍一下，希望对大家的知识积累有所帮助，助力实战开发！

我正在尝试使用 pq 驱动程序对 go 中的 postgresql 数据库执行以下查询：

select count(id)
from tags
where id in (1, 2, 3)

其中 1, 2, 3 在切片 tags := []string{"1", "2", "3"} 处传递。

我尝试过很多不同的事情，例如：

s := "(" + strings.join(tags, ",") + ")"
if err := db.queryrow(`
    select count(id)
    from tags
    where id in $1`, s,
).scan(&num); err != nil {
    log.println(err)
}

这会导致 pq: 在“$1” 处或附近出现语法错误。我也尝试过

if err := Db.QueryRow(`
    SELECT COUNT(id)
    FROM tags
    WHERE id IN ($1)`, strings.Join(stringTagIds, ","),
).Scan(&num); err != nil {
    log.Println(err)
}

也因 pq 失败：整数输入语法无效：“1,2,3”

我还尝试直接传递整数/字符串切片，并得到 sql: conversion exec argument #0's type: unsupported type []string, a slice。

那么我如何在 go 中执行这个查询呢？

解决方案

预构建 sql 查询（防止 sql 注入）

如果您要为每个值生成带有参数占位符的 sql 字符串，则立即生成最终 sql 会更容易。

请注意，由于值是 strings，因此存在 sql 注入攻击的地方，因此我们首先测试所有 string 值是否确实是数字，只有这样我们才继续：

tags := []string{"1", "2", "3"}
buf := bytes.newbufferstring("select count(id) from tags where id in(")
for i, v := range tags {
    if i > 0 {
        buf.writestring(",")
    }
    if _, err := strconv.atoi(v); err != nil {
        panic("not number!")
    }
    buf.writestring(v)
}
buf.writestring(")")

执行：

num := 0
if err := db.queryrow(buf.string()).scan(&num); err != nil {
    log.println(err)
}

使用 any

您还可以使用Postgresql's ANY，其语法如下：

expression operator any (array expression)

使用它，我们的查询可能如下所示：

select count(id) from tags where id = any('{1,2,3}'::int[])

在这种情况下，您可以将数组的文本形式声明为参数：

select count(id) from tags where id = any($1::int[])

可以简单地像这样构建：

tags := []string{"1", "2", "3"}
param := "{" + strings.join(tags, ",") + "}"

请注意，在这种情况下不需要检查，因为数组表达式不允许 sql 注入（而是会导致查询执行错误）。

完整代码：

tags := []string{"1", "2", "3"}

q := "select count(id) from tags where id = any($1::int[])"
param := "{" + strings.join(tags, ",") + "}"

num := 0
if err := db.queryrow(q, param).scan(&num); err != nil {
    log.println(err)
}

这并不是真正的 golang 问题，您正在使用字符串与 sql 请求中的整数 (id) 进行比较。这意味着，sql 接收：

select count(id)
from tags
where id in ("1, 2, 3")

而不是你想要给它的东西。您只需将标签转换为整数并将其传递给查询即可。

编辑： 由于您尝试将多个值传递给查询，那么您应该告诉它：

params := make([]string, 0, len(tags))
for i := range tags {
    params = append(params, fmt.sprintf("$%d", i+1))
}
query := fmt.sprintf("select count(id) from tags where id in (%s)", strings.join(params, ", "))

这将以“($1, $2, $3...”结束查询，然后将标签转换为 int：

values := make([]int, 0, len(tags))
for _, s := range tags {
    val, err := strconv.atoi(s)
    if err != nil {
        // do whatever is required with the error
        fmt.println("err : ", err)
    } else {
        values = append(values, val)
    }
}

最后，您可以在查询中使用它：

Db.QueryRow(query, values...)

这应该可以。

本篇关于《Postgres 中的 Go 和 IN 子句》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！