You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@iotdb.apache.org by ha...@apache.org on 2023/06/28 07:39:08 UTC

[iotdb-client-go] 01/01: fix StatementId doesn't exist in this session issue

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

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

commit e116b5bd2ac922a78426c0c9a8004bd9aaae4912
Author: HTHou <hh...@outlook.com>
AuthorDate: Wed Jun 28 15:38:55 2023 +0800

    fix StatementId doesn't exist in this session issue
---
 client/session.go | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/client/session.go b/client/session.go
index e0173c6..1d1ee4c 100644
--- a/client/session.go
+++ b/client/session.go
@@ -431,6 +431,7 @@ func (s *Session) ExecuteStatement(sql string) (*SessionDataSet, error) {
 	if err != nil && resp == nil {
 		if s.reconnect() {
 			request.SessionId = s.sessionId
+			request.StatementId = s.requestStatementId
 			resp, err = s.client.ExecuteStatement(context.Background(), &request)
 		}
 	}
@@ -450,6 +451,7 @@ func (s *Session) ExecuteNonQueryStatement(sql string) (r *rpc.TSStatus, err err
 	if err != nil && resp == nil {
 		if s.reconnect() {
 			request.SessionId = s.sessionId
+			request.StatementId = s.requestStatementId
 			resp, err = s.client.ExecuteStatement(context.Background(), &request)
 		}
 	}
@@ -469,6 +471,7 @@ func (s *Session) ExecuteQueryStatement(sql string, timeoutMs *int64) (*SessionD
 	} else {
 		if s.reconnect() {
 			request.SessionId = s.sessionId
+			request.StatementId = s.requestStatementId
 			resp, err = s.client.ExecuteQueryStatement(context.Background(), &request)
 			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
@@ -769,6 +772,7 @@ func (s *Session) ExecuteRawDataQuery(paths []string, startTime int64, endTime i
 	if err != nil && resp == nil {
 		if s.reconnect() {
 			request.SessionId = s.sessionId
+			request.StatementId = s.requestStatementId
 			resp, err = s.client.ExecuteRawDataQuery(context.Background(), &request)
 		}
 	}
@@ -788,6 +792,7 @@ func (s *Session) ExecuteUpdateStatement(sql string) (*SessionDataSet, error) {
 	if err != nil && resp == nil {
 		if s.reconnect() {
 			request.SessionId = s.sessionId
+			request.StatementId = s.requestStatementId
 			resp, err = s.client.ExecuteUpdateStatement(context.Background(), &request)
 		}
 	}