Установка dbname условно из блока PostgreSQL

Question

Я должен подключиться к БД, если таковой имеется, а если ее нет, мы должны создать БД с другим именем и подключиться к ней.Следовательно, нам нужно соединение с БД, чье имя будет меняться в зависимости от вышеуказанного условия.

Мой текущий код выглядит так:

DO $$
BEGIN
   IF EXISTS (SELECT 1 FROM pg_database WHERE datname = 'xdb') THEN
      RAISE NOTICE 'Database already exists';
   ELSE
      RAISE NOTICE 'Database does not exist';
      PERFORM dblink_exec('dbname=' || current_database()||' user=puser'  -- current db
                        , 'CREATE DATABASE "ydb" with owner puser');
   END IF;
END $$;
\connect 'xdb' ---->> This DB should be either xdb or ydb based on the above condition
## Other insert statements

Есть ли способ, которым я могу этого достичь?Мы используем Postgres 11.

Answer 1

Или, возможно:

select case when exists (select 1 from pg_database where datname = 'xdb') 
          then 'xdb'::text
          else 'ydb' 
       end as conndb \gset
\connect :conndb 

Answer 2

Я предполагаю, что это сценарий psql.

Вы можете использовать условную обработку в psql:

\if :reconnect
\connect xdb
\endif

Переменная может быть установлена ​​с помощью

\set reconnect 1

в соответствующей ветке.