Postgres函数,它接受一个间隔并返回一个结果集

时间:2013-09-24 21:49:37

标签: sql postgresql

我希望能够创建一个函数,该函数接收一个“标准化”查询集,该查询集具有整数id字段和时间戳作为列,然后修改查询集以按每个30中的数字的计数聚合最小块。如果用真实查询替换Q并执行WITH(QUERY)Q,下面的代码可以工作30分钟。但我希望能够传入任何符合该类型的查询,然后返回新的SETOF range_durations_type。

CREATE TYPE range_durations_type AS (dtime TIMESTAMP, counts BIGINT);
CREATE TYPE object_datetime AS (id INTEGER, dtime TIMESTAMP);

CREATE OR REPLACE FUNCTION time_range_durations(Q range_durations_type, start_date TIMESTAMP, end_date TIMESTAMP)
  RETURNS SETOF range_durations_type AS $$
BEGIN

  RETURN QUERY WITH q_cleaned AS (
    SELECT
      Q.id,
      Q.dtime,
      Q.dtime - (
                  (extract(EPOCH FROM Q.dtime :: TIMESTAMP)) :: INTEGER -
                  extract(EPOCH FROM Q.dtime :: TIMESTAMP) :: INTEGER / 1800 * 1800
                ) * INTERVAL '1 second' AS rounded_dtime
    FROM Q
  )
               SELECT
                 series.dtime,
                 count(q_cleaned.id)
               FROM (
                      SELECT
                        *
                      FROM
                        generate_series(start_date, end_date, '30 minutes') dtime) series
                 LEFT JOIN q_cleaned ON series.dtime = q_cleaned.rounded_dtime
               GROUP BY 1
               ORDER BY 1;

END;
$$ LANGUAGE plpgsql;

1 个答案:

答案 0 :(得分:0)

我认为一个完整的解决方案在这里将超出一个单一的答案,但我建议直接看一些重要的部分。

  1. 范围类型。在我看来,你正在寻找一组间隔。这将允许您执行以下操作:

      CREATE OR REPLACE FUNCTION time_range_durations(bounds rsrange, inc interval)
    

    这将允许你做一些非常不同的事情,即范围和可用于在其中递增的间隔。因此,您可以通过任意增量查看两个时间戳内的内容。

  2. 然后您可以将inc传递给您的generate_series函数。