最近搞Linq To Sql（以下简称LS）上瘾了，越玩越好玩，废话不多说。切入正题批更新和批删除早就被作为LS的软肋被广大程序员们嗤之以鼻，园子里也有人做了些扩展来满足批操作需求，我本来打算直接COPY这些代码在项目中使用的，遗憾的是我没找到批更新的，批删除的有几种，但我都觉得要么太取巧，要又是不完善，我这里也献丑一下，做个批操作扩展，既然大家都做了批删除，我挑个冷门的，批更新吧，其实删除和更新都差不多吧。
      我们想要的其实很简单，UPDATE [TABLE] SET ..... WHERE TABLE.XXX = XXX....，尴尬的是这样的语句在LS面前几乎成了奢求，Attach方法勉强可以用用，但是每次重新生成LS实体都要重新设置非条件属性的 UpdateCheck = UpdateChek.Never让人无法接受，况且他也就只能做做 WHERE TABLE.XXX = XXX，如果我需要 WHERE TABLE.XXX = TABLE.XXX != XXX或者其他更复杂的条件时这种方法就显得很无力了，扩展，自我扩展，这似乎是我最近在LS娱乐城里玩的最多的游戏了，今天我就要实现这么一个调用 Table.Update(Expression> predicate, Expression> evaluator)：
      还是得说说JeffreyZhao(他动作太快了，每次我想写的内容一到他博客里看都是成年旧醋了), 他已经做过批更新的，具体可看这里，遗憾的是如他所说：“放弃了对于复杂表达式树的解析，不支持item.Introduction.Length < 10这种条件”，另有BLACK JACK通过正则式匹配查询语句，可看这里，但是我对那个正则表达式不是很放心，毕竟SQL的变化太多。结合了前面两种做法，我也提出我的思路，我们知道IQueryable的查询语句是可以获得的，那么我们能否在批操作时候直接运用这个SQL呢？答案是肯定的，例如我有这么一个SELECT的SQL(OrderId是主键)：
                           SELECT * from Order WHERE Order.OrderId = 123;
     那么可以想办法更新这些查询结果中的记录，SQL语句可以这么写：

                         UPDATE [Order]
                          SET [XXX] = xxxx

FROM [dbo].[Order ] AS T1 INNER JOIN (

                          SELECT *
                          FROM [Order]
                          WHERE [Order].[OrderId ] = 123
                         ) AS T2 ON (T1.[OrderId] = T2.[OrderId])

很显然，子查询里的语句是直接获取的，这样我们的任务就变成构造SET [XXX] = xxxx....的字符串和join语句了，SET后面会说，这里直接给JOIN语句的构造代码:

private static string GetJoinCondition<TEntity>(this Table<TEntity> table)

where TEntity : class

{

MetaTable metaTable = table.Context.Mapping.GetTable(typeof(TEntity));

var keys = from member in metaTable.RowType.DataMembers

where member.IsPrimaryKey

select new { member.MappedName };

StringBuilder joinCondition = new StringBuilder();

foreach (var key in keys)

{

joinCondition.AppendFormat("T1.[{0}] = T2.[{0}] AND ", key.MappedName);

}

string join = joinCondition.Remove(0, joinCondition.Length - 5); //"remove and string"

return join;

}

private static string GetTableName<TEntity>(this Table<TEntity> table) where TEntity : class

{

var entityType = typeof(TEntity);

var metaTable = table.Context.Mapping.GetTable(entityType);

string name = metaTable.TableName;

string[] arr = name.Split('.');

foreach (string str in arr)

{

name = name.Replace(str, String.Format("[{0}]", str));

}

return name;

}

private static string GetFormatCmdText<TEntity>(Table<TEntity> table, IQueryable<TEntity> entities ) where TEntity : class

{

string joinCondition = table.GetJoinCondition();

DbCommand selectCommand = table.Context.GetCommand( entities );

string selectSql = selectCommand.CommandText;

string join = String.Format("FROM {0} AS T1 INNER JOIN (\r\n\r\n{1}\r\n\r\n) AS T2 ON ({2})\r\n", table.GetTableName(), selectSql, joinCondition);

return "UPDATE " + table.GetTableName() + " SET {0}" + join;

}

       按照上面的举例，直接调用上面的GetFormatCmdText扩展方法我们将得到如下SQL：

                         UPDATE [Order]
                          SET {0}

FROM [dbo].[Order ] AS T1 INNER JOIN (

                          SELECT *
                          FROM [Order]
                          WHERE [Order].[OrderId ] = 123
                         ) AS T2 ON (T1.[OrderId] = T2.[OrderId])

     现在的问题就是有SET后的赋值SQL串了，这也是批更新的难点，表达式树的解析很复杂，所幸的是有这么一个老外的博客提供了现成的代码给我们用：

    http://blogs.msdn.com/mattwar/archive/2007/07/31/linq-building-an-iqueryable-provider-part-ii.aspx

     里面有ExpressionVisitor类可供使用，代码我就不贴出来了，需要的朋友可浏览上面的链接。通过代码我们可以获取赋值表达式的MemberInitExpression对象，把它转换成SET的SQL语句片段也很简单：

private static string GetDbSetString<TEntity>( MemberInitExpression memberInitExpression, Table<TEntity> table, DbCommand updateCommand ) where TEntity : class

{

Type entityType = typeof(TEntity);

if (memberInitExpression.Type != entityType)

{

throw new NotImplementedException( string.Format( "the expression should be '{0}'.", entityType ) );

}

StringBuilder setSB = new StringBuilder();

string tableName = table.GetDbName();

MetaTable metaTable = table.Context.Mapping.GetTable( entityType );

var cols = from member in metaTable.RowType.DataMembers

select new { member.MappedName };

foreach ( var binding in memberInitExpression.Bindings )

{

MemberAssignment assignment = binding as MemberAssignment;

if ( binding == null )

{

throw new NotImplementedException( String.Format("the property '{0}' bindings is not a type MemberAssignment.", binding.Member.Name) );

}

ParameterExpression entityParam = null;

ExpressionVisitor<ParameterExpression>.Visit(assignment.Expression, p =>

{

if (p.Type == entityType)

entityParam = p;

return p;

}

);

string name = binding.Member.Name;

var col = ( from c in cols

where c.MappedName == name

select c ).FirstOrDefault();

if ( col == null )

{

throw new ArgumentOutOfRangeException(name, string.Format("not found the field on the {0} table", tableName));

}

if (entityParam == null)

{

object constant = Expression.Lambda(assignment.Expression, null).Compile().DynamicInvoke();

if (constant == null)

{

setSB.AppendFormat("[{0}] = NULL, ", col.MappedName);
点击图片可在新窗口打

来自 “ ITPUB博客 ” ，链接：https://blog.itpub.net/552044/viewspace-434514/ ，如需转载，请注明出处，否则将追究法律责任。

上一篇： asp.net修改网页扩展名

下一篇：扩展LINQ to SQL：使用Lambda Expression批量删除数据

请登录后发表评论登录

全部评论

ITPub博客

老调新弹，也玩Linq To Sql批操作

最新文章