Postgres generate_series - PullRequest
Новая
       20

Postgres generate_series

4 голосов
/ 21 августа 2011

Я хочу сделать статистику по таблице, и для этого я использую generate_series();

Вот что я делаю: 

SELECT x.month, amount
FROM (SELECT generate_series(
                 min(date_trunc('month', date)),
                 max(date_trunc('month', date)),
                 '1 month'
      ) AS month
      FROM table
      WHERE user_id = 55 AND ...
) x
LEFT JOIN (
      SELECT SUM(amount) AS amount, date_trunc('month', date) AS month
      FROM table
      WHERE user_id = 55 AND ...
      GROUP BY month
) q ON q.month = x.month
ORDER BY month

Это работает хорошо, но когда я хочу применить фильтры, такие как получить сумму для конкретных пользователей, я должен применить их дважды. Есть ли способ избежать фильтрации дважды или переписать это более эффективным способом, потому что я не уверен, что это правильный способ сделать это?

Ответы [ 2 ]

9 голосов
/ 21 августа 2011

Вы можете написать WITH запрос для этого: 

WITH month_amount AS
(
    SELECT
        sum(amount) AS amount,
        date_trunc('month', date) AS month
    FROM Amount
    WHERE user_id = 55 -- AND ...
    GROUP BY month
)
SELECT month, amount
FROM
    (SELECT generate_series(min(month), max(month), '1 month') AS month
    FROM month_amount) x
LEFT JOIN month_amount
USING (month)
ORDER BY month;

Пример результата: 

SELECT * FROM amount WHERE user_id = 55;
 amount_id | user_id | amount |    date    
-----------+---------+--------+------------
         3 |      55 |      7 | 2011-03-16
         4 |      55 |      5 | 2011-03-22
         5 |      55 |      2 | 2011-05-07
         6 |      55 |     18 | 2011-05-27
         7 |      55 |      4 | 2011-06-14
(5 rows)

WITH month_amount ..
         month          | amount 
------------------------+--------
 2011-03-01 00:00:00+01 |     12
 2011-04-01 00:00:00+02 |       
 2011-05-01 00:00:00+02 |     20
 2011-06-01 00:00:00+02 |      4
(4 rows)
6 голосов
/ 21 августа 2011

Вы можете выполнить свой запрос в предложении WITH, а затем использовать SELECT для добавления пропущенных месяцев: 

WITH query AS (
      SELECT SUM(amount) AS amount, date_trunc('month', date) AS month
      FROM table
      WHERE user_id = 55 AND ...
      GROUP BY month
)
SELECT date(d.month) AS month, q.amount
FROM (
    SELECT generate_series(min(month), max(month), '1 month') AS month
    FROM query
    ) d
    LEFT JOIN query q ON q.month = d.month
ORDER BY month
...