Sobre la base de esta tabla (sin usar la palabra clave SQL "date" como nombre de columna.):
CREATE TABLE tbl(
pid int
, the_date date
, PRIMARY KEY (pid, the_date)
);
Consulta:
SELECT pid, the_date
, row_number() OVER (PARTITION BY pid, grp ORDER BY the_date) AS in_streak
FROM (
SELECT *
, the_date - '2000-01-01'::date
- row_number() OVER (PARTITION BY pid ORDER BY the_date) AS grp
FROM tbl
) sub
ORDER BY pid, the_date;
Restar una date de otra date produce un integer . Dado que está buscando días consecutivos, cada fila siguiente sería mayor en uno . Si restamos row_number() a partir de ahí, toda la racha termina en el mismo grupo (grp ) por pid . Entonces es sencillo repartir números por grupo.
grp se calcula con dos restas, que deberían ser las más rápidas. Una alternativa igualmente rápida podría ser:
the_date - row_number() OVER (PARTITION BY pid ORDER BY the_date) * interval '1d' AS grp
Una multiplicación, una resta. La concatenación de cadenas y la conversión son más costosas. Prueba con EXPLAIN ANALYZE .
No olvides particionar por pid adicionalmente en ambos pasos, o sin querer mezclará grupos que deberían estar separados.
Usar una subconsulta, ya que suele ser más rápida que un CTE . No hay nada aquí que una simple subconsulta no pueda hacer.
Y ya que lo mencionaste:dense_rank() obviamente no necesario aquí. Básico row_number()
hace el trabajo.