Bagaimana cara menambahkan kolom jika tidak ada di PostgreSQL?

Pertanyaannya sederhana. Bagaimana cara menambahkan kolom xke tabel y, tetapi hanya ketika xkolom tidak ada? Saya hanya menemukan solusi di sini cara memeriksa apakah kolom ada.

SELECT column_name 
FROM information_schema.columns 
WHERE table_name='x' and column_name='y';

Berikut ini versi singkat dan manis menggunakan pernyataan "DO":

DO $$ 
    BEGIN
        BEGIN
            ALTER TABLE <table_name> ADD COLUMN <column_name> <column_type>;
        EXCEPTION
            WHEN duplicate_column THEN RAISE NOTICE 'column <column_name> already exists in <table_name>.';
        END;
    END;
$$

Anda tidak dapat melewatkan ini sebagai parameter, Anda harus melakukan substitusi variabel dalam string di sisi klien, tetapi ini adalah kueri mandiri yang hanya memancarkan pesan jika kolom sudah ada, menambahkan jika tidak dan akan terus gagal pada kesalahan lain (seperti tipe data yang tidak valid).

Saya tidak menyarankan melakukan salah satu dari metode ini jika ini adalah string acak yang berasal dari sumber eksternal. Tidak peduli metode apa yang Anda gunakan (string dinamis sisi-sisi atau sisi-server dieksekusi sebagai kueri), itu akan menjadi resep untuk bencana karena membuka Anda terhadap serangan injeksi SQL.

Dengan Postgres 9.6 ini dapat dilakukan menggunakan opsiif not exists

ALTER TABLE table_name ADD COLUMN IF NOT EXISTS column_name INTEGER;

CREATE OR REPLACE function f_add_col(_tbl regclass, _col  text, _type regtype)
  RETURNS bool AS
$func$
BEGIN
   IF EXISTS (SELECT 1 FROM pg_attribute
              WHERE  attrelid = _tbl
              AND    attname = _col
              AND    NOT attisdropped) THEN
      RETURN FALSE;
   ELSE
      EXECUTE format('ALTER TABLE %s ADD COLUMN %I %s', _tbl, _col, _type);
      RETURN TRUE;
   END IF;
END
$func$  LANGUAGE plpgsql;

Panggilan:

SELECT f_add_col('public.kat', 'pfad1', 'int');

Pengembalian TRUEsukses, kalau tidak FALSE(kolom sudah ada).
Menimbulkan pengecualian untuk tabel atau nama jenis tidak valid.

Mengapa versi lain?

• Ini bisa dilakukan dengan DOpernyataan, tetapi DOpernyataan tidak bisa mengembalikan apa pun. Dan jika itu untuk penggunaan berulang, saya akan membuat fungsi.

• Saya menggunakan jenis pengidentifikasi objek regclass dan regtypeuntuk _tbldan _typeyang) mencegah injeksi SQL dan b) memeriksa validitas keduanya segera (cara termurah yang mungkin). Nama kolom _colmasih harus dibersihkan untuk EXECUTEdengan quote_ident(). Penjelasan lebih lanjut dalam jawaban terkait ini:

  • Nama tabel sebagai parameter fungsi PostgreSQL

• format()membutuhkan Postgres 9.1+. Untuk versi lama digabungkan secara manual:

  EXECUTE 'ALTER TABLE ' || _tbl || ' ADD COLUMN ' || quote_ident(_col) || ' ' || _type;

• Anda dapat memenuhi kualifikasi skema nama tabel Anda, tetapi Anda tidak harus melakukannya.
  Anda dapat mengutip dua kali pengidentifikasi dalam pemanggilan fungsi untuk menjaga kasing dan unta kata-kata yang dilindungi (tapi toh Anda tidak boleh menggunakan semua ini).

• Saya meminta pg_catalogbukan information_schema. Penjelasan detail:

  • Cara memeriksa apakah ada tabel dalam skema yang diberikan

• Blok yang mengandung EXCEPTIONklausa seperti jawaban yang saat ini diterima jauh lebih lambat. Ini umumnya lebih sederhana dan lebih cepat. Dokumentasi:

Tip: Blok yang mengandung EXCEPTIONklausa jauh lebih mahal untuk masuk dan keluar daripada blok tanpa klausa. Karena itu, jangan gunakan EXCEPTIONtanpa perlu.

Permintaan pilih berikut akan kembali true/false, menggunakan EXISTS()fungsi.

EXISTS () :
Argumen EXISTS adalah pernyataan SELECT sewenang-wenang, atau subquery. Subquery dievaluasi untuk menentukan apakah ia mengembalikan baris. Jika mengembalikan setidaknya satu baris, hasil dari EXISTS adalah "true"; jika subquery tidak mengembalikan baris, hasil dari EXISTS adalah "false"

SELECT EXISTS(SELECT  column_name 
                FROM  information_schema.columns 
               WHERE  table_schema = 'public' 
                 AND  table_name = 'x' 
                 AND  column_name = 'y'); 

dan gunakan pernyataan SQL dinamis berikut untuk mengubah tabel Anda

