SNOW-502598: Refactor getQueryStatusThrowIfError

snowflakedb · Oct 31, 2023 · 357359a · 357359a
1 parent 6c00000
commit 357359a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/lib/connection/connection.js b/lib/connection/connection.js
@@ -487,17 +487,17 @@ function Connection(context)
    * @returns {String} the query status.
    */
   this.getQueryStatusThrowIfError = async function (queryId) {
-    const response = await getQueryResponse(queryId);
-    const queries = response['data']['queries'];
-    const status = extractQueryStatus(response);
+    const status = this.getQueryStatus(queryId);
+
     let message, code, sqlState = null;
 
     if (this.isAnError(status)) {
+      const response = await getQueryResponse(queryId);
       message = response['message'] || '';
       code = response['code'] || -1;
 
       if (response['data']) {
-        message += queries.length > 0 ? queries[0]['errorMessage'] : '';
+        message += response['data']['queries'].length > 0 ? response['data']['queries'][0]['errorMessage'] : '';
         sqlState = response['data']['sqlState'];
       }