为什么不能使用 DO ... END 进行参数化查询？

Question

以下工作正常：

conn = psycopg.connect(self.conn.params.conn_str)
cur = conn.cursor()
cur.execute("""
    SELECT 2, %s;
    """, (1,),
)

但是在

DO

里面：

cur.execute("""
DO $$
BEGIN
  SELECT 2, %s;
END$$;
""",  (1,),
)

它导致

psycopg.errors.UndefinedParameter: there is no parameter $1
LINE 1: SELECT 2, $1
                  ^
QUERY:  SELECT 2, $1
CONTEXT:  PL/pgSQL function inline_code_block line 3 at SQL statement

这是预期的吗？

Answer 1

是的，因为匿名代码块不接受参数：

代码块被视为没有参数的函数体，返回 void。它被解析并执行一次。

我还没有测试过这个，但这可能是一种解决方法：

DO $$
DECLARE
  my_param integer := %s;  -- Declare a variable to hold the parameter
  result integer;
BEGIN
  SELECT 2, my_param INTO result;  -- Store the result in a PL/pgSQL variable
END$$;