在事务中删除然后创建外键约束是否安全？

Question

我有一个引用表 B 的表 A。表 B 需要使用来自外部源的更新数据填充，为了提高效率，我使用 TRUNCATE 后跟一个 COPY。即使应用程序处于活动状态，也会这样做。

为了进一步提高效率，as suggested in the documentation，我想删除然后重新创建外键。

但是我有一些疑问。

如果我删除 FK、COPY 然后重新创建 FK在同一个事务中，我是否可以确保即使在事务期间插入到表 A 中的数据上也会保留约束？我问这个是因为理论上事务是原子的，但在文档中，关于临时删除 FK 说：

在缺少约束时，需要在数据加载速度和错误检查丢失之间进行权衡。

如果在此期间插入了错误的引用，当您尝试重新创建 FK 约束时会发生什么？

Answer 1

您可以使外键约束@​​987654321@（最初延迟）。这样在交易结束时只会检查一次。

ALTER TABLE
  xxx
ADD CONSTRAINT
  xxx_yyy_id_fk FOREIGN KEY (yyy_id)
REFERENCES
  yyy
DEFERRABLE INITIALLY DEFERRED;

在所有情况下，事务在 PostgreSQL 中都是完全原子的（不仅在理论上），包括 DDL 语句（例如 CREATE/DROP 约束），所以即使你删除了外键，然后插入数据，然后创建外键key 并在一个事务中完成所有操作，那么您是安全的 - 如果重新创建外键约束失败，则插入的数据也将被取消。

不过，最好切换到延迟外键，而不是删除然后创建它们。

Answer 2

TRUNCATE 不允许在外键引用的任何表上，除非您使用 TRUNCATE CASCADE，这也会截断引用表。约束的DEFERRABLE 状态不影响这一点。我认为没有办法解决这个问题。您将需要删除约束。

但是，这样做没有违反完整性的风险。 ALTER TABLE ... ADD CONSTRAINT 锁定有问题的表（TRUNCATE 也是如此），因此您的导入过程保证在其事务期间对表具有独占访问权。任何并发插入尝试都将挂起，直到导入提交，当它们被允许继续时，约束将恢复原状。

Answer 3

分析答案：测量新/相同/更新/删除记录的数量。 有四种情况：

• B 表中的键在 b_import 中不存在：删除
• b_import 中的键在旧 B 上不存在：插入
• key在旧B和新B中都有，但内容相同：忽略
• 键相同，但属性值不同：更新

---

        -- some test data for `A`, `B` and `B_import`:
CREATE TABLE b
        ( id INTEGER NOT NULL PRIMARY KEY
        , payload varchar
        );
INSERT INTO b(id,payload) SELECT gs, 'bb_' || gs::varchar
FROM generate_series(1,20) gs;

CREATE TABLE b_import
        ( id INTEGER NOT NULL PRIMARY KEY
        , payload varchar
        );
INSERT INTO b_import(id,payload) SELECT gs, 'bb_' || gs::varchar
FROM generate_series(10,15) gs;
        -- In real life this table will be filled by a `COPY b_import FROM ...`
INSERT INTO b_import(id,payload) SELECT gs, 'b2_' || gs::varchar
FROM generate_series(16,25) gs;

CREATE TABLE a
        ( id SERIAL NOT NULL PRIMARY KEY
        , b_id INTEGER references b(id) ON DELETE SET NULL
        , aaaaa varchar
        );
INSERT INTO a(b_id,aaaaa)
SELECT gs,'aaaaa_' || gs::text FROM generate_series(1,20) gs;
CREATE INDEX ON a(b_id); -- index supporting the FK

        -- show it
SELECT a.id, a.aaaaa
        ,b.id, b.payload AS oldpayload
FROM a
FULL JOIN b ON a.b_id=b.id
ORDER BY a.id;

        -- Do the actual I/U/D and report the numbers of affected rows
-- EXPLAIN
WITH ins AS (   -- INSERTS
        INSERT INTO b(id, payload)
        SELECT b_import.id, b_import.payload
        FROM b_import
                WHERE NOT EXISTS (
                SELECT 1 FROM b
                WHERE b.id = b_import.id
                )
        RETURNING b.id
        )
, del AS (      -- DELETES
        DELETE FROM b
        WHERE NOT EXISTS (
                SELECT 2 FROM b_import
                WHERE b_import.id  = b.id
                )
        RETURNING b.id
        )
, upd AS (      -- UPDATES
        UPDATE b
        SET payload=b_import.payload
        FROM b_import
        WHERE b_import.id = b.id
        AND b_import.payload IS DISTINCT FROM b.payload -- exclude idempotent updates
        -- AND NOT EXISTS (     -- exclude deleted records
                -- SELECT 3 FROM del
                -- WHERE del.id = b_import.id
                -- )
        -- AND NOT EXISTS (     -- avoid touching freshly inserted rows
                -- SELECT 4 FROM ins
                -- WHERE ins.id = b_import.id
                -- )
        RETURNING b.id
        )
SELECT COUNT(*) AS orgb
        , (SELECT COUNT(*) FROM b_import) AS newb
        , (SELECT COUNT(*) FROM ins) AS ninserted
        , (SELECT COUNT(*) FROM del) AS ndeleted
        , (SELECT COUNT(*) FROM upd) AS nupdated
FROM b
        ;

---

• 删除约束并在导入后重建它的成本很高：所有A 和B 中的记录都涉及。
• 暂时忽略约束是危险：新的B 表可能会丢失一些仍被A 的FK 引用的行。
• ergo：您最终可能会得到一个残缺的模型，并且您必须重建 As 引用（这基本上是不可能的，没有额外的信息（这将是多余的，顺便说一句））