Linq To SQL 批量更新方法汇总

方法一、官方例子

地球人都知道的，也是不少 Linq To SQL 反对者认为效率低下的一种方法。

NorthwindDataContext db = new NorthwindDataContext();
var customers = db.Customers.Where(c => c.CustomerID.StartsWith("BL"));
foreach (var customer in customers)
{
customer.Address = "Guangzhou";
customer.ContactName = "CoolCode";
customer.CompanyName = "Microsoft";
}
db.SubmitChanges();

这种方法必须要查询出要更新的数据，确实有点不雅，也是Linq To SQL 略显尴尬的一面。

方法二、使用ExpressionVisitor获取Lambda表达式生成的SQL条件语句

此方法是基于Jeffrey Zhao 的《扩展LINQ to SQL：使用Lambda Expression批量删除数据》，从该文章得到一点启发，继而有了批量更新。使用示例：

db.Customers.Update(c => c.CustomerID == "Bruce",
c => new Customer
{
Address = "Guangzhou",
ContactName = "CoolCode",
CompanyName = "Microsoft"
});

方法原型：

/// <summary>
/// 批量更新
/// </summary>
/// <typeparam name="T"></typeparam>
/// <param name="table">表</param>
/// <param name="predicate">查询条件表达式</param>
/// <param name="updater">更新表达式</param>
/// <returns>影响的行数</returns>
public static int Update<T>(this Table<T> table, Expression<Func<T, bool>> predicate, Expression<Func<T,T>> updater) where T : class

实现原理：扩展Table<T>，解释表达式树成SQL语句。其中解释表达式树包括和更新表达式，后者相对容易处理，例如表达式：

c => new Customer { Address = "Guangzhou", ContactName = "CoolCode", CompanyName = "Microsoft" }

解释成

Address = @Address, ContactName = @ContactName, CompanyName = @CompanyName

而相应的值（"Guangzhou", "CoolCode", "Microsoft" ）作为SQL参数传递。

实现这一步，其实就是从表达式 Expression<Func<T,T>> 中取到初始化的属性名字和值就可以，具体做法可以使用Expression Tree Viewer来辅助，从下图可以了解到 Expression<Func<T,T>> 的树形结构。





然后我按上面的结构图“照葫芦画瓢”就得到要更新的属性名字和值：

//获取Update的赋值语句
var updateMemberExpr = (MemberInitExpression)updater.Body;
var updateMemberCollection = updateMemberExpr.Bindings.Cast<MemberAssignment>().Select(c => new
{
Name = c.Member.Name,
Value = ((ConstantExpression)c.Expression).Value
});

而解释where条件就相对没这么轻松了。

这里同 Jeffrey Zhao 的批量删除一样，同样是借助 ExpressionVisitor 来解释。ExpressionVisitor 是 Expression Tree 的遍历器，它自身不会帮你生成任何东西，通过继承 ExpressionVisitor 就可以取表达式的任何信息，本文就是通过让 ConditionBuilder 继承ExpressionVisitor 而生成 Where 条件的 SQL。

注：Jeffrey Zhao 的批量删除一文提供的源代码中，ConditionBuilder 并不支持生成Like操作，如 字符串的 StartsWith，Contains，EndsWith 并不能生成这样的SQL： Like ‘xxx%’, Like ‘%xxx%’ , Like ‘%xxx’ 。我通过分析 ExpressionVisitor ，也不难发现只要override VisitMethodCall 这个方法即可实现上述功能。

protected override Expression VisitMethodCall(MethodCallExpression m)
{
if (m == null) return m;

string format;
switch (m.Method.Name)
{
case "StartsWith":
format = "({0} LIKE {1}+'%')";
break;

case "Contains":
format = "({0} LIKE '%'+{1}+'%')";
break;

case "EndsWith":
format = "({0} LIKE '%'+{1})";
break;

default:
throw new NotSupportedException(m.NodeType + " is not supported!");
}

this.Visit(m.Object);
this.Visit(m.Arguments[0]);
string right = this.m_conditionParts.Pop();
string left = this.m_conditionParts.Pop();
this.m_conditionParts.Push(String.Format(format, left, right));

return m;
}

