Extra result set processing for stored procedure support

1 files changed, 38 insertions, 6 deletions

diff --git a/odb/mssql/statement.cxx b/odb/mssql/statement.cxx
index 15037a0..00180ee 100644
--- a/odb/mssql/statement.cxx
+++ b/odb/mssql/statement.cxx
@@ -839,23 +839,55 @@ namespace odb
     {
       SQLRETURN r (statement::execute ());
 
-      if (!SQL_SUCCEEDED (r))
-        translate_error (r, conn_, stmt_);
+      // Skip empty result sets that seem to be added as a result of
+      // executing DML statements in stored procedures (e.g., INSERT
+      // INTO EXEC).
+      //
+      if (r == SQL_NO_DATA)
+      {
+        r = SQLMoreResults (stmt_);
 
-#ifndef NDEBUG
-      SQLSMALLINT cols;
-      r = SQLNumResultCols (stmt_, &cols);
+        if (r == SQL_NO_DATA)
+        {
+          throw database_exception (
+            0,
+            "?????",
+            "another result set expected after SQL_NO_DATA");
+        }
+      }
 
       if (!SQL_SUCCEEDED (r))
         translate_error (r, conn_, stmt_);
 
+      // Skip result sets that have no columns. These seem to be added
+      // by DML statements that don't produce any result (e.g., EXEC).
+      //
+      SQLSMALLINT cols (0);
+
+      while (cols == 0)
+      {
+        r = SQLNumResultCols (stmt_, &cols);
+
+        if (!SQL_SUCCEEDED (r))
+          translate_error (r, conn_, stmt_);
+
+        if (cols == 0)
+        {
+          r = SQLMoreResults (stmt_);
+
+          if (r == SQL_NO_DATA)
+            break;
+          else if (!SQL_SUCCEEDED (r))
+            translate_error (r, conn_, stmt_);
+        }
+      }
+
       // Make sure that the number of columns in the result returned by
       // the database matches the number that we expect. A common cause
       // of this assertion is a native view with a number of data members
       // not matching the number of columns in the SELECT-list.
       //
       assert (static_cast<SQLUSMALLINT> (cols) == result_count_ + long_count_);
-#endif
     }
 
     select_statement::result select_statement::