Bagaimana Anda menggunakan variabel dalam skrip PostgreSQL sederhana?

Misalnya, di MS-SQL, Anda dapat membuka jendela kueri dan menjalankan yang berikut ini:

DECLARE @List AS VARCHAR(8)

SELECT @List = 'foobar'

SELECT *
FROM   dbo.PubLists
WHERE  Name = @List

Bagaimana ini dilakukan di PostgreSQL? Bisakah itu dilakukan?

Jawaban lengkapnya ada di dokumentasi resmi PostgreSQL .

Anda dapat menggunakan fitur blok kode anonim PG9.0 baru ( http://www.postgresql.org/docs/9.1/static/sql-do.html )

DO $$
DECLARE v_List TEXT;
BEGIN
  v_List := 'foobar' ;
  SELECT *
  FROM   dbo.PubLists
  WHERE  Name = v_List;
  -- ...
END $$;

Anda juga bisa mendapatkan id sisipan terakhir :

DO $$
DECLARE lastid bigint;
BEGIN
  INSERT INTO test (name) VALUES ('Test Name') 
  RETURNING id INTO lastid;

  SELECT * FROM test WHERE id = lastid;
END $$;

DO $$
DECLARE  
   a integer := 10;  
   b integer := 20;  
   c integer;  
BEGIN  
   c := a + b;
    RAISE NOTICE'Value of c: %', c;
END $$;

Kamu bisa memakai:

\set list '''foobar'''
SELECT * FROM dbo.PubLists WHERE name = :list;

Itu akan berhasil

Berikut adalah contoh penggunaan variabel di plpgsql:

create table test (id int);
insert into test values (1);
insert into test values (2);
insert into test values (3);

create function test_fn() returns int as $$
    declare val int := 2;
    begin
        return (SELECT id FROM test WHERE id = val);
    end;
$$ LANGUAGE plpgsql;

SELECT * FROM test_fn();
 test_fn 
---------
       2

Silahkan lihat di docs plpgsql untuk informasi lebih lanjut.

Saya telah menemukan beberapa dokumen lain yang mereka gunakan \setuntuk mendeklarasikan variabel scripting tetapi nilainya tampaknya seperti nilai konstan dan saya menemukan cara yang dapat bertindak seperti variabel bukan variabel konstan.

Ex:

\set Comm 150

select sal, sal+:Comm from emp

Berikut saladalah nilai yang ada di tabel 'emp' dan commmerupakan nilai konstanta.

Saya harus melakukan sesuatu seperti ini

CREATE OR REPLACE FUNCTION MYFUNC()
RETURNS VOID AS $$
DO
$do$
BEGIN
DECLARE
 myvar int;
 ...
END
$do$
$$ LANGUAGE SQL;

Postgresql tidak memiliki variabel kosong, Anda dapat menggunakan tabel sementara. variabel hanya tersedia dalam blok kode atau sebagai fitur antarmuka pengguna.

Jika Anda membutuhkan variabel kosong, Anda dapat menggunakan tabel sementara:

CREATE TEMP TABLE list AS VALUES ('foobar');

SELECT dbo.PubLists.*
FROM   dbo.PubLists,list
WHERE  Name = list.column1;

Berdasarkan jawaban @ nad2000 dan jawaban @ Pavel di sini , di sinilah saya berakhir untuk skrip migrasi Jalur Terbang saya. Penanganan skenario di mana skema database diubah secara manual.

DO $$
BEGIN
    IF NOT EXISTS(
        SELECT TRUE FROM pg_attribute 
        WHERE attrelid = (
            SELECT c.oid
            FROM pg_class c
            JOIN pg_namespace n ON n.oid = c.relnamespace
            WHERE 
                n.nspname = CURRENT_SCHEMA() 
                AND c.relname = 'device_ip_lookups'
            )
        AND attname = 'active_date'
        AND NOT attisdropped
        AND attnum > 0
        )
    THEN
        RAISE NOTICE 'ADDING COLUMN';        
        ALTER TABLE device_ip_lookups
            ADD COLUMN active_date TIMESTAMP;
    ELSE
        RAISE NOTICE 'SKIPPING, COLUMN ALREADY EXISTS';
    END IF;
END $$;

Untuk klien CLI resmi "psql" lihat di sini . Dan "pgAdmin3" 1.10 (masih dalam versi beta) memiliki " pgScript ".

Untuk menggunakan variabel, misalnya mengubah tabel:

DO $$ 
DECLARE name_pk VARCHAR(200);
BEGIN
select constraint_name
from information_schema.table_constraints
where table_schema = 'schema_name'
      and table_name = 'table_name'
      and constraint_type = 'PRIMARY KEY' INTO name_pk;
IF (name_pk := '') THEN
EXECUTE 'ALTER TABLE schema_name.table_name DROP CONSTRAINT ' || name_pk;