如何创建将一个表中的序列号添加到另一个表的 INSERT 查询

Question

我在 Postgres 10.7 数据库中有一个表 sample_1，其中包含一些纵向研究数据和每个 key 的递增 sequence 数字。我需要从暂存表 (sample_2) 中的 INSERT 数据相应地维护 sequence 列。

sequence 数字从 0 开始。我假设我需要一个查询来查找sample_1 中每个key 的最大sequence 数字，并将其添加到每个新行的后续序列号中。在这一步，我主要在序列号算术上苦苦挣扎。试过这个：

INSERT INTO sample_1 (KEY, SEQUENCE, DATA)
SELECT KEY, sample_2.SEQUENCE + max(sample_1.SEQUENCE), DATA
FROM sample_2;

但是，我收到错误消息，说我不能在第 2 行中使用“sample_1.SEQUENCE”，因为这是要插入的表。我不知道如何用我的插入序列进行算术运算！

样本数据：

sample_1

| KEY         | SEQUENCE | DATA |
+-------------+----------+------+
| YMH_0001_XX |     0    |  a   |
| YMH_0001_XX |     1    |  b   |
| YMH_0002_YY |     0    |  c   |

sample_2

| KEY         | SEQUENCE | DATA |
+-------------+----------+------+
| YMH_0001_XX |     1    |  d   |
| YMH_0002_YY |     1    |  e   |
| YMH_0002_YY |     2    |  f   |

对于插入的行，我想继续对每个 key 的 sequence 数字进行升序。

为了清楚起见，此示例中的结果表将是 3 列和 6 行：

sample_1

| KEY         | SEQUENCE | DATA |
+-------------+----------+------+
| YMH_0001_XX |     0    |  a   |
| YMH_0001_XX |     1    |  b   |
| YMH_0001_XX |     2    |  d   |
| YMH_0002_YY |     0    |  c   |
| YMH_0002_YY |     1    |  e   |
| YMH_0002_YY |     2    |  f   |

Answer 1

那应该做你所追求的：

INSERT INTO sample_1 (key, sequence, data)
SELECT s2.key
     , COALESCE(s1.seq_base, -1)
     + row_number() OVER (PARTITION BY s2.key ORDER BY s2.sequence)
     , s2.data
FROM   sample_2 s2
LEFT   JOIN (
   SELECT key, max(sequence) AS seq_base
   FROM   sample_1
   GROUP  BY 1
   ) s1 USING (key);

笔记

• 您需要在 sample_1 中的每个 key 的现有最大值 sequence 上进行构建。 （我将其命名为seq_base。）在子查询中计算并加入它。

• 将row_number() 添加到其中，如图所示。这保留了输入行的顺序，丢弃了绝对数字。

• 我们需要 LEFTJOIN 来避免使用来自 sample_2 的新键丢失行。
  同样，我们需要COALESCE 来为新键启动一个新序列。默认为 -1 以在添加基于 1 的行号后有效地以 0 开始序列。

• 这对于并发执行是不安全的，但我认为这不是您的用例。