到此刻，已经解决了解释表达式树的难题，那么实现通过表达式树生成完整的 Update SQL语句这个设想也不是什么难事了。

/// <summary>
/// 批量更新
/// </summary>
/// <typeparam name="T"></typeparam>
/// <param name="table">表</param>
/// <param name="predicate">查询条件表达式</param>
/// <param name="updater">更新表达式</param>
/// <returns>影响的行数</returns>
public static int Update<T>(this Table<T> table, Expression<Func<T, bool>> predicate, Expression<Func<T,T>> updater) where T : class{
//获取表名
string tableName = table.Context.Mapping.GetTable(typeof(T)).TableName;

//查询条件表达式转换成SQL的条件语句
ConditionBuilder builder = new ConditionBuilder();
builder.Build(predicate.Body);
string sqlCondition = builder.Condition;

//获取Update的赋值语句
var updateMemberExpr = (MemberInitExpression)updater.Body;
var updateMemberCollection = updateMemberExpr.Bindings.Cast<MemberAssignment>().Select(c => new
{
Name = c.Member.Name,
Value = ((ConstantExpression)c.Expression).Value
});

int i = builder.Arguments.Length;
string sqlUpdateBlock = string.Join(", ", updateMemberCollection.Select(c => string.Format("[{0}]={1}", c.Name, "{" + (i++) + "}")).ToArray());

//SQL命令
string commandText = string.Format("UPDATE {0} SET {1} WHERE {2}", tableName, sqlUpdateBlock, sqlCondition);

//获取SQL参数数组 (包括查询参数和赋值参数)
var args = builder.Arguments.Union(updateMemberCollection.Select(c => c.Value)).ToArray();

//执行
return table.Context.ExecuteCommand(commandText, args);
}

例如上面提到的示例所生成的 Updae SQL语句是：

UPDATE dbo.Customers SET [Address]={1}, [ContactName]={2}, [CompanyName]={3} WHERE ([CustomerID] = {0})

相应参数："Bruce", "Guangzhou", "CoolCode", "Microsoft"

据不完全统计，实际开发中用的 Update SQL 90%是很简单的，以上扩展基本上符合要求。

方法三、使用 LinqToSQL 自身的解析器来获取Lambda表达式生成的SQL条件语句

该方法与方法二基本上是同一思路，只是在获取Lambda表达式生成的SQL条件上有点不一样。

通过 DataContext 的 GetCommand 可以获取到 DbCommand，所以通过生成的SQL查询语句中截取Where后面的条件，再用方法二生成Update 的赋值语句，两者拼凑起来即可。





该方法比方法二支持更多Lambda表达式（实际上就是所有LinqToSQL支持的）生成SQL条件。

