返回具有所有三个序列号的 id 的所有行

Question

我有一个看起来像这样的表：

table1

+----+-----+------+
| id | seq | test |
+----+-----+------+
|  1 |   1 | HR   |
|  1 |   2 | RR   |
|  2 |   1 | HR   |
|  2 |   2 | RR   |
|  2 |   3 | OXY  |
|  3 |   1 | HR   |
|  3 |   2 | RR   |
|  4 |   1 | HR   |
|  4 |   2 | RR   | 
|  4 |   3 | OXY  | 
+----+-----+------+

我想得到如下结果表。也就是说，只有当特定 id 的所有三个序列号都存在时，我才需要拥有特定 id 的所有行：

+----+-----+------+
| id | seq | test |
+----+-----+------+
|  2 |   1 | HR   |
|  2 |   2 | RR   |
|  2 |   3 | OXY  |
|  4 |   1 | HR   |
|  4 |   2 | RR   |
|  4 |   3 | OXY  | 
+----+-----+------+

我期待着编写一个 plpgsql 函数来为我提供解决方案。我对 plpgsql 和一般编程比较陌生。如果有人帮助我获得结果，那就太好了。

到目前为止，这是我的功能的样子，但它不完整：

CREATE OR REPLACE FUNCTION test()
returns SETOF table1 AS $$
DECLARE
    cur CURSOR FOR
        SELECT *
        FROM table1
        ORDER by id;
    rec_cur RECORD;
    counter INTEGER DEFAULT 0;

BEGIN 
    OPEN cur;

    FETCH FIRST FROM cur INTO rec_cur;
    MOVE RELATIVE +1 FROM cur;

    LOOP
        FETCH cur INTO rec_cur;
        EXIT WHEN NOT FOUND;

        IF rec_cur.seq = 1 AND counter = 0 THEN
        RETURN NEXT rec_cursor;
        END IF;

    END LOOP;
    CLOSE cur;
    RETURN;

END ; $$ 
LANGUAGE PLPGSQL STABLE PARALLEL SAFE;

Answer 1

您的问题不完整。
如果我们可以假设存在带有seq = 1 和seq = 2 的行，如果有一行带有seq = 3 的行对于相同的id，然后它变得便宜又简单：

SELECT *
FROM  (SELECT id FROM table1 WHERE seq = 3) x
JOIN   table1 t USING (id)
-- ORDER BY id, seq;  -- unclear whether you need sorted output.

还假设 (id, seq) 定义为 UNIQUE 和两列 NOT NULL。

如果您需要优化读取性能，请添加部分索引：

CREATE INDEX foo ON table1 (id) WHERE seq = 3;

Since Postgres 9.6 this can be used in an index-only scan.

当然，您还需要(id) 的索引。 (id, seq) 上的索引如果你说UNIQUE 约束也可以完成这项工作。相关：

• Does a Postgres UNIQUE constraint imply an index?
• Is a composite index also good for queries on the first field?
• PostgreSQL does not use a partial index

不管怎样，都是relational-division的情况。如果我们不能假设seq 中的顺序值，这里有一个技术库来识别符合条件的id：

• How to filter SQL results in a has-many-through relation

Answer 2

游标绝对不是正确的方法。您可以使用聚合和having 轻松获取 id：

select id
from t
where seq in (1, 2, 3)
group by id
having count(seq) = 3;

那么获取原始行，有多种方式：

select t.*
from t join
     (select id
      from t
      where seq in (1, 2, 3)
      group by id
      having count(seq) = 3
     ) tt
     on t.id = tt.id;

编辑：

如果序列号总是从 1 开始并且没有间隙，那么窗口函数就是要走的路：

select t.*
from (select t.*, max(t.seq) over (partition by t.id) as maxseq
      from t
     ) t
where maxseq = 3;