Sql 如何在Postgres中为单值输入多行

发布于10月27日

我有一张桌子

CREATE TABLE test_tbl(id integer, num integer, name text);

数据在哪里

INSERT INTO test_tbl (id, num, name)
VALUES(1, 2000, 'fdghgf'),(2, 3000, 'fdghg'),(3, 400, 'fhgf'),(4, 2200, 'dhgf');

我想将此数据插入到另一个表tbl_diburaction中

CREATE TABLE tbl_bifuraction(id integer, num integer, name text, is_bifurcated integer);

因此，当金额大于1000时，它会插入多个包含1000和剩余金额的行(当剩余金额为0时没有额外的行).

插入的数据集应类似于:

id  num  name   is_bifurcated
1   1000 fdghgf 1
1   1000 fdghgf 1
2   1000 fdghg  1
2   1000 fdghg  1
2   1000 fdghg  1
3   400  fhgf   0
4   1000 dhgf   1
4   1000 dhgf   1
4   200  dhgf   1

为bifuracted_query添加了小提琴链接

推荐答案

关键是使用generate_series()执行横向联接，这样您就可以获得所需的结果行数:

SELECT t.id,
       CASE WHEN (g.c + 1) * 1000 > t.num
            THEN t.num - g.c * 1000
            ELSE 1000
       END AS num,
       t.name,
       CAST(t.num > 1000 AS integer) AS is_bifurcated
FROM test_tbl AS t
   CROSS JOIN LATERAL generate_series(0, (num - 1) / 1000) AS g(c);

 id │ num  │  name  │ is_bifurcated 
════╪══════╪════════╪═══════════════
  1 │ 1000 │ fdghgf │             1
  1 │ 1000 │ fdghgf │             1
  2 │ 1000 │ fdghg  │             1
  2 │ 1000 │ fdghg  │             1
  2 │ 1000 │ fdghg  │             1
  3 │  400 │ fhgf   │             0
  4 │ 1000 │ dhgf   │             1
  4 │ 1000 │ dhgf   │             1
  4 │  200 │ dhgf   │             1
(9 rows)