modify type of endpointList

Author: shuwenwei

client/session.go

@@ -21,7 +21,6 @@ package client
 
 import (
 	"bytes"
-	"container/list"
 	"context"
 	"encoding/binary"
 	"errors"
@@ -77,7 +76,7 @@ type Session struct {
 	trans              thrift.TTransport
 	requestStatementId int64
 	protocolFactory    thrift.TProtocolFactory
-	endPointList       *list.List
+	endPointList       []endPoint
 }
 
 type endPoint struct {
@@ -1139,11 +1138,10 @@ func NewSession(config *Config) Session {
 }
 
 func newSessionWithSpecifiedSqlDialect(config *Config) Session {
-	endPointList := list.New()
-	endPoint := endPoint{}
-	endPoint.Host = config.Host
-	endPoint.Port = config.Port
-	endPointList.PushBack(endPoint)
+	endPointList := []endPoint{{
+		Host: config.Host,
+		Port: config.Port,
+	}}
 	return Session{
 		config:       config,
 		endPointList: endPointList,
@@ -1157,16 +1155,17 @@ func NewClusterSession(clusterConfig *ClusterConfig) (Session, error) {
 
 func newClusterSessionWithSqlDialect(clusterConfig *ClusterConfig) (Session, error) {
 	session := Session{}
-	node := endPoint{}
-	session.endPointList = list.New()
+	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]
-		session.endPointList.PushBack(node)
+		session.endPointList[i] = node
 	}
 	var err error
-	for e := session.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
@@ -1177,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
 			}
@@ -1254,13 +1253,14 @@ func (s *Session) reconnect() bool {
 	var connectedSuccess = false
 
 	for i := 0; i < s.config.ConnectRetryMax; i++ {
-		for e := s.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 {