fix: panic on query canceled (#168)

Author: everpcpc

VERSION

@@ -1 +1 @@
-0.8.2
+0.8.3

client.go

@@ -414,10 +414,8 @@ func (c *APIClient) applySessionState(response *QueryResponse) {
 }
 
 func (c *APIClient) PollUntilQueryEnd(ctx context.Context, resp *QueryResponse) (*QueryResponse, error) {
-	var err error
 	for !resp.ReadFinished() {
-		data := resp.Data
-		resp, err = c.PollQuery(ctx, resp.NextURI)
+		nextResponse, err := c.PollQuery(ctx, resp.NextURI)
 		if err != nil {
 			if errors.Is(err, context.Canceled) {
 				// context might be canceled due to timeout or canceled. if it's canceled, we need call
@@ -427,6 +425,8 @@ func (c *APIClient) PollUntilQueryEnd(ctx context.Context, resp *QueryResponse)
 			}
 			return nil, err
 		}
+		data := resp.Data
+		resp = nextResponse
 		if resp.Error != nil {
 			return nil, errors.Wrap(resp.Error, "query page has error")
 		}

rows.go

@@ -30,9 +30,8 @@ func waitForData(ctx context.Context, dc *DatabendConn, response *QueryResponse)
 	if response.Error != nil {
 		return nil, response.Error
 	}
-	var err error
 	for !response.ReadFinished() && len(response.Data) == 0 && response.Error == nil {
-		response, err = dc.rest.PollQuery(ctx, response.NextURI)
+		nextResponse, err := dc.rest.PollQuery(ctx, response.NextURI)
 		if err != nil {
 			if errors.Is(err, context.Canceled) {
 				// context might be canceled due to timeout or canceled. if it's canceled, we need call
@@ -44,6 +43,7 @@ func waitForData(ctx context.Context, dc *DatabendConn, response *QueryResponse)
 			}
 			return nil, err
 		}
+		response = nextResponse
 		if response.Error != nil {
 			_ = dc.rest.CloseQuery(ctx, response)
 			return nil, fmt.Errorf("query error: %+v", response.Error)