Question

我正在使用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_id和C.b_id分别是表id和A的{{1}}列的外键。

我要做的是使用纯SQL在这三个表中的每一个表中插入一行，而不将ID硬编码到SQL中（在运行此脚本之前对数据库进行假设似乎是不合需要的，因为如果这些假设发生变化，我将不得不返回并重新计算所有测试数据的正确ID。）

请注意，我确实意识到我可以以编程方式执行此操作，但通常编写纯SQL比编写执行SQL的程序代码更简洁，因此对测试套件数据更有意义。

无论如何，这是我写的查询，我认为可行：

然而，这不起作用，导致PostgreSQL告诉我：

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;

是否有正确的方法来编写这种类型的查询，而不对ID值进行硬编码？

（您可以找到包含此示例的小提琴here。）

Answer 1

不要嵌套公用表表达式，只需一个接一个地写：

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;

PostgreSQL嵌套INSERTs / WITHs用于外键插入

1 个答案: