fix: there is no errors when sql is incorrectErr/check

Co-authored-by: rick <[email protected]>

Author: LinuxSuRen

client/session.go

@@ -25,14 +25,15 @@ import (
 	"encoding/binary"
 	"errors"
 	"fmt"
-	"github.com/apache/iotdb-client-go/common"
 	"log"
 	"net"
 	"reflect"
 	"sort"
 	"strings"
 	"time"
 
+	"github.com/apache/iotdb-client-go/common"
+
 	"github.com/apache/iotdb-client-go/rpc"
 	"github.com/apache/thrift/lib/go/thrift"
 )
@@ -415,26 +416,34 @@ func (s *Session) SetTimeZone(timeZone string) (r *common.TSStatus, err error) {
 	return r, err
 }
 
-func (s *Session) ExecuteStatement(sql string) (*SessionDataSet, error) {
+func (s *Session) ExecuteStatementWithContext(ctx context.Context, sql string) (*SessionDataSet, error) {
 	request := rpc.TSExecuteStatementReq{
 		SessionId:   s.sessionId,
 		Statement:   sql,
 		StatementId: s.requestStatementId,
 		FetchSize:   &s.config.FetchSize,
 	}
-	resp, err := s.client.ExecuteStatement(context.Background(), &request)
+	resp, err := s.client.ExecuteStatement(ctx, &request)
 
 	if err != nil && resp == nil {
 		if s.reconnect() {
 			request.SessionId = s.sessionId
 			request.StatementId = s.requestStatementId
-			resp, err = s.client.ExecuteStatement(context.Background(), &request)
+			resp, err = s.client.ExecuteStatement(ctx, &request)
 		}
 	}
 
+	if statusErr := VerifySuccess(resp.Status); statusErr != nil {
+		return nil, statusErr
+	}
+
 	return s.genDataSet(sql, resp), err
 }
 
+func (s *Session) ExecuteStatement(sql string) (*SessionDataSet, error) {
+	return s.ExecuteStatementWithContext(context.Background(), sql)
+}
+
 func (s *Session) ExecuteNonQueryStatement(sql string) (r *common.TSStatus, err error) {
 	request := rpc.TSExecuteStatementReq{
 		SessionId:   s.sessionId,

test/e2e/e2e_test.go

@@ -20,14 +20,16 @@
 package e2e
 
 import (
+	"context"
 	"fmt"
-	"github.com/apache/iotdb-client-go/common"
 	"log"
 	"math/rand"
 	"strings"
 	"testing"
 	"time"
 
+	"github.com/apache/iotdb-client-go/common"
+
 	"github.com/apache/iotdb-client-go/client"
 	"github.com/stretchr/testify/suite"
 )
@@ -392,3 +394,9 @@ func (s *e2eTestSuite) Test_InsertAlignedTablets() {
 	assert.Equal(status, "8")
 	s.session.DeleteStorageGroup("root.ln.**")
 }
+
+func (s *e2eTestSuite) Test_InvalidSQL() {
+	_, err := s.session.ExecuteStatementWithContext(context.Background(), "select1 from device")
+	assert := s.Require()
+	assert.Error(err)
+}