SQLServer 批量修改或插入场景

今天在工作中遇到这么一个场景，我需要根据条件对表A做批量更新或插入。因为条件比较复杂，所以我使用了临时表B，先把需要更新或插入的数据查询出来放入临时表。然后更新表A的某字段，更新条件是A.id = B.id，更新效果是若记录存在表A中，则更新表A的字段，若不在表A中，则插入一条记录到表A。

初始方案

首先想到的是写两条SQL：

  1. update使用A.id in (select id from B)，这时会更新所有存在的记录的字段，不存在则不更新
  2. 接下来再使用insert，条件是A.id not in (select id from B)，把A不存在未更新的数据插入到A

两条SQL执行完，则完成了所有数据的更新和插入。

优化方案

后面有朋友提示可以使用SQL Merge语句做这件事情，由于对数据库并不熟悉，所以不知道还有这么一个语法。我查阅了相关资料，发现Merge确实很合适这个场景。参考SQL Server Merge语句 – SQL Server教程 (yiibai.com)重构了代码

select ... into #temp from A  where ... merge [A] a using #B b      on a.id = b.id      when matched      then update set a.field = 1     when not matched by target      then insert (field)  values (b.id);