Update using-sqlgetdiagrec-and-sqlgetdiagfield.md

v-chojas · web-flow · commit bec34917ec90 · 2018-02-15T21:38:48.000Z
diff --git a/docs/odbc/reference/develop-app/using-sqlgetdiagrec-and-sqlgetdiagfield.md b/docs/odbc/reference/develop-app/using-sqlgetdiagrec-and-sqlgetdiagfield.md
@@ -46,10 +46,12 @@ GetSQLStmt(SQLStmt);
   
 // Execute the SQL statement and return any errors or warnings.  
 rc1 = SQLExecDirect(hstmt, SQLStmt, SQL_NTS);  
-if ((rc1 == SQL_SUCCESS_WITH_INFO) || (rc1 == SQL_ERROR)) {  
-   // Get the status records.  
+if ((rc1 == SQL_SUCCESS_WITH_INFO) || (rc1 == SQL_ERROR)) {
+   SQLLEN numRecs = 0;
+   SQLGetDiagField(SQL_HANDLE_STMT, hstmt, 0, SQL_DIAG_NUMBER, &numRecs, 0, 0);
+   // Get the status records.
    i = 1;  
-   while ((rc2 = SQLGetDiagRec(SQL_HANDLE_STMT, hstmt, i, SqlState, &NativeError,  
+   while (i <= numRecs && (rc2 = SQLGetDiagRec(SQL_HANDLE_STMT, hstmt, i, SqlState, &NativeError,  
             Msg, sizeof(Msg), &MsgLen)) != SQL_NO_DATA) {  
       DisplayError(SqlState,NativeError,Msg,MsgLen);  
       i++;  
