chore(endpoint): Refactor Type for simplification and better performance

test result

cpu: AMD Ryzen 7 4800H with Radeon Graphics
BenchmarkEndpoint_Type-16

Cut + switch     5053628 238.7 ns/op 0 B/op 0 allocs/op
Cut + map        2714998 438.6 ns/op 0 B/op 0 allocs/op
switch HasPerfix 2683519 430.9 ns/op 0 B/op 0 allocs/op

Author: qianlongzt

config/endpoint/endpoint.go

@@ -161,28 +161,33 @@ func (e *Endpoint) IsEnabled() bool {
 
 // Type returns the endpoint type
 func (e *Endpoint) Type() Type {
-	switch {
-	case e.DNSConfig != nil:
+	if e.DNSConfig != nil {
 		return TypeDNS
-	case strings.HasPrefix(e.URL, "tcp://"):
+	}
+	before, _, ok := strings.Cut(e.URL, ":")
+	if !ok {
+		return TypeUNKNOWN
+	}
+	switch before {
+	case "tcp":
 		return TypeTCP
-	case strings.HasPrefix(e.URL, "sctp://"):
+	case "sctp":
 		return TypeSCTP
-	case strings.HasPrefix(e.URL, "udp://"):
+	case "udp":
 		return TypeUDP
-	case strings.HasPrefix(e.URL, "icmp://"):
+	case "icmp":
 		return TypeICMP
-	case strings.HasPrefix(e.URL, "starttls://"):
+	case "starttls":
 		return TypeSTARTTLS
-	case strings.HasPrefix(e.URL, "tls://"):
+	case "tls":
 		return TypeTLS
-	case strings.HasPrefix(e.URL, "http://") || strings.HasPrefix(e.URL, "https://"):
+	case "http", "https":
 		return TypeHTTP
-	case strings.HasPrefix(e.URL, "ws://") || strings.HasPrefix(e.URL, "wss://"):
+	case "ws", "wss":
 		return TypeWS
-	case strings.HasPrefix(e.URL, "ssh://"):
+	case "ssh":
 		return TypeSSH
-	case strings.HasPrefix(e.URL, "domain://"):
+	case "domain":
 		return TypeDomain
 	default:
 		return TypeUNKNOWN

config/endpoint/endpoint_test.go

@@ -301,116 +301,112 @@ func TestEndpoint_IsEnabled(t *testing.T) {
 	}
 }
 
+type testEndpoint_typeArgs struct {
+	URL string
+	DNS *dns.Config
+	SSH *ssh.Config
+}
+
+var testEndpoint_typeData = []struct {
+	args testEndpoint_typeArgs
+	want Type
+}{
+	{
+		args: testEndpoint_typeArgs{
+			URL: "8.8.8.8",
+			DNS: &dns.Config{
+				QueryType: "A",
+				QueryName: "example.com",
+			},
+		},
+		want: TypeDNS,
+	},
+	{
+		args: testEndpoint_typeArgs{
+			URL: "tcp://127.0.0.1:6379",
+		},
+		want: TypeTCP,
+	},
+	{
+		args: testEndpoint_typeArgs{
+			URL: "icmp://example.com",
+		},
+		want: TypeICMP,
+	},
+	{
+		args: testEndpoint_typeArgs{
+			URL: "sctp://example.com",
+		},
+		want: TypeSCTP,
+	},
+	{
+		args: testEndpoint_typeArgs{
+			URL: "udp://example.com",
+		},
+		want: TypeUDP,
+	},
+	{
+		args: testEndpoint_typeArgs{
+			URL: "starttls://smtp.gmail.com:587",
+		},
+		want: TypeSTARTTLS,
+	},
+	{
+		args: testEndpoint_typeArgs{
+			URL: "tls://example.com:443",
+		},
+		want: TypeTLS,
+	},
+	{
+		args: testEndpoint_typeArgs{
+			URL: "https://twin.sh/health",
+		},
+		want: TypeHTTP,
+	},
+	{
+		args: testEndpoint_typeArgs{
+			URL: "wss://example.com/",
+		},
+		want: TypeWS,
+	},
+	{
+		args: testEndpoint_typeArgs{
+			URL: "ws://example.com/",
+		},
+		want: TypeWS,
+	},
+	{
+		args: testEndpoint_typeArgs{
+			URL: "ssh://example.com:22",
+			SSH: &ssh.Config{
+				Username: "root",
+				Password: "password",
+			},
+		},
+		want: TypeSSH,
+	},
+	{
+		args: testEndpoint_typeArgs{
+			URL: "domain://example.org",
+		},
+		want: TypeDomain,
+	},
+	{
+		args: testEndpoint_typeArgs{
+			URL: "invalid://example.org",
+		},
+		want: TypeUNKNOWN,
+	},
+	{
+		args: testEndpoint_typeArgs{
+			URL: "no-scheme",
+		},
+		want: TypeUNKNOWN,
+	},
+}
+
 func TestEndpoint_Type(t *testing.T) {
-	type args struct {
-		URL string
-		DNS *dns.Config
-		SSH *ssh.Config
-	}
-	tests := []struct {
-		args args
-		want Type
-	}{
-		{
-			args: args{
-				URL: "8.8.8.8",
-				DNS: &dns.Config{
-					QueryType: "A",
-					QueryName: "example.com",
-				},
-			},
-			want: TypeDNS,
-		},
-		{
-			args: args{
-				URL: "tcp://127.0.0.1:6379",
-			},
-			want: TypeTCP,
-		},
-		{
-			args: args{
-				URL: "icmp://example.com",
-			},
-			want: TypeICMP,
-		},
-		{
-			args: args{
-				URL: "sctp://example.com",
-			},
-			want: TypeSCTP,
-		},
-		{
-			args: args{
-				URL: "udp://example.com",
-			},
-			want: TypeUDP,
-		},
-		{
-			args: args{
-				URL: "starttls://smtp.gmail.com:587",
-			},
-			want: TypeSTARTTLS,
-		},
-		{
-			args: args{
-				URL: "tls://example.com:443",
-			},
-			want: TypeTLS,
-		},
-		{
-			args: args{
-				URL: "https://twin.sh/health",
-			},
-			want: TypeHTTP,
-		},
-		{
-			args: args{
-				URL: "wss://example.com/",
-			},
-			want: TypeWS,
-		},
-		{
-			args: args{
-				URL: "ws://example.com/",
-			},
-			want: TypeWS,
-		},
-		{
-			args: args{
-				URL: "ssh://example.com:22",
-				SSH: &ssh.Config{
-					Username: "root",
-					Password: "password",
-				},
-			},
-			want: TypeSSH,
-		},
-		{
-			args: args{
-				URL: "domain://example.org",
-			},
-			want: TypeDomain,
-		},
-		{
-			args: args{
-				URL: "domain://example.org",
-			},
-			want: TypeDomain,
-		},
-		{
-			args: args{
-				URL: "invalid://example.org",
-			},
-			want: TypeUNKNOWN,
-		},
-		{
-			args: args{
-				URL: "no-scheme",
-			},
-			want: TypeUNKNOWN,
-		},
-	}
-	for _, tt := range tests {
+	for _, tt := range testEndpoint_typeData {
 		t.Run(string(tt.want), func(t *testing.T) {
 			endpoint := Endpoint{
 				URL:       tt.args.URL,
@@ -423,6 +419,20 @@ func TestEndpoint_Type(t *testing.T) {
 	}
 }
 
+func BenchmarkEndpoint_Type(b *testing.B) {
+	for b.Loop() {
+		for _, tt := range testEndpoint_typeData {
+			endpoint := Endpoint{
+				URL:       tt.args.URL,
+				DNSConfig: tt.args.DNS,
+			}
+			if got := endpoint.Type(); got != tt.want {
+				b.Errorf("Endpoint.Type() = %v, want %v", got, tt.want)
+			}
+		}
+	}
+}
+
 func TestEndpoint_ValidateAndSetDefaults(t *testing.T) {
 	endpoint := Endpoint{
 		Name:               "website-health",