如果不存在,创建表

在一个MySQL脚本中,你可以写:

CREATE TABLE IF NOT EXISTS foo ...;

... 其他的…

然后可以多次运行脚本,而无需重新创建表。

在PostgreSQL中如何做到这一点?

372393 次浏览

没有创建表如果不存在…但是你可以为此写一个简单的过程,比如:

CREATE OR REPLACE FUNCTION execute(TEXT) RETURNS VOID AS $$
BEGIN
EXECUTE $1;
END; $$ LANGUAGE plpgsql;




SELECT
execute($$
CREATE TABLE sch.foo
(
i integer
)
$$)
WHERE
NOT exists
(
SELECT *
FROM information_schema.tables
WHERE table_name = 'foo'
AND table_schema = 'sch'
);

这个特性已经Postgres 9.1中实现:

CREATE TABLE IF NOT EXISTS myschema.mytable (i integer);


对于旧版本,这里有一个函数来处理它:

CREATE OR REPLACE FUNCTION create_mytable()
RETURNS void
LANGUAGE plpgsql AS
$func$
BEGIN
IF EXISTS (SELECT FROM pg_catalog.pg_tables
WHERE  schemaname = 'myschema'
AND    tablename  = 'mytable') THEN
RAISE NOTICE 'Table myschema.mytable already exists.';
ELSE
CREATE TABLE myschema.mytable (i integer);
END IF;
END
$func$;

电话:

SELECT create_mytable();        -- call as many times as you want.

笔记

pg_tables中的列schemanametablename区分大小写。如果在CREATE TABLE语句中使用双引号标识符,则需要使用完全相同的拼写。如果没有,则需要使用小写字符串。看到的:

pg_tables只包含实际的。标识符仍然可能被相关对象占用。看到的:

如果角色执行这个函数没有必要的权限来创建表,你可能想要为这个函数使用SECURITY DEFINER,并由另一个具有必要权限的角色使其拥有。这个版本足够安全。

我从现有的答案中创建了一个通用的解决方案,可以重复用于任何表:

CREATE OR REPLACE FUNCTION create_if_not_exists (table_name text, create_stmt text)
RETURNS text AS
$_$
BEGIN


IF EXISTS (
SELECT *
FROM   pg_catalog.pg_tables
WHERE    tablename  = table_name
) THEN
RETURN 'TABLE ' || '''' || table_name || '''' || ' ALREADY EXISTS';
ELSE
EXECUTE create_stmt;
RETURN 'CREATED';
END IF;


END;
$_$ LANGUAGE plpgsql;

用法:

select create_if_not_exists('my_table', 'CREATE TABLE my_table (id integer NOT NULL);');

如果要从查询参数中提取表名,可以进一步简化为只取一个参数。我还省略了模式。

这个解决方案有点类似于Erwin Brandstetter的答案,但只使用sql语言。

并非所有PostgreSQL安装都默认使用plpqsql语言,这意味着你可能必须在创建函数之前调用CREATE LANGUAGE plpgsql,然后必须再次删除该语言,以使数据库保持与之前相同的状态(但只有在数据库一开始就没有plpgsql语言时)。看到复杂性是如何增长的了吗?

如果您在本地运行脚本,添加plpgsql可能不是问题,但是,如果脚本用于在客户中设置模式,则不希望在客户数据库中留下这样的更改。

这个解决方案的灵感来自Andreas Scherbaum的一篇文章

-- Function which creates table
CREATE OR REPLACE FUNCTION create_table () RETURNS TEXT AS $$
CREATE TABLE table_name (
i int
);
SELECT 'extended_recycle_bin created'::TEXT;
$$
LANGUAGE 'sql';


-- Test if table exists, and if not create it
SELECT CASE WHEN (SELECT true::BOOLEAN
FROM   pg_catalog.pg_tables
WHERE  schemaname = 'public'
AND    tablename  = 'table_name'
) THEN (SELECT 'success'::TEXT)
ELSE (SELECT create_table())
END;


-- Drop function
DROP FUNCTION create_table();

试试这个:

CREATE TABLE IF NOT EXISTS app_user (
username varchar(45) NOT NULL,
password varchar(450) NOT NULL,
enabled integer NOT NULL DEFAULT '1',
PRIMARY KEY (username)
)

没有创建表如果不存在…但是你可以为此写一个简单的过程,比如:

CREATE OR REPLACE FUNCTION prc_create_sch_foo_table() RETURNS VOID AS $$
BEGIN


EXECUTE 'CREATE TABLE /* IF NOT EXISTS add for PostgreSQL 9.1+ */ sch.foo (
id serial NOT NULL,
demo_column varchar NOT NULL,
demo_column2 varchar NOT NULL,
CONSTRAINT pk_sch_foo PRIMARY KEY (id));
CREATE INDEX /* IF NOT EXISTS add for PostgreSQL 9.5+ */ idx_sch_foo_demo_column ON sch.foo(demo_column);
CREATE INDEX /* IF NOT EXISTS add for PostgreSQL 9.5+ */ idx_sch_foo_demo_column2 ON sch.foo(demo_column2);'
WHERE NOT EXISTS(SELECT * FROM information_schema.tables
WHERE table_schema = 'sch'
AND table_name = 'foo');


EXCEPTION WHEN null_value_not_allowed THEN
WHEN duplicate_table THEN
WHEN others THEN RAISE EXCEPTION '% %', SQLSTATE, SQLERRM;


END; $$ LANGUAGE plpgsql;