fix EsURL parse

jtsunne · Oct 24, 2022 · fc6c00a · fc6c00a
1 parent 9f2c3e3
commit fc6c00a
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/Utils/Common.go b/Utils/Common.go
@@ -3,9 +3,11 @@ package Utils
 import (
 	"encoding/json"
 	"fmt"
+	"net"
 	"net/http"
 	"net/url"
 	"os"
+	"strings"
 	"time"
 )
 
@@ -42,6 +44,10 @@ func ParseEsUrl(u string) string {
 	if s.Port() != "" {
 		port = s.Port()
 	}
-	domain = s.Host
+	if strings.Contains(s.Host, ":") {
+		domain, port, _ = net.SplitHostPort(s.Host)
+	} else {
+		domain = s.Host
+	}
 	return fmt.Sprintf("%s://%s:%s", schema, domain, port)
 }