Remove global endPointList to avoid cluster init error

Author: shuwenwei

client/session.go

@@ -21,7 +21,6 @@ package client
 
 import (
 	"bytes"
-	"container/list"
 	"context"
 	"encoding/binary"
 	"errors"
@@ -77,15 +76,14 @@ type Session struct {
 	trans              thrift.TTransport
 	requestStatementId int64
 	protocolFactory    thrift.TProtocolFactory
+	endPointList       []endPoint
 }
 
 type endPoint struct {
 	Host string
 	Port string
 }
 
-var endPointList = list.New()
-
 func (s *Session) Open(enableRPCCompression bool, connectionTimeoutInMs int) error {
 	if s.config.FetchSize <= 0 {
 		s.config.FetchSize = DefaultFetchSize
@@ -1140,11 +1138,14 @@ func NewSession(config *Config) Session {
 }
 
 func newSessionWithSpecifiedSqlDialect(config *Config) Session {
-	endPoint := endPoint{}
-	endPoint.Host = config.Host
-	endPoint.Port = config.Port
-	endPointList.PushBack(endPoint)
-	return Session{config: config}
+	endPointList := []endPoint{{
+		Host: config.Host,
+		Port: config.Port,
+	}}
+	return Session{
+		config:       config,
+		endPointList: endPointList,
+	}
 }
 
 func NewClusterSession(clusterConfig *ClusterConfig) (Session, error) {
@@ -1154,15 +1155,17 @@ func NewClusterSession(clusterConfig *ClusterConfig) (Session, error) {
 
 func newClusterSessionWithSqlDialect(clusterConfig *ClusterConfig) (Session, error) {
 	session := Session{}
-	node := endPoint{}
+	session.endPointList = make([]endPoint, len(clusterConfig.NodeUrls))
 	for i := 0; i < len(clusterConfig.NodeUrls); i++ {
+		node := endPoint{}
 		node.Host = strings.Split(clusterConfig.NodeUrls[i], ":")[0]
 		node.Port = strings.Split(clusterConfig.NodeUrls[i], ":")[1]
-		endPointList.PushBack(node)
+		session.endPointList[i] = node
 	}
 	var err error
-	for e := endPointList.Front(); e != nil; e = e.Next() {
-		session.trans = thrift.NewTSocketConf(net.JoinHostPort(e.Value.(endPoint).Host, e.Value.(endPoint).Port), &thrift.TConfiguration{
+	for i := range session.endPointList {
+		ep := session.endPointList[i]
+		session.trans = thrift.NewTSocketConf(net.JoinHostPort(ep.Host, ep.Port), &thrift.TConfiguration{
 			ConnectTimeout: time.Duration(0), // Use 0 for no timeout
 		})
 		// session.trans = thrift.NewTFramedTransport(session.trans)	// deprecated
@@ -1173,7 +1176,7 @@ func newClusterSessionWithSqlDialect(clusterConfig *ClusterConfig) (Session, err
 			if err != nil {
 				log.Println(err)
 			} else {
-				session.config = getConfig(e.Value.(endPoint).Host, e.Value.(endPoint).Port,
+				session.config = getConfig(ep.Host, ep.Port,
 					clusterConfig.UserName, clusterConfig.Password, clusterConfig.FetchSize, clusterConfig.TimeZone, clusterConfig.ConnectRetryMax, clusterConfig.Database, clusterConfig.sqlDialect)
 				break
 			}
@@ -1250,13 +1253,14 @@ func (s *Session) reconnect() bool {
 	var connectedSuccess = false
 
 	for i := 0; i < s.config.ConnectRetryMax; i++ {
-		for e := endPointList.Front(); e != nil; e = e.Next() {
-			err = s.initClusterConn(e.Value.(endPoint))
+		for i := range s.endPointList {
+			ep := s.endPointList[i]
+			err = s.initClusterConn(ep)
 			if err == nil {
 				connectedSuccess = true
 				break
 			} else {
-				log.Println("Connection refused:", e.Value.(endPoint))
+				log.Println("Connection refused:", ep)
 			}
 		}
 		if connectedSuccess {

test/e2e/e2e_test.go

@@ -75,6 +75,16 @@ func (s *e2eTestSuite) checkError(status *common.TSStatus, err error) {
 	}
 }
 
+func (s *e2eTestSuite) Test_WrongURL() {
+	clusterConfig := client.ClusterConfig{
+		NodeUrls: strings.Split("iotdb1:6667", ","),
+		UserName: "root",
+		Password: "root",
+	}
+	_, err := client.NewClusterSession(&clusterConfig)
+	s.Require().Error(err)
+}
+
 func (s *e2eTestSuite) Test_CreateTimeseries() {
 	var (
 		path       = "root.tsg1.dev1.status"