環路

Question

的Postgres 9.2 PL/pgSQL的簡單更新我有如下表：

+----+----------+----------+ 
| id | trail_id | position | 
+----+----------+----------+ 
| 11 | 16  | NULL  | 
| 12 | 121  | NULL  | 
| 15 | 121  | NULL  | 
| 19 | 42  | NULL  | 
| 20 | 42  | NULL  | 
| 21 | 42  | NULL  | 
+----+----------+----------+ 

而且我在尋找一個簡單的方法來更新position與遞增的整數（每個父）。所以我完成後，該表應該是這樣的：

+----+-----------+----------+ 
| id | trail_id | position | 
+----+-----------+----------+ 
| 11 | 16  | 1  | 
| 12 | 121  | 1  | 
| 15 | 121  | 2  | 
| 19 | 42  | 1  | 
| 20 | 42  | 2  | 
| 21 | 42  | 3  | 
+----+-----------+----------+ 

我想我需要的，是一個功能，即遍歷一個給定trail所有行，有一個簡單的遞增索引並更新position柱。不過，我是一名pgSQL新手，所以我很樂意聽到有更簡單的方法來做到這一點。

我想，現在的解決方案是這樣的

CREATE FUNCTION fill_positions(tid integer) RETURNS integer AS $$ 
DECLARE 
    pht RECORD; 
    i INTEGER := 0; 
BEGIN 
    FOR pht IN SELECT * FROM photos WHERE photos.trail_id = tid LOOP 
     i := i + 1; 
     UPDATE photos SET position = i WHERE id = pht.id; 
    END LOOP; 
    RETURN i; 
END; 
$$ LANGUAGE plpgsql; 

我敢肯定它可能是更清潔，它不具有使用功能。

Answer 1

您不需要爲此存儲的函數。你可以用一條語句來做到這一點。

with new_numbers as (
    select id, 
      trail_id, 
      row_number() over (partition by trail_id order by id) as position 
    from photos 
) 
update photos 
    set position = nn.position 
from new_numbers nn 
where nn.id = photos.id;