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

#58)
sigmacomputing · Mar 12, 2024 · 2db3d43 · 2db3d43
1 parent 611dc5e
commit 2db3d43
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/monitoring.go b/monitoring.go
@@ -239,14 +239,16 @@ func (sc *snowflakeConn) rowsForRunningQuery(
 		return err
 	}
 	if !resp.Success {
+		message := resp.Message
 		code, err := strconv.Atoi(resp.Code)
 		if err != nil {
-			return err
+			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:  resp.Message,
+			Message:  message,
 			QueryID:  resp.Data.QueryID,
 		}).exceptionTelemetry(sc)
 	}