Added support for EF5 queries. Closes #8

Giorgi · Mar 28, 2016 · d2b7858 · d2b7858
1 parent f0b16e3
commit d2b7858
Show file tree

Hide file tree

Showing 3 changed files with 47 additions and 1 deletion.
diff --git a/src/QueryPlanVisualizer/Helpers/DatabaseHelper.cs b/src/QueryPlanVisualizer/Helpers/DatabaseHelper.cs
@@ -1,5 +1,6 @@
 using System;
 using System.Data.Common;
+using System.Data.Entity.Infrastructure;
 using System.Data.Linq;
 using System.Linq;
 using System.Reflection;
@@ -39,6 +40,20 @@ public static DatabaseHelper Create<T>(DataContextBase dataContextBase, IQueryab
                 }
             }
 
+            var query = queryable as DbQuery<T>;
+            if (query != null)
+            {
+                var bindingFlags = BindingFlags.Instance | BindingFlags.NonPublic | BindingFlags.GetProperty;
+
+                var internalQuery = query.GetType().GetProperty("InternalQuery", bindingFlags)?.GetValue(query);
+                var objectQuery = internalQuery?.GetType().GetProperty("ObjectQuery")?.GetValue(internalQuery) as System.Data.Objects.ObjectQuery<T>;
+
+                if (objectQuery != null) //EF5 uses ObjectQuery from System.Data.Objects namespace, EF6 uses System.Data.Entity.Core.Objects so it will be null
+                {
+                    return new EntityFramework5DatabaseHelper(objectQuery);
+                }
+            }
+
             return new EntityFrameworkDatabaseHelper();
         }
 

diff --git a/src/QueryPlanVisualizer/Helpers/EntityFrameworkDatabaseHelper.cs b/src/QueryPlanVisualizer/Helpers/EntityFrameworkDatabaseHelper.cs
@@ -1,10 +1,41 @@
 using System;
 using System.Data.Common;
 using System.Data.Entity.Infrastructure.Interception;
+using System.Data.EntityClient;
+using System.Data.Objects;
 using System.Linq;
 
 namespace ExecutionPlanVisualizer.Helpers
 {
+    class EntityFramework5DatabaseHelper : DatabaseHelper
+    {
+        private ObjectParameterCollection parameters;
+
+        public EntityFramework5DatabaseHelper(ObjectQuery objectQuery)
+        {
+            Connection = (objectQuery.Context.Connection as EntityConnection)?.StoreConnection;
+            parameters = objectQuery.Parameters;
+        }
+
+        protected override DbCommand CreateCommand(IQueryable queryable)
+        {
+            var command = Connection.CreateCommand();
+            command.CommandText = queryable.ToString();
+
+            var copiedParameters = parameters.Select(parameter =>
+                                              {
+                                                  var parameterCopy = command.CreateParameter();
+                                                  parameterCopy.ParameterName = parameter.Name;
+                                                  parameterCopy.Value = parameter.Value;
+                                                  return parameterCopy;
+                                              }).ToArray();
+
+            command.Parameters.AddRange(copiedParameters);
+
+            return command;
+        }
+    }
+
     internal class EntityFrameworkDatabaseHelper : DatabaseHelper
     {
         protected override DbCommand CreateCommand(IQueryable queryable)
@@ -47,7 +78,6 @@ protected override DbCommand CreateCommand(IQueryable queryable)
 
             command.Parameters.AddRange(copiedParameters);
 
-
             return command;
         }
 

diff --git a/src/QueryPlanVisualizer/QueryPlanVisualizer.csproj b/src/QueryPlanVisualizer/QueryPlanVisualizer.csproj
@@ -46,6 +46,7 @@
     <Reference Include="System" />
     <Reference Include="System.ComponentModel.DataAnnotations" />
     <Reference Include="System.Core" />
+    <Reference Include="System.Data.Entity" />
     <Reference Include="System.Data.Linq" />
     <Reference Include="System.Drawing" />
     <Reference Include="System.Windows.Forms" />