Início / dba / Perguntas / 341593
Accepted
Thomas
Asked: 2024-08-13 00:14:48 +0800 CST

Crie grupos sequenciais de linhas que ultrapassem uma soma mínima

  • 772

Digamos que eu tenha uma tabela com um registro de data e hora ( ts) e um valor ( a).

Com alguns dados de amostra:

ts, a
1, 10
2, 30
3, 10
4, 20
5, 40
6, 10
7, 20
8, 30
etc

Gostaria de agrupar linhas, na ordem ts, somando ae gerando uma linha onde a soma de aseja > 50.

Neste caso, a saída deve ser dois registros de data e hora (primeira e última linha):

1, 3  -- 10 + 30 + 10
4, 5  -- 20 + 40
6, 8  -- 10 + 20 + 30

Mas não tenho a mínima ideia de como proceder. Todas as tentativas que fiz com o Copilot produziram código complexo que nunca termina.

postgresql

1 respostas

  1. Best Answer

    Cada próxima linha depende de todas as linhas anteriores. Isso se presta a uma solução procedural .
    Crie esta função PL/pgSQL uma vez:

    CREATE OR REPLACE FUNCTION f_tbl_chunks(_chunk int)
  RETURNS TABLE (ts1 int, ts2 int, total int)  -- adapt to actual types!
  LANGUAGE plpgsql STABLE AS
$func$
DECLARE
   _r record;
BEGIN
   total := 0;  -- init

   FOR _r IN
      SELECT ts, a
      FROM   tbl
      ORDER  BY ts
   LOOP
      IF total = 0 THEN
         ts1 := _r.ts;  -- add first ts
      END IF;

      total := total + _r.a;

      IF total >= _chunk THEN
         ts2 := _r.ts;  -- add last ts
         RETURN NEXT;
         total := 0;    -- reset total
      END IF;
   END LOOP; 

   -- return remainder, if any?
   IF total > 0 THEN
      ts2 = _r.ts;
      RETURN NEXT;
   END IF;
END
$func$;

    Chamar:

    SELECT * FROM f_tbl_chunks(50);

    violino

    Adicionei o total por grupo, o que é fácil de suprimir.

    Ou use uma função temporária se você não quiser (ou não puder) persistir uma. Veja:

    Você pode dobrar a natureza baseada em conjunto do SQL com um CTE recursivo. Mas isso é difícil de manejar e lento em comparação. Veja:

    • 0

relate perguntas