diff --git a/pkg/velaql/providers/query/handler_test.go b/pkg/velaql/providers/query/handler_test.go index bb32b26be..4e19101ea 100644 --- a/pkg/velaql/providers/query/handler_test.go +++ b/pkg/velaql/providers/query/handler_test.go @@ -960,12 +960,12 @@ options: { fmt.Sprintf("http://%s:30229", gatewayIP), "http://10.10.10.10", "http://text.example.com", - "tcp://10.10.10.10:81", - "tcp://text.example.com:81", + "10.10.10.10:81", + "text.example.com:81", // helmRelease fmt.Sprintf("http://%s:30002", gatewayIP), "http://ingress.domain.helm", - "tcp://1.1.1.1:80/seldon/test", + "1.1.1.1:80/seldon/test", } endValue, err := v.Field("list") Expect(err).Should(BeNil()) diff --git a/pkg/velaql/providers/query/types/type.go b/pkg/velaql/providers/query/types/type.go index b4fa61eca..22b0b4ddd 100644 --- a/pkg/velaql/providers/query/types/type.go +++ b/pkg/velaql/providers/query/types/type.go @@ -58,6 +58,9 @@ func (s *ServiceEndpoint) String() string { if (protocol == HTTPS && s.Endpoint.Port == 443) || (protocol == HTTP && s.Endpoint.Port == 80) { return fmt.Sprintf("%s://%s%s", protocol, s.Endpoint.Host, path) } + if protocol == "tcp" { + return fmt.Sprintf("%s:%d%s", s.Endpoint.Host, s.Endpoint.Port, path) + } return fmt.Sprintf("%s://%s:%d%s", protocol, s.Endpoint.Host, s.Endpoint.Port, path) } diff --git a/references/cli/velaql_test.go b/references/cli/velaql_test.go index 0a9f2d3a7..d81b98ed6 100644 --- a/references/cli/velaql_test.go +++ b/references/cli/velaql_test.go @@ -393,8 +393,8 @@ var _ = Describe("Test velaQL", func() { fmt.Sprintf("http://%s:30229", gatewayIP), "http://10.10.10.10", "http://text.example.com", - "tcp://10.10.10.10:81", - "tcp://text.example.com:81", + "10.10.10.10:81", + "text.example.com:81", // helmRelease fmt.Sprintf("http://%s:30002", gatewayIP), "http://ingress.domain.helm",