重复的postgresql架构,包括序列

问题描述 投票:6回答:2

我的数据库布局需要为每个新客户创建新架构。目前我使用我在网上找到的内部功能并进行了一些修改。

CREATE FUNCTION copy_schema(
    source_schema character varying, 
    target_schema character varying, 
    copy_data boolean)
RETURNS integer AS
$BODY$
DECLARE
    t_ex integer := 0;
    s_ex integer := 0;
    src_table character varying;
    trg_table character varying;
BEGIN
    if (select 1 from pg_namespace where nspname = source_schema) THEN
        -- we have defined target schema
        s_ex := 1;
    END IF;

    IF (s_ex = 0) THEN
        -- no source schema exist
        RETURN 0;
    END IF;

    if (select 1 from pg_namespace where nspname = target_schema) THEN
        -- we have defined target schema need to sync all table layout
        t_ex := 1;
    ELSE
        EXECUTE 'CREATE SCHEMA '||target_schema||' AUTHORIZATION user';
    END IF;

    FOR src_table IN 
       SELECT table_name 
       FROM information_schema.TABLES 
       WHERE table_schema = source_schema
    LOOP
        trg_table := target_schema||'.'||src_table;
        EXECUTE 
            'CREATE TABLE ' || trg_table || ' (LIKE ' || source_schema || '.' || src_table || ' INCLUDING ALL)';
        IF (copy_data = true) THEN
            EXECUTE 'INSERT INTO ' || trg_table || '(SELECT * FROM ' || source_schema || '.' || src_table || ')';
        END IF;
    END LOOP;

    return t_ex;
END;
$BODY$
LANGUAGE plpgsql VOLATILE
COST 100;

此脚本的问题是新架构中的表继续使用源架构的序列。有没有办法使用sql语句(或其他可靠的方法)来获取新创建的表的序列的新副本(或者甚至是另一种可靠的方法来复制整个模式)?

sql postgresql database-schema multi-tenant
2个回答
5
投票

The root of the problem

与旧序列的连接来自所涉及列的普通默认值。我引用the manual here

仅当指定了INCLUDING DEFAULTS时,才会复制复制的列定义的默认表达式。默认行为是排除默认表达式,导致新表中复制的列具有null默认值。

由于您使用创建新表

INCLUDING ALL

和:

INCLUDING ALLINCLUDING DEFAULTS INCLUDING CONSTRAINTS INCLUDING INDEXES INCLUDING STORAGE INCLUDING COMMENTS的缩写形式。

..你得到相同的默认值。您可以在创建新表后排除默认值或显式更改默认值,包括nextval()。我认为没有任何中间立场。


Simpler with dump / hack the dump / restore

甚至是另一种可靠的复制整个架构的方法

您可以使用pg_dump转储模式的架构(相同的单词,不同的含义):

pg_dump $DB -p $PORT -n $SCHEMA -sf /var/lib/postgresql/your_name.pgsql

破解转储(意思是:在其上使用文本编辑器或编写脚本):在转储顶部交换模式名称,在SET search_path中交换所有其他实例,并作为序列的模式限定,可能更多。如果您为架构选择了一个唯一的名称,那么使用您最喜欢的工具(sedvim或......)进行一次全局搜索和替换就可以完成这项工作。

然后使用psql对相同或任何其他数据库运行SQL脚本:

psql $DB -p $PORT -f /var/lib/postgresql/your_name.pgsql > /dev/null

与我最初发布的内容相反,串行列仍然在转储中被拆分(至少在PostgreSQL 9.1.5中)。 SQL脚本单独创建序列,使用以下命令将它们附加到串行列:

ALTER SEQUENCE seq OWNED BY tbl.col;

并分别设置默认值。

暂且不说:当满足所有要求时,当前版本的pgAdmin在DDL脚本中反向工程serial列。


5
投票

所以经过一番思考后,我继续更新我的第一篇文章中提到的sql函数,所以现在它看起来像这样:

CREATE FUNCTION copy_schema(
    source_schema character varying, 
    target_schema character varying, 
    copy_data boolean)
RETURNS integer AS
$BODY$
DECLARE
    t_ex integer := 0;
    s_ex integer := 0;
    src_table character varying;
    trg_table character varying;
BEGIN
    if (select 1 from pg_namespace where nspname = source_schema) THEN
        -- we have defined target schema
        s_ex := 1;
    END IF;

    IF (s_ex = 0) THEN
        -- no source schema exist
        RETURN 0;
    END IF;

    if (select 1 from pg_namespace where nspname = target_schema) THEN
        -- we have defined target schema need to sync all table layout
        t_ex := 1;
    ELSE
        EXECUTE 'CREATE SCHEMA '||target_schema||' AUTHORIZATION user';
    END IF;

    FOR src_table IN 
        SELECT table_name 
        FROM information_schema.TABLES 
        WHERE table_schema = source_schema
    LOOP
        trg_table := target_schema||'.'||src_table;
        EXECUTE 'CREATE TABLE ' || trg_table || ' (LIKE ' || source_schema || '.' || src_table || ' INCLUDING ALL)';
        EXECUTE 'CREATE SEQUENCE ' || trg_table || '_id_seq OWNED BY '||trg_table || '.id';
        EXECUTE 'ALTER TABLE ' || trg_table || ' ALTER COLUMN id SET DEFAULT nextval('''|| trg_table || '_id_seq''::regclass)';
        IF (copy_data = true) THEN
            EXECUTE 'INSERT INTO ' || trg_table || '(SELECT * FROM ' || source_schema || '.' || src_table || ')';
        END IF;
    END LOOP;
    return t_ex;
END;
$BODY$
LANGUAGE plpgsql VOLATILE
COST 100;

对于每个人来说,这不是一个通用的解决方案,但是因为我在架构中的所有表都有名为id的串行字段,所以它适合我。

@ erwin-brandstetter建议使用dump / hack转储文件/恢复转储文件再次返回的版本通常在论坛上看作是要走的路。

在专用服务器的情况下,它可以工作,在共享主机(或需要较少依赖外部脚本)的情况下,内部功能的方式似乎更好。

© www.soinside.com 2019 - 2024. All rights reserved.