Tạo VAI TRÒ PostgreSQL (người dùng) nếu nó không tồn tại

Làm cách nào để viết tập lệnh SQL để tạo VAI TRÒ trong PostgreSQL 9.1, nhưng không phát sinh lỗi nếu nó đã tồn tại?

Tập lệnh hiện tại chỉ có:

CREATE ROLE my_user LOGIN PASSWORD 'my_password';

Điều này không thành công nếu người dùng đã tồn tại. Tôi muốn một cái gì đó như:

IF NOT EXISTS (SELECT * FROM pg_user WHERE username = 'my_user')
BEGIN
    CREATE ROLE my_user LOGIN PASSWORD 'my_password';
END;

... nhưng điều đó không hoạt động - IFdường như không được hỗ trợ trong SQL thuần túy.

Tôi có một tệp lô tạo cơ sở dữ liệu PostgreSQL 9.1, vai trò và một số thứ khác. Nó gọi psql.exe, truyền vào tên của một tập lệnh SQL để chạy. Cho đến nay tất cả các tập lệnh này đều là SQL thuần túy và tôi muốn tránh PL / pgSQL và những thứ tương tự, nếu có thể.

Đơn giản hóa theo cách tương tự như những gì bạn đã nghĩ:

DO
$do$
BEGIN
   IF NOT EXISTS (
      SELECT FROM pg_catalog.pg_roles  -- SELECT list can be empty for this
      WHERE  rolname = 'my_user') THEN

      CREATE ROLE my_user LOGIN PASSWORD 'my_password';
   END IF;
END
$do$;

(Dựa trên câu trả lời của @a_horse_with_no_name và được cải thiện với nhận xét của @ Gregory .)

Không giống như, chẳng hạn, với CREATE TABLEkhông có IF NOT EXISTSmệnh đề cho CREATE ROLE(tối đa ít nhất là pg 12). Và bạn không thể thực thi các câu lệnh DDL động trong SQL thuần túy.

Yêu cầu của bạn để "tránh PL / pgSQL" là không thể ngoại trừ bằng cách sử dụng PL khác. Câu DOlệnh sử dụng plpgsql làm ngôn ngữ thủ tục mặc định. Cú pháp cho phép bỏ qua khai báo rõ ràng:

DO [ LANGUAGE lang_name ] code
... Tên của ngôn ngữ thủ tục mà mã được viết. Nếu bỏ qua, giá trị mặc định là .
lang_name
plpgsql

Câu trả lời được chấp nhận gặp phải điều kiện chạy đua nếu hai tập lệnh như vậy được thực thi đồng thời trên cùng một cụm Postgres (máy chủ DB), điều này thường xảy ra trong môi trường tích hợp liên tục .

Nói chung sẽ an toàn hơn nếu cố gắng tạo vai trò và giải quyết các vấn đề một cách khéo léo khi tạo vai trò đó:

DO $$
BEGIN
  CREATE ROLE my_role WITH NOLOGIN;
  EXCEPTION WHEN DUPLICATE_OBJECT THEN
  RAISE NOTICE 'not creating role my_role -- it already exists';
END
$$;

Hoặc nếu vai trò không phải là chủ sở hữu của bất kỳ đối tượng db nào thì người ta có thể sử dụng:

DROP ROLE IF EXISTS my_user;
CREATE ROLE my_user LOGIN PASSWORD 'my_password';

Nhưng chỉ khi thả người dùng này sẽ không gây hại gì.

Bash thay thế (cho Bash scripting ):

psql -h localhost -U postgres -tc \
"SELECT 1 FROM pg_user WHERE usename = 'my_user'" \
| grep -q 1 \
|| psql -h localhost -U postgres \
-c "CREATE ROLE my_user LOGIN PASSWORD 'my_password';"

(không phải là câu trả lời cho câu hỏi! nó chỉ dành cho những người có thể hữu ích)

Đây là một giải pháp chung sử dụng plpgsql:

CREATE OR REPLACE FUNCTION create_role_if_not_exists(rolename NAME) RETURNS TEXT AS
$$
BEGIN
    IF NOT EXISTS (SELECT * FROM pg_roles WHERE rolname = rolename) THEN
        EXECUTE format('CREATE ROLE %I', rolename);
        RETURN 'CREATE ROLE';
    ELSE
        RETURN format('ROLE ''%I'' ALREADY EXISTS', rolename);
    END IF;
END;
$$
LANGUAGE plpgsql;

Sử dụng:

posgres=# SELECT create_role_if_not_exists('ri');
 create_role_if_not_exists 
---------------------------
 CREATE ROLE
(1 row)
posgres=# SELECT create_role_if_not_exists('ri');
 create_role_if_not_exists 
---------------------------
 ROLE 'ri' ALREADY EXISTS
(1 row)

Một số câu trả lời được đề xuất để sử dụng mẫu: kiểm tra nếu vai trò không tồn tại và nếu không thì hãy ra CREATE ROLElệnh. Điều này có một bất lợi: điều kiện chủng tộc. Nếu ai đó khác tạo một vai trò mới giữa kiểm tra và phát hành CREATE ROLElệnh thì CREATE ROLErõ ràng là không thành công với lỗi nghiêm trọng.

