用于外键插入的PostgreSQL嵌套INSERT/WITH

cbjzeqam  于 2023-05-17  发布在  PostgreSQL
关注(0)|答案(1)|浏览(86)

我使用的是PostgreSQL 9.3,我试图编写一个SQL脚本来插入一些单元测试的数据,我遇到了一点问题。
假设我们有三个表,结构如下:

------- Table A -------    -------- Table B --------    -------- Table C --------
id  | serial NOT NULL      id   | serial NOT NULL       id   | serial NOT NULL
foo | character varying    a_id | integer NOT NULL      b_id | integer NOT NULL
                           bar  | character varying     baz  | character varying

B.a_idC.b_id分别是表ABid列的外键。
我想做的是用纯SQL向这三个表中的每一个插入一行,而不是将ID硬编码到SQL中(在运行此脚本之前对数据库进行假设似乎是不可取的,因为如果这些假设发生变化,我将不得不返回并重新计算所有测试数据的正确ID)。
请注意,我确实意识到我可以通过编程来实现这一点,但是一般来说,编写纯SQL比编写执行SQL的程序代码要少得多,所以它对测试套件数据更有意义。
无论如何,这是我写的查询,我认为它会起作用:

WITH X AS (
    WITH Y AS (
        INSERT INTO A (foo)
        VALUES ('abc')
        RETURNING id
    )
    INSERT INTO B (a_id, bar)
    SELECT id, 'def'
    FROM Y
    RETURNING id
)
INSERT INTO C (b_id, baz)
SELECT id, 'ghi'
FROM X;

然而,这不起作用,并导致PostgreSQL告诉我:

ERROR:  WITH clause containing a data-modifying statement must be at the top level

有没有一种正确的方法来编写这种类型的查询,而不需要对ID值进行硬编码?
(You可以在这里找到一个包含这个例子的小提琴。

ndh0cuux

ndh0cuux1#

不要嵌套公共表表达式,只需一个接一个地编写:

WITH Y AS (
  INSERT INTO A (foo)
  VALUES ('abc')
  RETURNING id
), X as (
  INSERT INTO B (a_id, bar)
  SELECT id, 'def'
  FROM Y
  RETURNING id
)
INSERT INTO C (b_id, baz)
SELECT id, 'ghi'
FROM X;

相关问题