ChatGPT解决这个技术问题 Extra ChatGPT

How to declare a variable in a PostgreSQL query

How do I declare a variable for use in a PostgreSQL 8.3 query?

In MS SQL Server I can do this:

DECLARE @myvar INT
SET @myvar = 5

SELECT *
FROM somewhere
WHERE something = @myvar

How do I do the same in PostgreSQL? According to the documentation variables are declared simply as "name type;", but this gives me a syntax error:

myvar INTEGER;

Could someone give me an example of the correct syntax?

It can be done in just PostgreSQL. See the answer to this related question: stackoverflow.com/questions/766657/…
This related answer has better answers: stackoverflow.com/questions/13316773/…

a
a_horse_with_no_name

I accomplished the same goal by using a WITH clause, it's nowhere near as elegant but can do the same thing. Though for this example it's really overkill. I also don't particularly recommend this.

WITH myconstants (var1, var2) as (
   values (5, 'foo')
)
SELECT *
FROM somewhere, myconstants
WHERE something = var1
   OR something_else = var2;

This works great for most instances where you would want variables. However, if you wanted to use a variable for LIMIT (which can't contain variables), then you'd want to use \set as suggested in Shahriar Aghajani's answer.
This is ideal for when I have a migration script where I want to import some relational data. Obviously I won't know the sequence id the relational data is given.
I just tried this approach, and found a perhaps better way: JOIN myconstants ON true and then there is no need to do the sub-select.
This only works within a single query, you can't share a WITH CTE across queries in a transaction.
Old question, but here here’s a variation: WITH constants AS (SELECT 5 AS var) SELECT * FROM somewhere CROSS JOIN constants WHERE someting=var;. The CROSS JOIN, being a with a single-row table expression, virtually duplicates the data for all of the rows in the real table, and simplifies the expression.
J
J.Wincewicz

There is no such feature in PostgreSQL. You can do it only in pl/PgSQL (or other pl/*), but not in plain SQL.

An exception is WITH () query which can work as a variable, or even tuple of variables. It allows you to return a table of temporary values.

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) > ...;

I tried this method of CTEs being used as vriables. But than i quickly ran into a problem where different data modifying queries in CTEs are not guaranteed to see each other's effects. I had to use multiple CTEs as i needed to use that variable in multiple queries.
e
elixenide

You could also try this in 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;

The above requires Postgres 9.0 or later.


The DO statement was added in PostgreSQL 9.0 and does not work in 8.3.
Use CREATE TEMPORARY TABLE or CREATE TEMP TABLE, not CREATE TABLE. But otherwise fine.
R
Rob Bednark

Dynamic Config Settings

you can "abuse" dynamic config settings for this:

-- choose some prefix that is unlikely to be used by postgres
set session my.vars.id = '1';

select *
from person 
where id = current_setting('my.vars.id')::int;

Config settings are always varchar values, so you need to cast them to the correct data type when using them. This works with any SQL client whereas \set only works in psql

The above requires Postgres 9.2 or later.

For previous versions, the variable had to be declared in postgresql.conf prior to being used, so it limited its usability somewhat. Actually not the variable completely, but the config "class" which is essentially the prefix. But once the prefix was defined, any variable could be used without changing postgresql.conf


@BrijanElwadhi: yes that's transactional.
As a side note: some words are reserved, for example changing set session my.vars.id = '1'; to set session my.user.id = '1'; will yield ERROR: syntax error at or near "user"
@BrijanElwadhi: To make variable transaction specific you must use: SET LOCAL .... The session variable will be in effect as long as you connection is. The local is scoped to transaction.
@dominik You can get around that limitation with quotes, eg., set session "my.user.id" = '1'; The current_setting('my.user.id') call works as expected.
But it seems to be impossible to deal with datetime values in session variable. Something like SET SESSION "vars.tomorrow" = CURRENT_DATE + '1 DAY'::interval; does not work, even with casting to text. In my opinion a great restriction.
K
Kemin Zhou

It depends on your client.

However, if you're using the psql client, then you can use the following:

my_db=> \set myvar 5
my_db=> SELECT :myvar  + 1 AS my_var_plus_1;
 my_var_plus_1 
---------------
             6

If you are using text variables you need to quote.

\set myvar 'sometextvalue'
select * from sometable where name = :'myvar';

\set must be lowercase
db=# \set profile_id 102 db=# :profile_id; ERROR: syntax error at or near "102" LINE 1: 102; ^
@AlxVallejo you have to use it in statement and psql console. db=> \set someid 8292 db=> SELECT * FROM sometable WHERE id = :someid;
J
Jorge Luis

This solution is based on the one proposed by fei0x but it has the advantages that there is no need to join the value list of constants in the query and constants can be easily listed at the start of the query. It also works in recursive queries.

Basically, every constant is a single-value table declared in a WITH clause which can then be called anywhere in the remaining part of the query.

Basic example with two constants:

WITH
    constant_1_str AS (VALUES ('Hello World')),
    constant_2_int AS (VALUES (100))
SELECT *
FROM some_table
WHERE table_column = (table constant_1_str)
LIMIT (table constant_2_int)

Alternatively you can use SELECT * FROM constant_name instead of TABLE constant_name which might not be valid for other query languages different to postgresql.


Very neat, I'll be using this often. Just curious - what does the TABLE keyword do in this context? I'm having no luck searching for it since it's such a generic term.
@user323094 it's same as 'select * from XX'
it works only once. if you write select query to use same value twice, it gives error saying "SQL Error [42P01]: ERROR: relation "constant_1_str" does not exist Position: 20"
@SatishPatro Yes that's the only downside to the CTE approach - it only exists for the first query that follows the creating of the CTE. This example is probably the nicest version of the CTE approach to variables though, IMO
E
Evan Carroll

Using a Temp Table outside of pl/PgSQL

Outside of using pl/pgsql or other pl/* language as suggested, this is the only other possibility I could think of.

begin;
select 5::int as var into temp table myvar;
select *
  from somewhere s, myvar v
 where s.something = v.var;
commit;

C
Community

I want to propose an improvement to @DarioBarrionuevo's answer, to make it simpler leveraging temporary tables.

DO $$
    DECLARE myvar integer = 5;
BEGIN
    CREATE TEMP TABLE tmp_table ON COMMIT DROP AS
        -- put here your query with variables:
        SELECT * 
        FROM yourtable
        WHERE id = myvar;
END $$;

SELECT * FROM tmp_table;

nice solution for solving DO block can't return data set!
On PostgreSQL 11.0, such a query returns 1 (presumably the row count) rather than the contents of tmp_table.
M
Michael.Medvedskiy

True, there is no vivid and unambiguous way to declare a single-value variable, what you can do is

with myVar as (select "any value really")

then, to get access to the value stored in this construction, you do

(select * from myVar)

for example

with var as (select 123)    
... where id = (select * from var)

I am able to use only once, 2nd time I am trying to use it, it's showing "SQL Error [42P01]: ERROR: relation "varName" does not exist Position: 143"
g
gavenkoa

You may resort to tool special features. Like for DBeaver own proprietary syntax:

@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);

This is closer to usable: i'm going to look into whether DBeaver supports lists and looping: i need to apply the same sql to multiple schemas and the list would be of the schemas to apply them to.
M
Martin Zinovsky

Here is an example using PREPARE statements. You still can't use ?, but you can use $n notation:

PREPARE foo(integer) AS
    SELECT  *
    FROM    somewhere
    WHERE   something = $1;
EXECUTE foo(5);
DEALLOCATE foo;

T
The Coder

In DBeaver you can use parameters in queries just like you can from code, so this will work:

SELECT *
FROM somewhere
WHERE something = :myvar

When you run the query DBeaver will ask you for the value for :myvar and run the query.


M
Manngo

As you will have gathered from the other answers, PostgreSQL doesn’t have this mechanism in straight SQL, though you can now use an anonymous block. However, you can do something similar with a Common Table Expression (CTE):

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

You can, of cause have as many variables as you like, and they can also be derived. For example:

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;

The process is:

Generate a one-row cte using SELECT without a table (in Oracle you will need to incllude FROM DUAL).

CROSS JOIN the cte with the other table. Although there is a CROSS JOIN syntax, the older comma syntax is slightly more readable.

Note that I have cast the dates to avoid possible issues in the SELECT clause. I used PostgreSQL’s shorter syntax, but you could have used the more formal CAST('1980-01-01' AS date) for cross-dialect compatibility.

Normally, you want to avoid cross joins, but since you’re only cross joining a single row, this has the effect of simply widening the table with the variable data.

In many cases, you don’t need to include the vars. prefix if the names don’t clash with the names in the other table. I include it here to make the point clear.

Also, you can go on to add more CTEs.

This also works in all current versions of MSSQL and MySQL, which do support variables, as well as SQLite which doesn’t, and Oracle which sort of does and sort of doesn’t.


Downvoted as this only works for a single select block. The reason we use variables is so that it effects multiple lines of code
@user1034912 I did say similar. The reason we use variables is also to set arbitrary values or to pre-calculate values to be used later in the query, and they might well be used multiple times in the query. You will notice that OP’s sample could very easily have been done with a CTE.
K
Kemin Zhou

Here is a code segment using plain variable in postges terminal. I have used it a few times. But need to figure a better way. Here I am working with string variable. Working with integer variable, you don't need the triple quote. Triple quote becomes single quote at query time; otherwise you got syntax error. There might be a way to eliminate the need of triple quote when working with string variables. Please update if you find a way to improve.

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

select *
from covid19strain
where name = :strainname ;