Merge pull request #80 from ceshihao/fix_GetTotalLink

fix GetTotalLink by using SourceAddressList instead of SourceAddress

src/redis-shake/common/common.go

@@ -1,10 +1,10 @@
 package utils
 
 import (
-	"net"
+	"bytes"
 	"fmt"
+	"net"
 	"strings"
-	"bytes"
 
 	"pkg/libs/bytesize"
 	"redis-shake/configure"
@@ -26,7 +26,7 @@ const (
 
 	TencentCluster = "tencent_cluster"
 	AliyunCluster  = "aliyun_cluster"
-	UCloudCluster = "ucloud_cluster"
+	UCloudCluster  = "ucloud_cluster"
 )
 
 var (
@@ -56,7 +56,7 @@ func ReadRESPEnd(c net.Conn) (string, error) {
 func RemoveRESPEnd(input string) string {
 	length := len(input)
 	if length >= 2 {
-		return input[: length - 2]
+		return input[:length-2]
 	}
 	return input
 }
@@ -75,8 +75,8 @@ func ParseInfo(content []byte) map[string]string {
 }
 
 func GetTotalLink() int {
-	if len(conf.Options.SourceAddress) != 0 {
-		return len(conf.Options.SourceAddress)
+	if len(conf.Options.SourceAddressList) != 0 {
+		return len(conf.Options.SourceAddressList)
 	} else {
 		return len(conf.Options.RdbInput)
 	}
@@ -87,4 +87,4 @@ func PickTargetRoundRobin(n int) int {
 		TargetRoundRobin = (TargetRoundRobin + 1) % n
 	}()
 	return TargetRoundRobin
-}
+}