在 pgx lib 中命名准备好的语句,它是如何工作的?

Tim*_*Tim 5 postgresql go pgx

介绍

数据库/sql

在 Go 标准 sql 库中,该*Stmt类型具有如下定义的方法:

func (s *Stmt) Exec(args ...interface{}) (Result, error)
func (s *Stmt) Query(args ...interface{}) (*Rows, error)
Run Code Online (Sandbox Code Playgroud)

新的(未命名的)语句由以下人员准备:

func (db *DB) Prepare(query string) (*Stmt, error)
Run Code Online (Sandbox Code Playgroud)
  1. 连接池是抽象的,不能直接访问
  2. 在单个连接上准备事务
  3. 如果连接在语句执行时不可用,它将在新连接上重新准备。

PGX

该PreparedStatement类型没有定义任何方法。一个新命名的准备好的语句是由以下方式准备的:

func (p *ConnPool) Prepare(name, sql string) (*PreparedStatement, error)
Run Code Online (Sandbox Code Playgroud)
  1. 操作直接在连接池上
  2. 事务在池的所有连接上准备就绪
  3. 没有明确的方法如何执行准备好的语句

在Github 评论中,作者更好地解释了 pgx 和 database/sql 之间架构的差异。的文档Prepare还指出(强调我的):

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?
}
Run Code Online (Sandbox Code Playgroud)

问题)

  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
                }
            }
Run Code Online (Sandbox Code Playgroud)
  1. PosgreSQL 本身需要它来做某事(假设)。

Tim*_*Tim 6

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

因此,您可以有效地执行以下操作:

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)
}
Run Code Online (Sandbox Code Playgroud)