1

我想知道是否可以将参数传递给用 PL/pgSQL 编写的查询？

我试过这个，但它失败了pq: got 1 parameters but the statement requires 0

package main

import (
    "database/sql"
    "fmt"
    "log"

    _ "github.com/lib/pq"
)

func main() {
    db, err := sql.Open("postgres", "host=localhost dbname=db user=user sslmode=disable password=pw")
    if err != nil {
        log.Fatal(err)
    }
    row := db.QueryRow(`
DO $$
BEGIN
    IF true THEN
        SELECT $1;
    END IF;
END$$
`, 1)
    var num int
    err = row.Scan(&num)
    if err != nil {
        log.Fatal(err)
    }
    fmt.Println(num)
}

另一个相关的问题是我想使用事务，但是sql包提供的 API 似乎每次在 tx 中执行查询时都会连接到 db。如果可能的话，我希望所有事情都能一口气完成。例如，使用 go 你应该使用这样的事务

tx, err := db.Begin()
rows, err := tx.Query(sql1)
result, err := tx.Exec(sql2)
tx.Commit()

问题是，如果我没记错的话，调用tx.Query并两次访问 PostgreSQL 服务器。tx.Exec我想要实现的是合并sql1and sql2，将它们包装在里面BEGIN并END一次执行它们。我的问题是：

  1. 你觉得有必要吗？我想如果有足够的流量，性能差异可能会很明显，但我不确定。
  2. 如果是这样，执行此合并交易的最佳方式是什么？创建一个函数并在 PL/pgSQL 中运行事务（因为我可能需要使用条件语句等）？
4

1 回答 1

2

您收到错误，因为 PL/pgSQL 应该在服务器端定义为functionor procedure，但在您的情况下，它是从客户端调用的。下面是一个关于如何定义和调用带参数的函数的简单示例：

CREATE OR REPLACE FUNCTION myadd(a integer, b integer) RETURNS integer AS $$
    BEGIN
            RETURN a + b;
    END;
$$ LANGUAGE plpgsql;

然后，您可以从客户端使用SELECT查询调用带有参数的函数。请注意，即使您的函数包含INSERT/UPDATE，也必须使用 usingSELECT语句调用该函数。

//...
a := 10
row := db.QueryRow(`SELECT * FROM myadd($1, $2)`, a, 130)
//...

下一个问题，关于事务和 PL/pgSQL。是的，使用 PL/pgSQL 可以减少网络流量。服务器端语言 (PL/pgSQL) 的几个优点是：

  1. 消除客户端-服务器往返
  2. 无需将中间结果传递给客户端，只传递最终结果。
  3. 避免多次解析查询（发送查询到服务器 --> 服务器解析查询 --> 执行数据库操作 --> 将结果返回给客户端等...）

处理数据库（大数据）时的规则是You need to avoid to move your data around，PL/pgSQL 符合此规则。但是，在某些情况下您不能（需要避免）使用 PL/pgSQL，例如 DB 管理员/服务器所有者不允许服务器端编程（安全/性能原因等）。

function和之间的关系在手册transaction中有明确的说明：

重要的是不要将在 PL/pgSQL 中用于分组语句的 BEGIN/END 与用于事务控制的类似名称的 SQL 命令混淆。PL/pgSQL 的 BEGIN/END 仅用于分组；他们不会开始或结束交易。函数和触发器过程总是在由外部查询建立的事务中执行——它们不能启动或提交该事务，因为它们没有执行的上下文。但是，包含 EXCEPTION 子句的块有效地形成了一个子事务，可以在不影响外部事务的情况下进行回滚

总之，使用 PL/pgSQL 可以提高性能。多少？这是依赖。请记住，在使用 PL/pgSQL 之后，您需要管理多个代码库，并且有时很难调试。

于 2017-04-26T04:40:49.653 回答