SQL点滴18—SqlServer中的merge操作，相当地风骚

原文:SQL点滴18—SqlServer中的merge操作，相当地风骚

今天在一个存储过程中看见了merge这个关键字，第一个想法是，这个是配置管理中的概念吗，把相邻两次的更改合并到一起。后来在technet上搜索发现别有洞天，原来是另外一个sql关键字，t-sql的语法还是相当地丰富的。本篇是一篇学习笔记，没有什么新意，这里给出technet上的地址连接供大家参考权威：http://technet.microsoft.com/zh-cn/library/bb510625.aspx，这里具体的语法不去深究了，只是把几个例子实际运行，剖析一番。

　　

使用merge同时执行insert和update操作

我们经常会有这样的需求，根据某个字段或多个字段查找表中的一行或多行数据，如果查找成功得到匹配项，更新其中的其他一个或多个字段；如果查找失败则将“某个字段或多个字段”作为新的一行中的数据插入到表中。第一种方法是先更新，然后根据@@rowcount判断是否有匹配项，如果没有则插入。先使用下面的 代码创建一个存储过程。

 use AdventureWorks
 go
 create procedure dbo.InsertUnitMeasure @UnitMeasureCode nchar(3),@Name nvarchar(25)
 as
 begin
 set nocount on;
 update Production.UnitMeasure set Name=@Name where UnitMeasureCode=@UnitMeasureCode
 if(@@ROWCOUNT=0)
   begin
   insert into Production.UnitMeasure(Name,UnitMeasureCode)values(@Name,@UnitMeasureCode)
   end
 end
 go

记得见过这样的笔试题目，要求是插入不存在的行，只要把上面语句中的update改成select就可以了，当时没有写出来，现在恍然大悟，也许是在考察@@ROWCOUNT的用法吧。这个语句也可以使用merge语句实现。下面我们使用merge关键字来修改这个存储过程。

 alter procedure dbo.InsertUnitMeasure @UnitMeasureCode nchar(3),@Name nvarchar(25)
 as
 begin
 set nocount on
 merge Production.UnitMeasure as target
 using (select @UnitMeasureCode,@Name) as source (UnitMeasureCode,Name)
 on (target.UnitMeasureCode=source.UnitMeasureCode)
 when matched then update set Name=source.Name
 when not matched then insert(UnitMeasureCode,Name)values(source.UnitMeasureCode,Name)
 output deleted.*,$action,inserted.* into MyTempTable;
 end
 go

这个语句使用merge修改存储过程，这个语句中又出现我不太了解的关键字using和$action。Using是用来指定和表InsertUnitMeasure中相匹配的数据源，这里的数据源来自外部输入，是通过两个输入参数得到。$action可能是一个占位符，表示上面的when字句进行的操作。至于inserted.*和deleted.* 就是插入和删除的数据行了，这个我在其中一篇文章中也提到，他们有点类似类中的this关键字，过可以看看：SQL点滴14—编辑数据。注意为了记录修改的过程我们需要创建一个临时表#MyTempTable来跟踪修改过程，所以在调用这个存储过程之前我们需要新建这个表，语句如下：

 create table MyTempTable(
 ExistingCode  nchar(3),
 ExistingName  nvarchar(50),
 ExistingDate  datetime,
 ActionTaken   nvarchar(50),
 NewCode       nchar(3),
 [NewName]     nvarchar(50),      
 NewDate       datetime
 )
 Go

现在我们来执行下面的语句看看有什么样的结果：

 exec InsertUnitMeasure @UnitMeasureCode = 'ABC',@Name='New Test Value1'
 EXEC InsertUnitMeasure @UnitMeasureCode = 'XYZ', @Name = 'Test Value';
 EXEC InsertUnitMeasure @UnitMeasureCode = 'ABC', @Name = 'Another Test Valuea';
 Go

首先使用语句：select * from Production.UnitMeasure order by ModifiedDate desc 来查看目标表中的数据变化如图1：

图1

这里虽然三次执行了存储过程，但是由于第一次和第三次的@UnitMeasureCode的值是相同的’ABC’所以第二次肯定是进行更新操作。所以最后表中新增了两条记录。然后使用下面的语句查看记录表MyTempTable中的跟踪信息如图2

图2

我们可以看到前面两条语句执行的是插入操作，所以原有的值都是空，因为在插入之前他们还不存在。第三条新型的是更新操作，更新UnitMeasureCode为’ABC’的记录。
 
　　

使用merge在单个语句中执行insert和update操作

