Add IsInMemoryQuery + Fix Take

Add IsInMemoryQuery + Fix Take

Author: zzzprojects

src/shared/Z.EF.Plus.BatchUpdate.Shared/BatchUpdate.cs

@@ -15,6 +15,7 @@
 #if EF5
 using System.Data.Objects;
 using System.Data.SqlClient;
+using System.Reflection;
 using Z.EntityFramework.Plus.Internal.Core.SchemaObjectModel;
 
 #elif EF6
@@ -192,6 +193,30 @@ public int Execute<T>(IQueryable<T> query, Expression<Func<T, T>> updateFactory)
             }
 
 #if EF5 || EF6
+            if (BatchUpdateManager.IsInMemoryQuery)
+            {
+                var list = query.ToList();
+                var compiled = updateFactory.Compile();
+                var memberBindings = ((MemberInitExpression)updateFactory.Body).Bindings;
+                var accessors = memberBindings
+                    .Select(x => x.Member.Name)
+                    .Select(x => new PropertyOrFieldAccessor(typeof(T).GetProperty(x, BindingFlags.Public | BindingFlags.NonPublic | BindingFlags.Instance)))
+                    .ToList();
+
+                foreach (var item in list)
+                {
+                    var newItem = compiled(item);
+
+                    foreach (var accessor in accessors)
+                    {
+                        var value = accessor.GetValue(newItem);
+                        accessor.SetValue(item, value);
+                    }
+                }
+
+                return list.Count;
+            }
+
             var dbContext = query.GetDbContext();
 
 #if EF6
@@ -994,6 +1019,11 @@ public List<Tuple<string, object>> GetInnerValues<T>(IQueryable<T> query, Expres
                     valueSql = valueSql.Replace("AS [C1]", "");
                     valueSql = valueSql.Replace("AS `C1`", "");
 
+                    if (valueSql.Trim().StartsWith("TOP") && valueSql.IndexOf(")") != -1)
+                    {
+                        valueSql = valueSql.Substring(valueSql.IndexOf(")") + 1);
+                    }
+
                     var listReplace = new List<string>()
                     {
                         "[Extent1]",

src/shared/Z.EF.Plus.BatchUpdate.Shared/BatchUpdateManager.cs

@@ -25,6 +25,12 @@ public class BatchUpdateManager
         /// <summary>Gets or sets the factory to create an InMemory DbContext.</summary>
         /// <value>The factory to create an InMemory DbContext.</value>
         public static Func<DbContext> InMemoryDbContextFactory { get; set; }
+#else
+        /// <summary>
+        /// Gets or sets a value indicating whether this object is in memory query.
+        /// </summary>
+        /// <value>True if this object is in memory query, false if not.</value>
+        public static bool IsInMemoryQuery { get; set; }
 #endif
     }
 }