DO
$$
BEGIN
IF NOT EXISTS (SELECT column_name 
                 FROM  information_schema.columns 
                WHERE  table_schema = 'public' 
                  AND  table_name = 'x' 
                  AND  column_name = 'y') THEN
ALTER TABLE x ADD COLUMN y int DEFAULT NULL;
ELSE
RAISE NOTICE 'Already exists';
END IF;
END
$$

Bagi mereka yang menggunakan Postgre 9.5+ (saya percaya sebagian besar dari Anda melakukannya), ada solusi yang cukup sederhana dan bersih

ALTER TABLE if exists <tablename> add if not exists <columnname> <columntype>

fungsi di bawah ini akan memeriksa kolom jika ada mengembalikan pesan yang sesuai lain itu akan menambahkan kolom ke tabel.

create or replace function addcol(schemaname varchar, tablename varchar, colname varchar, coltype varchar)
returns varchar 
language 'plpgsql'
as 
$$
declare 
    col_name varchar ;
begin 
      execute 'select column_name from information_schema.columns  where  table_schema = ' ||
      quote_literal(schemaname)||' and table_name='|| quote_literal(tablename) || '   and    column_name= '|| quote_literal(colname)    
      into   col_name ;   

      raise info  ' the val : % ', col_name;
      if(col_name is null ) then 
          col_name := colname;
          execute 'alter table ' ||schemaname|| '.'|| tablename || ' add column '|| colname || '  ' || coltype; 
      else
           col_name := colname ||' Already exist';
      end if;
return col_name;
end;
$$

Ini pada dasarnya adalah solusi dari sola, tetapi baru dibersihkan sedikit. Cukup berbeda sehingga saya tidak hanya ingin "meningkatkan" solusinya (ditambah, saya kira itu tidak sopan).

Perbedaan utama adalah bahwa ia menggunakan format EXECUTE. Yang menurut saya sedikit lebih bersih, tapi saya percaya berarti Anda harus menggunakan PostgresSQL 9.1 atau yang lebih baru.

Ini telah diuji pada 9.1 dan berfungsi. Catatan: Ini akan memunculkan kesalahan jika skema / table_name / atau data_type tidak valid. Itu bisa "diperbaiki", tetapi mungkin perilaku yang benar dalam banyak kasus.

CREATE OR REPLACE FUNCTION add_column(schema_name TEXT, table_name TEXT, 
column_name TEXT, data_type TEXT)
RETURNS BOOLEAN
AS
$BODY$
DECLARE
  _tmp text;
BEGIN

  EXECUTE format('SELECT COLUMN_NAME FROM information_schema.columns WHERE 
    table_schema=%L
    AND table_name=%L
    AND column_name=%L', schema_name, table_name, column_name)
  INTO _tmp;

  IF _tmp IS NOT NULL THEN
    RAISE NOTICE 'Column % already exists in %.%', column_name, schema_name, table_name;
    RETURN FALSE;
  END IF;

  EXECUTE format('ALTER TABLE %I.%I ADD COLUMN %I %s;', schema_name, table_name, column_name, data_type);

  RAISE NOTICE 'Column % added to %.%', column_name, schema_name, table_name;

  RETURN TRUE;
END;
$BODY$
LANGUAGE 'plpgsql';

pemakaian:

select add_column('public', 'foo', 'bar', 'varchar(30)');

Dapat ditambahkan ke fungsi memanggil skrip migrasi dan drop ketika selesai.

create or replace function patch_column() returns void as
$$
begin
    if exists (
        select * from information_schema.columns
            where table_name='my_table'
            and column_name='missing_col'
     )
    then
        raise notice 'missing_col already exists';
    else
        alter table my_table
            add column missing_col varchar;
    end if;
end;
$$ language plpgsql;

select patch_column();

drop function if exists patch_column();

Dalam kasus saya, untuk alasan pembuatannya, agak sulit bagi skrip migrasi kami untuk memotong skema yang berbeda.

Untuk mengatasi ini, kami menggunakan pengecualian yang hanya menangkap dan mengabaikan kesalahan. Ini juga memiliki efek samping yang bagus karena jauh lebih mudah dilihat.

Namun, berhati-hatilah karena solusi lain memiliki kelebihan sendiri yang mungkin lebih penting daripada solusi ini:

DO $$
BEGIN
  BEGIN
    ALTER TABLE IF EXISTS bobby_tables RENAME COLUMN "dckx" TO "xkcd";
  EXCEPTION
    WHEN undefined_column THEN RAISE NOTICE 'Column was already renamed';
  END;
END $$;

Anda bisa melakukannya dengan cara berikut.

ALTER TABLE tableName drop column if exists columnName; 
ALTER TABLE tableName ADD COLUMN columnName character varying(8);

Jadi itu akan menjatuhkan kolom jika sudah ada. Dan kemudian tambahkan kolom ke tabel tertentu.

Cukup periksa apakah kueri mengembalikan nama_kolom.

Jika tidak, jalankan sesuatu seperti ini:

ALTER TABLE x ADD COLUMN y int;

Di mana Anda meletakkan sesuatu yang berguna untuk 'x' dan 'y' dan tentu saja tipe data yang cocok di mana saya menggunakan int.