обновление 2 столбца на основе значений из другой таблицы

Question

У меня есть 2 таблицы, и я хочу написать 2 обновления.
tbl_stage с идентификатором, флагом (null) и счетом (null).
tbl_source с идентификатором, счетом, обновленной колонкой.

Перед обновлением

tbl_stage     tbl_source 
id flag count  id count updated 
abc NULL NULL  abc 9.0  false 
          def 3.6  false 

Я хочу обновить tbl_stage.count = tbl_source.count и tbl_stage.flag = истинное, где tbl_stage.id = tbl_source.id.
И еще один, tbl_source.updated = true, где tbl_stage.id = tbl_source.id.

После обновления должно выглядеть следующим образом.

tbl_stage     tbl_source 
id flag count  id count updated 
abc true 9.0   abc 9.0  true 
          def 3.6  false 

Answer 1

Вам необходимо использовать 2 оператора UPDATE, так как обновление 2 таблиц в одном из операторов недоступно.

update tbl_stage 
set [count]=tbl_source.count 
    ,flag = 1 
FROM tbl_source WHERE tbl_stage.id=tbl_source.id 

update tbl_source 
set updated = 1 
from tbl_source WHERE tbl_stage.id=tbl_source.id 
and tbl_stage.count=tbl_source.count 

Answer 2

Вы можете сделать это с помощью одного оператора, меняя обе таблицы на одном дыхании:

with stage_update as (
    update stage 
    set cnt = s.cnt, 
     flag = true 
    from source s 
    where stage.id = s.id 
    returning s.id 
) 
update source 
    set updated = true 
where id in (select id from stage_update); 

---

^{(я выбрал cnt для имени столбца, потому что count является зарезервированным словом, и не следует использовать в качестве идентификатора)}