How to SUM() over column with reset condition?

后端 未结 2 546
逝去的感伤
逝去的感伤 2021-01-03 16:05

I\'m using Postgresql 9.2 and I need to sum quantitys from buttom up with initial value of 100, however if I encounter a row with name X i need to restart the SUM from the

相关标签:
2条回答
  • 2021-01-03 16:31

    You need a RECURSIVE query

    SQL FIDDLE DEMO

    WITH RECURSIVE t("itemorder", "name", "qty", "modifyed_sum", "level") AS (
        SELECT "itemorder", 
               "name", 
               "qty", 
               100 +  CASE 
                         WHEN "name" = 'A' THEN "qty"
                         ELSE 0
                      END as "modifyed_sum",
               "itemorder" as "level"
         FROM myTable
         WHERE "itemorder" = (SELECT max("itemorder") FROM myTable)
         UNION ALL
         SELECT myTable."itemorder", 
                myTable."name", 
                myTable."qty", 
                CASE 
                    WHEN myTable."name" = 'A' THEN t."modifyed_sum" + myTable."qty"
                    WHEN myTable."name" = 'X' THEN myTable."qty"
                    ELSE t."modifyed_sum"          
                END as "modifyed_sum",
               myTable."itemorder" as "level"
         FROM myTable 
         JOIN t
           ON myTable."itemorder" + 1 = t."itemorder" 
          AND myTable."itemorder" = t."level" - 1
         WHERE  myTable."itemorder" > 0        
    )
    SELECT *
    FROM t
    ORDER BY "itemorder"
    

    OUTPUT

    | itemorder | name | qty | modifyed_sum | level |
    |-----------|------|-----|--------------|-------|
    |         1 |    A | -20 |           20 |     1 |
    |         2 |   A2 | 350 |           40 |     2 |
    |         3 |    X |  40 |           40 |     3 |
    |         4 |    A |  50 |          140 |     4 |
    |         5 |    A | -10 |           90 |     5 |
    |         6 |   A2 |  10 |          100 |     6 |
    
    0 讨论(0)
  • 2021-01-03 16:42

    To reset a Windowed Aggregate based on a condition you need to add another level to define the different groups of rows:

    with cte as
     (
       select itemorder, name, qty,
          case -- start qty for each group
             when itemorder = max(itemorder) over () then 100 -- newest row
             when name = 'X' then qty
             else 0
          end 
               -- only sum 'A' rows
          + case when name = 'A' then qty else 0 end as new_qty, 
    
               -- grp value increases only when name = 'X'
               -- this assigns the same value to each row in a group
               -- will be used in the next step to PARTITION BY
          sum(case when name = 'X' then 1 else 0 end) 
          over (order by itemorder desc
                rows unbounded preceding) as grp
       from myTable
     )
    
    select itemorder, name, qty,
       sum(new_qty)
       over (partition by grp
             order by itemorder desc
             rows unbounded preceding ) as modified_sum
    from cte
    order by item order
    

    See fiddle

    0 讨论(0)
提交回复
热议问题