我正在尝试在存储过程中测试序列是否已经存在。
IF EXISTS SEQUENCE seq_name
RAISE EXCEPTION 'sequence % already exists!', seq_name
END IF;
我已经尝试了上面片段的几种变体,但没有运气。我一定是给 Google 提供了错误的术语,因为我似乎找不到任何关于该主题的内容。任何帮助表示赞赏!
我正在尝试在存储过程中测试序列是否已经存在。
IF EXISTS SEQUENCE seq_name
RAISE EXCEPTION 'sequence % already exists!', seq_name
END IF;
我已经尝试了上面片段的几种变体,但没有运气。我一定是给 Google 提供了错误的术语,因为我似乎找不到任何关于该主题的内容。任何帮助表示赞赏!
您应该能够查询 pg_class 表以查看 relname 是否存在。
IF EXISTS (SELECT 0 FROM pg_class where relname = '<my sequence name here>' )
THEN
--stuff here
END IF;
如果您确定名称只能对序列有效(即,您确信它不会用于普通表、索引、视图、复合类型、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;
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()
使用模式限定名称失败 - 会模棱两可。
schema 参数有一个默认值,因此您可以在调用中省略它。如果没有给出模式,则函数默认为current_schema
. 根据文档:
current_schema
返回搜索路径中第一个模式的名称(如果搜索路径为空,则返回 null 值)。这是将用于在未指定目标架构的情况下创建的任何表或其他命名对象的架构。
pgclass.relkind
手册中的类型列表。
如何使用信息模式:
SELECT COUNT(*)
FROM information_schema.sequences
WHERE sequence_schema=? AND sequence_name=?
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
此查询可以检查模式内是否存在序列。
我不确定为什么必须检查序列的存在的实际意图。如果目标是在创建序列之前检查序列是否存在,则IF NOT EXISTS
可以使用 PostgreSQL 中的条件:
CREATE SEQUENCE IF NOT EXISTS 'name'
见https://www.postgresql.org/docs/9.5/static/sql-createsequence.html