Проверьте, существует ли последовательность в Postgres (plpgsql) - программирование
Подтвердить что ты не робот

Проверьте, существует ли последовательность в Postgres (plpgsql)

Я пытаюсь проверить в хранимой процедуре, существует ли последовательность.

IF EXISTS SEQUENCE seq_name
    RAISE EXCEPTION 'sequence % already exists!', seq_name
END IF;

Я попробовал несколько вариантов фрагмента выше, не повезло. Я должен давать Google неправильные условия, потому что я не могу найти что-либо по этой теме. Любая помощь приветствуется!

4b9b3361

Ответ 1

Вы должны иметь возможность запросить таблицу pg_class, чтобы увидеть, существует ли relname.

IF EXISTS (SELECT 0 FROM pg_class where relname = '<my sequence name here>' )
THEN
  --stuff here
END IF;

Ответ 2

Ответ от @rfusca работает, если вы уверены, что имя может быть действительным только для последовательности (т.е. вы уверены, что он не будет использоваться для обычной таблицы, индекса, представления, композитного типа, TOAST таблицы или внешней таблицы), и вас не интересуют несколько схем. Другими словами, он работает для большинства распространенных случаев, но это не совсем строго.

Если вы хотите проверить, существует ли последовательность с таким именем в конкретной схеме, это должно работать:

-- Clear the search path so that the regclass of the sequence
-- will be schema-qualified.
SET search_path = '';
-- Do your conditional code.
IF EXISTS (SELECT * FROM pg_class
             WHERE relkind = 'S'
               AND oid::regclass::text = 'public.' || quote_ident(seq_name))
  THEN
    RAISE EXCEPTION 'sequence public.% already exists!', seq_name
END IF;
-- Restore the normal search path.
RESET search_path;

Ответ 3

Обновление: просто тестирование на существование стало проще с to_regclass() в Postgres 9.4

SELECT to_regclass('schema_name.table_name');

Но прочитайте подробности:

Полная функция

Вам нужно проверить любой объект, похожий на таблицу, который будет конфликтовать с именем, а не только последовательностями.

Эта функция создает новую последовательность, если имя доступно и выдает значимые NOTICE/WARNING/EXCEPTION соответственно в других случаях:

CREATE OR REPLACE FUNCTION f_create_seq(_seq text, _schema text = NULL)
  RETURNS void AS
$func$
DECLARE
   _fullname text := format('%I.%I', COALESCE(_schema,current_schema),_seq);
   _relkind "char" := (SELECT c.relkind
                       FROM   pg_namespace n
                       JOIN   pg_class c ON c.relnamespace = n.oid
                       WHERE  n.nspname = COALESCE(_schema, current_schema)
                       AND    c.relname = _seq);
BEGIN
   IF _relkind IS NULL THEN   -- name is free
      EXECUTE 'CREATE SEQUENCE ' || _fullname;
      RAISE NOTICE 'New sequence % created.', _fullname;

   ELSIF _relkind = 'S' THEN  -- 'S' = sequence
      IF has_sequence_privilege(_fullname, 'USAGE') THEN
         RAISE WARNING 'Sequence % already exists.', _fullname;
      ELSE
         RAISE EXCEPTION
           'Sequence % already exists but you have no USAGE privilege.'
         , _fullname;
      END IF;

   ELSE
      RAISE EXCEPTION 'A(n) "%" named % already exists.'
      -- Table-like objects in pg 9.4:
      -- www.postgresql.org/docs/current/static/catalog-pg-class.html
         , CASE _relkind WHEN 'r' THEN 'ordinary table'
                         WHEN 'i' THEN 'index'
                      -- WHEN 'S' THEN 'sequence'  -- impossible here
                         WHEN 'v' THEN 'view'
                         WHEN 'm' THEN 'materialized view'
                         WHEN 'c' THEN 'composite type'
                         WHEN 't' THEN 'TOAST table'
                         WHEN 'f' THEN 'foreign table'
                         ELSE 'unknown object' END
         , _fullname;
   END IF;
END
$func$  LANGUAGE plpgsql;

COMMENT ON FUNCTION f_create_seq(text, text) IS
'Create sequence if name is free.
RAISE NOTICE on successful creation.
RAISE WARNING if it already exists.
RAISE EXCEPTION if it already exists and current user lacks USAGE privilege.
RAISE EXCEPTION if object of a different kind occupies the name.
$1 _seq    .. sequence name 
$2 _schema .. schema name (optional; default is CURRENT_SCHEMA)';

Вызов:

SELECT f_create_seq('myseq', 'myschema');

Или:

SELECT f_create_seq('myseq1');  -- defaults to current schema

Объяснить

  • Также прочитайте комментарий к функции в конце кода.

  • Работает в Postgres 9.1 +. Для более старых версий вам нужно заменить format(), который защищает от SQL-инъекции. Подробности:

  • Два отдельных параметра позволяют последовательности в любой схеме независимо от текущего search_path, а также позволяют quote_ident() выполнять свою работу. quote_ident() не выполняется с именами, закодированными с помощью схемы, будет неоднозначным.

  • Значение параметра схемы указано по умолчанию, поэтому вы можете опустить его из вызова. Если схема не указана, функция по умолчанию имеет значение current_schema. В документации:

    current_schema возвращает имя схемы, которая первая в путь поиска (или нулевое значение, если путь поиска пуст). Это схему, которая будет использоваться для любых таблиц или других именованных объектов, которые создаются без указания целевой схемы.

  • Список типов pgclass.relkind в руководстве.

  • Коды ошибок PostgreSQL.

Ответ 4

Как насчет использования информационной схемы:

SELECT COUNT(*) 
FROM information_schema.sequences 
WHERE sequence_schema=? AND sequence_name=?

Ответ 5

select relname, relnamespace
from pg_class join pg_catalog.pg_namespace n ON n.oid = pg_class.relnamespace
where n.nspname='metastore_1' and relname='updater_state_id_seq';

Результат:

       relname        | relnamespace 
-------------------------------------
 updater_state_id_seq |        32898

Этот запрос может проверить существование последовательности внутри схемы.