From 3b3c82b31116cc55f0e752fe69b15861e8f13d97 Mon Sep 17 00:00:00 2001 From: Boris Kolpackov Date: Tue, 18 Nov 2014 14:53:05 +0200 Subject: Extra result set processing for stored procedure support --- odb/mssql/statement.cxx | 44 ++++++++++++++++++++++++++++++++++++++------ 1 file 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 (cols) == result_count_ + long_count_); -#endif } select_statement::result select_statement:: -- cgit v1.1