[iotdb-client-go] branch main updated: fix the panic caused by the status is error

[chaow]

This is an automated email from the ASF dual-hosted git repository.

chaow pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/iotdb-client-go.git


The following commit(s) were added to refs/heads/main by this push:
     new 810fc54  fix the panic caused by the status is error
     new 1e7fa88  Merge pull request #20 from neuyilan/fix_panic
810fc54 is described below

commit 810fc54e4762ca04dd49cccaac4e3c6deed48d09
Author: HouliangQi <neuyi...@163.com>
AuthorDate: Wed May 19 11:55:04 2021 +0800

    fix the panic caused by the status is error
---
 client/session.go | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/client/session.go b/client/session.go
index 794c028..8b99965 100644
--- a/client/session.go
+++ b/client/session.go
@@ -274,7 +274,11 @@ func (s *Session) ExecuteQueryStatement(sql string, 
timeoutMs int64) (*SessionDa
        request := rpc.TSExecuteStatementReq{SessionId: s.sessionId, Statement: 
sql, StatementId: s.requestStatementId,
                FetchSize: &s.config.FetchSize, Timeout: &timeoutMs}
        if resp, err := s.client.ExecuteQueryStatement(context.Background(), 
&request); err == nil {
-               return NewSessionDataSet(sql, resp.Columns, resp.DataTypeList, 
resp.ColumnNameIndexMap, *resp.QueryId, s.client, s.sessionId, 
resp.QueryDataSet, resp.IgnoreTimeStamp != nil && *resp.IgnoreTimeStamp, 
s.config.FetchSize, timeoutMs), err
+               if statusErr := VerifySuccess(resp.Status); statusErr == nil {
+                       return NewSessionDataSet(sql, resp.Columns, 
resp.DataTypeList, resp.ColumnNameIndexMap, *resp.QueryId, s.client, 
s.sessionId, resp.QueryDataSet, resp.IgnoreTimeStamp != nil && 
*resp.IgnoreTimeStamp, s.config.FetchSize, timeoutMs), err
+               } else {
+                       return nil, statusErr
+               }
        } else {
                return nil, err
        }