如何将参数传递给用PL / pgSQL编写的查询?

时间:2017-04-25 10:21:38

标签: postgresql stored-procedures go plpgsql pq

我想知道是否可以将参数传递给用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)
}

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

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

问题是,如果我没错,调用tx.Querytx.Exec会使两次访问PostgreSQL服务器。我想要实现的是合并sql1sql2,将它们包含在BEGINEND中,并在一次旅行中执行它们。我的问题是:

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

1 个答案:

答案 0 :(得分:0)

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

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

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

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

关于事务和PL / pgSQL的下一个问题。是的,使用PL / pgSQL可以减少网络流量。 Several advantages服务器端语言(PL / pgSQL)是:

  1. 消除客户端 - 服务器往返
  2. 无需将中间结果传输到客户端,只传输最终结果。
  3. 避免多次解析查询(向服务器发送查询 - >服务器解析查询 - >执行数据库操作 - >将结果返回给客户端等等。)
  4. 处理数据库(大数据)时的规则是You need to avoid to move your data around,PL / pgSQL符合此规则。但是,在某些情况下,您不能(需要避免)使用PL / pgSQL,例如数据库管理员/服务器所有者不允许服务器端编程(安全/性能原因等)。

    the manual中明确说明functiontransaction之间的关系:

      

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

    总之,使用PL / pgSQL可以提高性能。多少?它依赖。请记住,使用PL / pgSQL后,您需要管理多个代码库,有时难以调试。