pgx lib 中准备语句的执行过程是怎样的？

在 Go 的 `database/sql` 库中，准备语句由 `*stmt` 类型表示，具有 `exec` 和 `query` 方法。而 `pgx` 库中，准备语句由 `preparedstatement` 类型表示，没有明确的方法执行。`pgx` 文档表明准备是幂等的，允许在不关心语句是否已准备好的情况下使用相同的名称和 SQL 参数多次调用 `prepare`。但是，`query` 和 `exec` 方法没有接受名称参数，让人困惑如何匹配准备好的语句。

简介

数据库/sql

在go标准sql库中，*stmt类型的方法定义如下：

func (s *stmt) exec(args ...interface{}) (result, error)
func (s *stmt) query(args ...interface{}) (*rows, error)

新的（未命名）语句由以下内容准备：

func (db *db) prepare(query string) (*stmt, error)

1. 连接池是抽象的，无法直接访问
2. 事务是在单个连接上准备的
3. 如果连接在语句执行时不可用，则会在新连接上重新准备。

pgx

preparedstatement 类型没有定义任何方法。一个新的命名准备好的语句是通过以下方式准备的：

func (p *connpool) prepare(name, sql string) (*preparedstatement, error)

1. 直接在连接池上进行操作
2. 在池的所有连接上准备事务
3. 没有明确的方法来执行准备好的语句

在github评论中，作者更好地解释了pgx和database/sql之间架构的差异。 prepare 的文档还指出（强调我的）：

准备是幂等的；即，使用相同的名称和 sql 参数多次调用“prepare”是安全的。这允许prepare和query/exec/prepareex的代码路径，而无需关心语句是否已经准备好。

小例子

package main

import (
    "github.com/jackc/pgx"
)

func main() {
    conf := pgx.connpoolconfig{
        connconfig: pgx.connconfig{
            host:     "/run/postgresql",
            user:     "postgres",
            database: "test",
        },
        maxconnections: 5,
    }
    db, err := pgx.newconnpool(conf)
    if err != nil {
        panic(err)
    }
    _, err = db.prepare("my-query", "select $1")
    if err != nil {
        panic(err)
    }
    // what to do with the prepared statement?
}

问题

1. name 参数给我的印象是它可以通过 name 调用来执行，但是如何执行？
2. 文档给人的印象是 query/exec 方法以某种方式利用了准备好的语句。但是，这些方法不采用 name 参数。它们如何匹配？
3. 推测，匹配是通过查询内容完成的。那么命名语句的全部意义是什么？

可能的答案

这就是我的进步：

1. 没有方法通过名称引用查询（假设）
2. 匹配是在 conn.execex() 中的查询主体上完成的。如果还没有准备好，就会完成：

ps, ok := c.preparedStatements[sql]
            if !ok {
                var err error
                ps, err = c.prepareEx("", sql, nil)
                if err != nil {
                    return "", err
                }
            }

1. posgresql 本身需要它来做某事（假设）。

解决方案

@mkopriva 指出 sql 文本误导了我。它在这里有双重功能。如果 sql 变量与 c.preparedstatements[sql] 映射中的键不匹配，则准备好 sql 中包含的查询，并将新的 *preparedstatement 结构指定给 ps。如果它确实匹配某个键，则 ps 变量将指向映射的条目。

因此，您可以有效地执行以下操作：

package main

import (
    "fmt"

    "github.com/jackc/pgx"
)

func main() {
    conf := pgx.ConnPoolConfig{
        ConnConfig: pgx.ConnConfig{
            Host:     "/run/postgresql",
            User:     "postgres",
            Database: "test",
        },
        MaxConnections: 5,
    }
    db, err := pgx.NewConnPool(conf)
    if err != nil {
        panic(err)
    }
    if _, err := db.Prepare("my-query", "select $1::int"); err != nil {
        panic(err)
    }
    row := db.QueryRow("my-query", 10)
    var i int
    if err := row.Scan(&i); err != nil {
        panic(err)
    }
    fmt.Println(i)
}

好了，本文到此结束，带大家了解了《pgx lib 中准备语句的执行过程是怎样的？》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！