PostgresのSELECT COUNTは... UPDATE

Question

ためにここで私がやろうとしている何のための擬似コードです：

rate_count = SELECT COUNT(id) FROM job WHERE last_processed_at >= ? 

current_limit = rate_limit - rate_count 
if current_limit > 0 
    UPDATE job SET state='processing' 
    WHERE id IN(
    SELECT id FROM job 
    WHERE state='pending' 
    LIMIT :current_limit 
) 

私はそれが並行性の問題を除いて取り組んでいます。同時に複数のセッションから実行すると、両方のセッションがSELECTとなり、同じものが更新されます:(

2番目のクエリは、SELECTサブクエリにFOR UPDATEを追加することでアトミックになりますが、追加できません最初のクエリに対するUPDATE FOR UPDATEが集約関数では使用できませんFOR

がどのように私はこの作品にアトミックトランザクションを作ることができるので、

Answer 1

あなたは、サブクエリ内FOR UPDATEを行うことができます？

rate_count := COUNT(id) 
       FROM (
       SELECT id FROM job 
       WHERE last_processed_at >= ? FOR UPDATE 
       ) a; 

このすべてのことを単一のクエリ：

UPDATE job SET state='processing' 
WHERE id IN (
    SELECT id FROM job 
    WHERE state='pending' 
    LIMIT (SELECT GREATEST(0, rate_limit - COUNT(id)) 
     FROM (SELECT id FROM job 
       WHERE last_processed_at >= ? FOR UPDATE) a 
     ) 
)