Issue zzzprojects#11: Added support for returning a DBDataReader base…

…d on xavierjohn's suggestion
jamesra · Apr 2, 2016 · 22888dd · 22888dd
1 parent 960f9b2
commit 22888dd
Show file tree

Hide file tree

Showing 3 changed files with 63 additions and 0 deletions.
diff --git a/EntityFrameworkExtras.EF5/DatabaseExtensions.cs b/EntityFrameworkExtras.EF5/DatabaseExtensions.cs
@@ -52,6 +52,27 @@ public static IEnumerable<T> ExecuteStoredProcedure<T>(this Database database, o
             return result;
         }
 
+        /// <summary>
+        /// Executes the specified stored procedure against a database
+        /// and returns a DbDataReader.  Used for procedures returning multiple result sets.
+        /// </summary>
+        /// <param name="database">The database to execute against.</param>
+        /// <param name="storedProcedure">The stored procedure to execute.</param>
+        /// <returns></returns>
+        public static System.Data.Common.DbDataReader ExecuteReader(this Database database, object storedProcedure)
+        {
+            if (storedProcedure == null)
+                throw new ArgumentNullException("storedProcedure");
+
+            var info = StoredProcedureParser.BuildStoredProcedureInfo(storedProcedure);
+
+            var cmd = database.Connection.CreateCommand();
+            cmd.CommandText = info.Sql;
+            cmd.Parameters.AddRange(info.SqlParameters);
+
+            return cmd.ExecuteReader();
+        }
+
         private static void SetOutputParameterValues(IEnumerable<SqlParameter> sqlParameters, object storedProcedure)
         {
             foreach (SqlParameter sqlParameter in sqlParameters.Where(p => p.Direction != ParameterDirection.Input))

diff --git a/EntityFrameworkExtras.EF6/DatabaseExtensions.cs b/EntityFrameworkExtras.EF6/DatabaseExtensions.cs
@@ -52,6 +52,27 @@ public static IEnumerable<T> ExecuteStoredProcedure<T>(this Database database, o
             return result;
         }
 
+        /// <summary>
+        /// Executes the specified stored procedure against a database
+        /// and returns a DbDataReader.  Used for procedures returning multiple result sets.
+        /// </summary>
+        /// <param name="database">The database to execute against.</param>
+        /// <param name="storedProcedure">The stored procedure to execute.</param>
+        /// <returns></returns>
+        public static System.Data.Common.DbDataReader ExecuteReader(this Database database, object storedProcedure)
+        {
+            if (storedProcedure == null)
+                throw new ArgumentNullException("storedProcedure");
+
+            var info = StoredProcedureParser.BuildStoredProcedureInfo(storedProcedure);            
+
+            var cmd = database.Connection.CreateCommand();
+            cmd.CommandText = info.Sql;
+            cmd.Parameters.AddRange(info.SqlParameters);    
+
+            return cmd.ExecuteReader();
+        }
+
         private static void SetOutputParameterValues(IEnumerable<SqlParameter> sqlParameters, object storedProcedure)
         {
             foreach (SqlParameter sqlParameter in sqlParameters.Where(p => p.Direction != ParameterDirection.Input))

diff --git a/EntityFrameworkExtras/DatabaseExtensions.cs b/EntityFrameworkExtras/DatabaseExtensions.cs
@@ -52,6 +52,27 @@ public static IEnumerable<T> ExecuteStoredProcedure<T>(this Database database, o
             return result;
         }
 
+        /// <summary>
+        /// Executes the specified stored procedure against a database
+        /// and returns a DbDataReader.  Used for procedures returning multiple result sets.
+        /// </summary>
+        /// <param name="database">The database to execute against.</param>
+        /// <param name="storedProcedure">The stored procedure to execute.</param>
+        /// <returns></returns>
+        public static System.Data.Common.DbDataReader ExecuteReader(this Database database, object storedProcedure)
+        {
+            if (storedProcedure == null)
+                throw new ArgumentNullException("storedProcedure");
+
+            var info = StoredProcedureParser.BuildStoredProcedureInfo(storedProcedure);
+
+            var cmd = database.Connection.CreateCommand();
+            cmd.CommandText = info.Sql;
+            cmd.Parameters.AddRange(info.SqlParameters);
+
+            return cmd.ExecuteReader();
+        }
+
         private static void SetOutputParameterValues(IEnumerable<SqlParameter> sqlParameters, object storedProcedure)
         {
             foreach (SqlParameter sqlParameter in sqlParameters.Where(p => p.Direction != ParameterDirection.Input))