Để giải quyết vấn đề trên, nhiều câu trả lời khác đã đề cập đến việc sử dụng PL/pgSQL, phát hành CREATE ROLEvô điều kiện và sau đó bắt các ngoại lệ từ cuộc gọi đó. Chỉ có một vấn đề với các giải pháp này. Họ âm thầm loại bỏ bất kỳ lỗi nào, bao gồm cả những lỗi không được tạo ra bởi thực tế rằng vai trò đã tồn tại. CREATE ROLEcũng có thể tạo ra các lỗi khác và mô phỏng IF NOT EXISTSchỉ nên tắt lỗi khi vai trò đã tồn tại.

CREATE ROLEném duplicate_objectlỗi khi vai trò đã tồn tại. Và trình xử lý ngoại lệ chỉ nên bắt một lỗi này. Như các câu trả lời khác đã đề cập, bạn nên chuyển lỗi nghiêm trọng thành thông báo đơn giản. Các IF NOT EXISTSlệnh PostgreSQL khác thêm , skippingvào thông điệp của chúng, vì vậy để nhất quán, tôi cũng sẽ thêm nó vào đây.

Đây là mã SQL đầy đủ để mô phỏng CREATE ROLE IF NOT EXISTSvới sự lan truyền ngoại lệ và sqlstate chính xác:

DO $$
BEGIN
CREATE ROLE test;
EXCEPTION WHEN duplicate_object THEN RAISE NOTICE '%, skipping', SQLERRM USING ERRCODE = SQLSTATE;
END
$$;

Kiểm tra đầu ra (được gọi hai lần qua DO và sau đó trực tiếp):

$ sudo -u postgres psql
psql (9.6.12)
Type "help" for help.

postgres=# \set ON_ERROR_STOP on
postgres=# \set VERBOSITY verbose
postgres=# 
postgres=# DO $$
postgres$# BEGIN
postgres$# CREATE ROLE test;
postgres$# EXCEPTION WHEN duplicate_object THEN RAISE NOTICE '%, skipping', SQLERRM USING ERRCODE = SQLSTATE;
postgres$# END
postgres$# $$;
DO
postgres=# 
postgres=# DO $$
postgres$# BEGIN
postgres$# CREATE ROLE test;
postgres$# EXCEPTION WHEN duplicate_object THEN RAISE NOTICE '%, skipping', SQLERRM USING ERRCODE = SQLSTATE;
postgres$# END
postgres$# $$;
NOTICE:  42710: role "test" already exists, skipping
LOCATION:  exec_stmt_raise, pl_exec.c:3165
DO
postgres=# 
postgres=# CREATE ROLE test;
ERROR:  42710: role "test" already exists
LOCATION:  CreateRole, user.c:337

Khi bạn đang sử dụng 9.x, bạn có thể gói nó thành một câu lệnh DO:

do 
$body$
declare 
  num_users integer;
begin
   SELECT count(*) 
     into num_users
   FROM pg_user
   WHERE usename = 'my_user';

   IF num_users = 0 THEN
      CREATE ROLE my_user LOGIN PASSWORD 'my_password';
   END IF;
end
$body$
;

Nhóm của tôi đã gặp phải tình huống có nhiều cơ sở dữ liệu trên một máy chủ, tùy thuộc vào cơ sở dữ liệu bạn đã kết nối, VAI TRÒ được đề cập không được trả lại SELECT * FROM pg_catalog.pg_userbởi @ erwin-brandstetter và @a_horse_with_no_name. Khối điều kiện được thực thi và chúng tôi đã nhấn role "my_user" already exists.

Rất tiếc, chúng tôi không chắc chắn về các điều kiện chính xác, nhưng giải pháp này khắc phục được sự cố:

        DO  
        $body$
        BEGIN
            CREATE ROLE my_user LOGIN PASSWORD 'my_password';
        EXCEPTION WHEN others THEN
            RAISE NOTICE 'my_user role exists, not re-creating';
        END
        $body$

Nó có thể được thực hiện cụ thể hơn để loại trừ các trường hợp ngoại lệ khác.

Bạn có thể làm điều đó trong tệp lô của mình bằng cách phân tích cú pháp đầu ra của:

SELECT * FROM pg_user WHERE usename = 'my_user'

và sau đó chạy psql.exelại một lần nữa nếu vai trò không tồn tại.

Giải pháp tương tự như đối với Mô phỏng TẠO CƠ SỞ DỮ LIỆU NẾU KHÔNG TỒN TẠI cho PostgreSQL?nên hoạt động - gửi một CREATE USER …đến \gexec.

Cách giải quyết từ bên trong psql

SELECT 'CREATE USER my_user'
WHERE NOT EXISTS (SELECT FROM pg_catalog.pg_roles WHERE rolname = 'my_user')\gexec

Cách giải quyết từ shell

echo "SELECT 'CREATE USER my_user' WHERE NOT EXISTS (SELECT FROM pg_catalog.pg_roles WHERE rolname = 'my_user')\gexec" | psql

Xem câu trả lời được chấp nhận ở đó để biết thêm chi tiết.