/// <summary>
/// 批量更新
/// </summary>
/// <typeparam name="T"></typeparam>
/// <param name="table">表</param>
/// <param name="predicate">查询条件表达式</param>
/// <param name="updater">更新表达式</param>
/// <returns>影响的行数</returns>
public static int Update<T>(this Table<T> table, Expression<Func<T, bool>> predicate, Expression<Func<T,T>> updater) where T : class{
//获取表名
string tableName = table.Context.Mapping.GetTable(typeof(T)).TableName;
DbCommand command = table.Context.GetCommand(table.Where(predicate));
string sqlCondition = command.CommandText;
sqlCondition = sqlCondition.Substring(sqlCondition.LastIndexOf("WHERE ", StringComparison.InvariantCultureIgnoreCase) + 6);

//获取Update的赋值语句
var updateMemberExpr = (MemberInitExpression)updater.Body;
var updateMemberCollection = updateMemberExpr.Bindings.Cast<MemberAssignment>().Select(c =>
{
var p = command.CreateParameter();
p.ParameterName = c.Member.Name;
p.Value = ((ConstantExpression)c.Expression).Value;
return p;
})
.ToArray();

string sqlUpdateBlock = string.Join(", ", updateMemberCollection.Select(c => string.Format("[{0}]=@{0}", c.ParameterName)).ToArray());

//SQL命令
string commandText = string.Format("UPDATE {0} SET {1} FROM {0} AS t0 WHERE {2}", tableName, sqlUpdateBlock, sqlCondition);

//获取SQL参数数组 (包括查询参数和赋值参数)
command.Parameters.AddRange(updateMemberCollection);
command.CommandText = commandText;

//执行
try
{
if (command.Connection.State != ConnectionState.Open)
{
command.Connection.Open();
}
return command.ExecuteNonQuery();
}
finally
{
command.Connection.Close();
command.Dispose();
}
}

同样使用文章开头的示例，生成的 Update SQL 跟方法二略有不同：

UPDATE dbo.Customers SET [Address]=@Address, [ContactName]=@ContactName, [CompanyName]=@CompanyName FROM dbo.Customers AS t0 WHERE [t0].[CustomerID] = @p0

方法四、支持多表关联的复杂条件

要知道，前面提到的方法二和三都不支持多表关联的复杂条件。可以用一个示例让大家更清楚为什么——

例如，更新CustomerID=“Bruce”的用户的所有订单的送货日前是一个月后。

db.Orders.Update(c => c.Customer.CustomerID == "Bruce",
c => new Order
{
ShippedDate =  DateTime.Now.AddMonths(1)
});

应该生成的 Update SQL 语句是：

UPDATE [dbo].[Orders] SET [ShippedDate] = @p1
FROM [dbo].[Orders] AS [t0]
LEFT OUTER JOIN [dbo].[Customers] AS [t1] ON [t1].[CustomerID] = [t0].[CustomerID]
WHERE [t1].[CustomerID] = @p0
--@p0 = 'Bruce', @p1 = '2010-08-11'

但遗憾的是无论用方法二或三都会抛异常，因为两者皆没法解释多表关联生成的语句： “LEFT OUTER JOIN [dbo].[Customers] AS [t1] ON [t1].[CustomerID] = [t0].[CustomerID] ”

一位叫 Terry Aney 的朋友在《Batch Updates and Deletes with LINQ to SQL》这篇博文中解决了这个问题。使用他提供的UpdateBatch 方法生成的 Update SQL 是：

UPDATE [dbo].[Orders]
SET [ShippedDate] = @p1
FROM [dbo].[Orders] AS j0 INNER JOIN (
SELECT [t0].[OrderID]
FROM [dbo].[Orders] AS [t0]
LEFT OUTER JOIN [dbo].[Customers] AS [t1] ON [t1].[CustomerID] = [t0].[CustomerID]
WHERE [t1].[CustomerID] = @p0
) AS j1 ON (j0.[OrderID] = j1.[OrderID])

-- @p0: Input NVarChar (Size = 5; Prec = 0; Scale = 0) [Bruce]
-- @p1: Input DateTime (Size = 0; Prec = 0; Scale = 0) [2010/8/11 19:51:59]

虽然跟我刚才手写的SQL略有不同，但 Update 的逻辑是对的。有兴趣的朋友不妨试试，Terry Aney在他的文章里有很详尽的介绍，这里不再详述。

相关博文：

Batch Updates and Deletes with LINQ to SQL
LINQ to SQL Batch Updates/Deletes: Fix for 'Could not translate expression'
I've Left Query Analyzer Hell For LINQPad Heaven

总结

Linq To SQL 有很多地方值得探索的，Expression Tree 是探索的基础， 嘿嘿！
完整代码（内含Terry Aney 的代码）
Linq2SQL批量更新.rar