[Fix] SIG-18794: Fix handling of results response for canceled queries (

#58)
sigmacomputing · Feb 8, 2023 · a94f204 · a94f204
1 parent fbac64f
commit a94f204
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/monitoring.go b/monitoring.go
@@ -251,6 +251,20 @@ func (sc *snowflakeConn) rowsForRunningQuery(
 		}
 		return err
 	}
+	if !resp.Success {
+		message := resp.Message
+		code, err := strconv.Atoi(resp.Code)
+		if err != nil {
+			code = ErrQueryStatus
+			message = fmt.Sprintf("%s: (failed to parse original code: %s: %s)", message, resp.Code, err.Error())
+		}
+		return (&SnowflakeError{
+			Number:   code,
+			SQLState: resp.Data.SQLState,
+			Message:  message,
+			QueryID:  resp.Data.QueryID,
+		}).exceptionTelemetry(sc)
+	}
 	rows.addDownloader(populateChunkDownloader(ctx, sc, resp.Data))
 	return nil
 }