fix check health for tcp/udp, refactor

agent/pkg/handler/check_health.go

@@ -1,9 +1,11 @@
 package handler
 
 import (
+	"fmt"
 	"net/http"
 	"net/url"
 	"os"
+	"strings"
 
 	apiUtils "github.com/yusing/go-proxy/internal/api/v1/utils"
 	"github.com/yusing/go-proxy/internal/net/types"
@@ -51,6 +53,14 @@ func CheckHealth(w http.ResponseWriter, r *http.Request) {
 			http.Error(w, http.StatusText(http.StatusBadRequest), http.StatusBadRequest)
 			return
 		}
+		hasPort := strings.Contains(host, ":")
+		port := query.Get("port")
+		if port != "" && !hasPort {
+			host = fmt.Sprintf("%s:%s", host, port)
+		} else {
+			http.Error(w, http.StatusText(http.StatusBadRequest), http.StatusBadRequest)
+			return
+		}
 		result, err = monitor.NewRawHealthChecker(types.NewURL(&url.URL{
 			Scheme: scheme,
 			Host:   host,

agent/pkg/handler/check_health_test.go

@@ -198,7 +198,7 @@ func TestCheckHealthTCPUDP(t *testing.T) {
 	for _, tt := range tests {
 		t.Run(tt.name, func(t *testing.T) {
 			query := url.Values{}
-			query.Set("scheme", "tcp")
+			query.Set("scheme", tt.scheme)
 			query.Set("host", tt.host)
 			query.Set("port", strconv.Itoa(tt.port))