Створіть PostgreSQL ROLE (користувача), якщо він не існує

Як написати сценарій SQL, щоб створити РОЛЬ у PostgreSQL 9.1, але без підвищення помилки, якщо він вже існує?

Поточний сценарій просто має:

CREATE ROLE my_user LOGIN PASSWORD 'my_password';

Це не вдається, якщо користувач вже існує. Мені б хотілося чогось типу:

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

... але це не працює - IFздається, не підтримується у звичайному SQL.

У мене є пакетний файл, який створює базу даних PostgreSQL 9.1, роль та кілька інших речей. Він викликає psql.exe, передаючи ім'я сценарію SQL для запуску. Поки всі ці сценарії є звичайними SQL, і я хотів би уникати PL / pgSQL і подібного, якщо це можливо.

Спростіть аналогічно тому, що ви мали на увазі:

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$;

(Спираючись на відповідь @a_horse_with_no_name та покращивши коментар @ Грегорі .)

На відміну, наприклад, CREATE TABLEвідсутнє IF NOT EXISTSзастереження CREATE ROLE(до принаймні pg 12). І ви не можете виконувати динамічні оператори DDL у звичайному SQL.

Ваш запит "уникнути PL / pgSQL" неможливий, за винятком використання іншого PL. У DOзаяві використовується plpgsql в якості процедурної мови за умовчанням. Синтаксис дозволяє опустити явне оголошення:

DO [ LANGUAGE lang_name ] code
... Ім'я процедурної мови, на якому записаний код. Якщо цей пункт пропущено, типовим є .
lang_name
plpgsql

Прийнята відповідь страждає від умови перегонів, якщо два такі сценарії виконуються одночасно на одному кластері Postgres (сервер БД), як це часто в середовищах безперервної інтеграції .

Зазвичай безпечніше намагатися створити роль і витончено вирішувати проблеми при її створенні:

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

Або якщо роль не є власником жодних db-об'єктів, ви можете використовувати:

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

Але тільки якщо випадання цього користувача не принесе шкоди.

Альтернатива Bash (для сценаріїв Bash ):

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';"

(не відповідь на запитання! це лише для тих, хто може бути корисним)

Ось загальне рішення з використанням 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;

Використання:

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)

Деякі відповіді пропонують використовувати шаблон: перевірте, чи немає ролі, а якщо ні, то видайте CREATE ROLEкоманду. У цього є один недолік: стан перегонів. Якщо хтось інший створює нову роль між перевіркою та видачею CREATE ROLEкоманди, то, CREATE ROLEочевидно, не вдається з фатальною помилкою.

Щоб вирішити вищезгадану проблему, більше інших відповідей, про які вже згадувалося PL/pgSQL, видаючи CREATE ROLEбеззастережно і потім вибираючи винятки з цього дзвінка. З цими рішеннями існує лише одна проблема. Вони мовчки скидають будь-які помилки, в тому числі й ті, які не породжені фактом, що роль вже існує. CREATE ROLEможе кидати й інші помилки, і моделювання IF NOT EXISTSповинно замовчувати помилку лише тоді, коли роль вже існує.

CREATE ROLEduplicate_objectпомилка кидка, коли роль вже існує. І обробник винятків повинен виявити лише цю одну помилку. Як згадували інші відповіді, це гарна ідея перетворити фатальну помилку в просте повідомлення. Інші IF NOT EXISTSкоманди PostgreSQL додають , skippingу своє повідомлення, тому для послідовності додаю його і тут.

Ось повний код SQL для моделювання CREATE ROLE IF NOT EXISTSз правильним винятком та розповсюдження sqlstate:

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

Тестовий вихід (викликається два рази через DO, а потім безпосередньо):

$ 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

Оскільки ви перебуваєте на 9.x, ви можете перетворити це в оператор 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$
;

Моя команда потрапила в ситуацію з декількома базами даних на одному сервері, залежно від того, до якої бази даних ви підключились, відповідна ROLE не була повернена SELECT * FROM pg_catalog.pg_user, як запропонували @ erwin-marketetter та @a_horse_with_no_name. Умовний блок виконали, і ми вдарили role "my_user" already exists.

На жаль, ми не впевнені в точних умовах, але це рішення вирішує проблему:

        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$

Можливо, це може бути конкретніше, щоб виключити інші винятки.

Ви можете зробити це у своєму пакетному файлі, розібравши вихід:

SELECT * FROM pg_user WHERE usename = 'my_user'

а потім psql.exeще раз запустити, якщо ролі не існує.

Те саме рішення, що і для імітації СТВОРИТИ ДАТАБАЗУ, ЯКЩО НЕ МАЄТЬСЯ для PostgreSQL? повинен працювати - надіслати CREATE USER …на \gexec.

Обхід зсередини psql

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

Обхід із оболонки

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

Дивіться прийняту відповідь там для отримання більш детальної інформації.