我如何在PostgreSQL 8.3查询中声明一个变量?

在MS SQL Server我可以这样做:

DECLARE @myvar INT
SET @myvar = 5

SELECT *
FROM somewhere
WHERE something = @myvar

我如何在PostgreSQL做同样的事情?根据文档,变量被简单地声明为“name type;”,但这给了我一个语法错误:

myvar INTEGER;

谁能给我一个正确语法的例子吗?


当前回答

你也可以在PLPGSQL中尝试:

DO $$
DECLARE myvar integer;
BEGIN
    SELECT 5 INTO myvar;

    DROP TABLE IF EXISTS tmp_table;
    CREATE TABLE tmp_table AS
    SELECT * FROM yourtable WHERE   id = myvar;
END $$;

SELECT * FROM tmp_table;

以上要求Postgres 9.0或更高版本。

其他回答

您可以求助于工具的特殊功能。比如DBeaver自己的专有语法:

@set name = 'me'
SELECT :name;
SELECT ${name};

DELETE FROM book b
WHERE b.author_id IN (SELECT a.id FROM author AS a WHERE a.name = :name);

在DBeaver中,你可以在查询中使用参数,就像在代码中一样,所以这是可以工作的:

SELECT *
FROM somewhere
WHERE something = :myvar

当你运行查询时,DBeaver会问你:myvar的值并运行查询。

正如您将从其他答案中了解到的那样,PostgreSQL在直接SQL中没有这种机制,尽管您现在可以使用匿名块。然而,你可以用公共表表达式(CTE)做类似的事情:

WITH vars AS (
    SELECT 5 AS myvar
)
SELECT *
FROM somewhere,vars
WHERE something = vars.myvar;

当然,你可以有任意多的变量,它们也可以被推导出来。例如:

WITH vars AS (
    SELECT
        '1980-01-01'::date AS start,
        '1999-12-31'::date AS end,
        (SELECT avg(height) FROM customers) AS avg_height
)
SELECT *
FROM customers,vars
WHERE (dob BETWEEN vars.start AND vars.end) AND height<vars.avg_height;

流程如下:

使用不带表的SELECT生成单行cte(在Oracle中需要包含FROM DUAL)。 交叉连接cte与另一个表。虽然有CROSS JOIN语法,但旧的逗号语法可读性稍好一些。 注意,我对日期进行了强制转换,以避免SELECT子句中可能出现的问题。我使用了PostgreSQL更短的语法,但您可以使用更正式的CAST('1980-01-01' AS日期)来实现跨方言兼容性。

通常,您希望避免交叉连接,但由于您只是交叉连接单行,因此这只会使用变量数据扩大表。

在许多情况下,不需要包含变量。如果名称与其他表中的名称不冲突,则添加前缀。我把它写在这里是为了说明这一点。

另外,你可以继续添加更多的cte。

这也适用于所有当前版本的MSSQL和MySQL,它们支持变量,以及不支持变量的SQLite,以及支持或不支持变量的Oracle。

PostgreSQL中没有这样的特性。您只能在pl/PgSQL(或其他pl/*)中这样做,但不能在纯SQL中这样做。

一个例外是WITH()查询,它可以作为一个变量,甚至是变量的元组。它允许您返回临时值的表。

WITH master_user AS (
    SELECT
      login,
      registration_date
    FROM users
    WHERE ...
)

SELECT *
FROM users
WHERE master_login = (SELECT login
                      FROM master_user)
      AND (SELECT registration_date
           FROM master_user) > ...;

下面是在postges终端中使用普通变量的代码段。我用过几次。但得想个更好的办法。这里我正在使用字符串变量。使用整型变量时,不需要三引号。三引号在查询时变成单引号;否则就会出现语法错误。在处理字符串变量时,可能有一种方法可以消除对三引号的需求。如果你找到了改进的方法,请及时更新。

\set strainname '''B.1.1.7'''

select *
from covid19strain
where name = :strainname ;