在AdventureWorks数据库中有ProductInventory表，存储的是存货信息，SalesOrderDetail表中存储的是订单信息，现在如果每天减去对SalesOrderDetail表中每种产品所下的订单数，更新ProductInventory表中的 Quantity列。如果随着时间推移订单数导致产品库存量下降到0或者更少，则从ProductInventory表中删除该产品对应的行。下面的语句创建一个存储过程实现上面的逻辑。

 CREATE PROCEDURE Production.usp_UpdateInventory
     @OrderDate datetime
 AS
 MERGE Production.ProductInventory AS target
 USING (SELECT ProductID, SUM(OrderQty) FROM Sales.SalesOrderDetail AS sod
     JOIN Sales.SalesOrderHeader AS soh
     ON sod.SalesOrderID = soh.SalesOrderID
     AND soh.OrderDate = @OrderDate
     GROUP BY ProductID) AS source (ProductID, OrderQty)
 ON (target.ProductID = source.ProductID)
 WHEN MATCHED AND target.Quantity - source.OrderQty <= 0
     THEN DELETE
 WHEN MATCHED 
     THEN UPDATE SET target.Quantity = target.Quantity - source.OrderQty, 
                     target.ModifiedDate = GETDATE()
 OUTPUT $action, Inserted.ProductID, Inserted.Quantity, Inserted.ModifiedDate, Deleted.ProductID,
     Deleted.Quantity, Deleted.ModifiedDate;
 GO

这个语句比第一个要复杂一点，注意当匹配成功并且总量小于0的时候直接使用一个delete就可以将此条记录删除，output语句直接把操作结果输出，相当地神奇。最后运行下面的 语句得到如图3的结果。注意这个语句相当于将2003年5月1号的订单量减去。如果多次运行的话就相当于多减了一次，整个表中数据条数会减少的。

EXECUTE Production.usp_UpdateInventory '20030501'

图3

　　

借助派生源表，使用merge对目标表执行update和insert操作

这次我们已知有一些表数据，我们要和Sales.SalesReason这个表中的数据做对比，如果和SalesReason表中的Name字段匹配时就更新表中的ReasonType列，如果没有匹配项的时候就插入这一行新的数据。在这里是使用表值构造函数指定源表的多个行，使用表变量存储更新记录，注意表变量的使用范围。代码如下：

 declare @SummaryOfChanges table(Change varchar(20))
 merge into Sales.SalesReason as target
 using(values('Recommendation','Other'),('Review','Marketing'),('Internet','Promotion')) as source([NewName],NewReasonType)
 on target.Name=source.[NewName]
 when matched then update set ReasonType=source.NewReasonType
 when not matched by target then insert(Name,ReasonType) values ([NewName],NewReasonType)
 output $action into @SummaryOfChanges;
 select Change,COUNT(*) as CountPerChange from @SummaryOfChanges group by Change

执行完上面的语句之后我们得到下面的结果说明执行了2次插入，1次更新，如图4。那么是不是这样的 呢，我们查看Sales.SalesReason这个表发现原来已经有’Review’这一条数据了，对它执行了更新，剩下的’Recommendation’，’Internet’执行的是插入操作。如果再次执行上面的语句就会得到UPDATE 3这样的结果，因为已经存在这三条数据了所以都执行UPDATE。

图4

　　

将merge执行的结果插入到另外一个表中

我们还可以将merge操作得到的结果写入到另外一个表中，如下的语句将更新的每条数据信息写入到一个新建的表Production.UpdatedInventory中，代码如下：

 INSERT INTO Production.UpdatedInventory
 SELECT ProductID, LocationID, NewQty, PreviousQty 
 FROM
 (    MERGE Production.ProductInventory AS target
      USING (SELECT ProductID, SUM(OrderQty) 
             FROM Sales.SalesOrderDetail AS sod
             JOIN Sales.SalesOrderHeader AS soh
             ON sod.SalesOrderID = soh.SalesOrderID
             AND soh.OrderDate BETWEEN '' AND ''
             GROUP BY ProductID) AS source (ProductID, OrderQty)
      ON target.ProductID = source.ProductID
     WHEN MATCHED AND target.Quantity - source.OrderQty >= 0 
         THEN UPDATE SET target.Quantity = target.Quantity - source.OrderQty
     WHEN MATCHED AND target.Quantity - source.OrderQty <= 0 
         THEN DELETE
     OUTPUT $action, Inserted.ProductID, Inserted.LocationID, Inserted.Quantity AS NewQty, Deleted.Quantity AS PreviousQty)
  AS Changes (Action, ProductID, LocationID, NewQty, PreviousQty) WHERE Action = 'UPDATE';
 GO

执行这个语句再查询表得到如下图5的结果，我们可以看到新的销售量总是比以前的销售量要少，因为执行一次就要减去订单量。

图5

这里我们只记录了更新的变化，如果想记录所有的操作可以去掉最后的一个限制条件WHERE Action = 'UPDATE'，那就要修改记录表的结构了，这个和第二个例子有些相似，只不过将记录在实际的表中，而第二个例子仅仅输出这些操作记录。