feat(infrastructure): Handle exact match URL too

Dcard · Apr 3, 2024 · c2e763f · c2e763f
1 parent 791b691
commit c2e763f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/internal/infrastructure/manager.go b/internal/infrastructure/manager.go
@@ -80,17 +80,17 @@ func newHTTPClient(config *rest.Config) (*http.Client, error) {
 
 const (
 	gatewayAPIPath        = "/apis/gateway.networking.k8s.io/"
-	gatewayAPIPathV1      = gatewayAPIPath + "v1/"
-	gatewayAPIPathV1Beta1 = gatewayAPIPath + "v1beta1/"
+	gatewayAPIPathV1      = gatewayAPIPath + "v1"
+	gatewayAPIPathV1Beta1 = gatewayAPIPath + "v1beta1"
 )
 
 type rewriteAPIVersion struct {
 	http.RoundTripper
 }
 
 func (r rewriteAPIVersion) RoundTrip(req *http.Request) (*http.Response, error) {
-	if strings.HasPrefix(req.URL.Path, gatewayAPIPathV1) {
-		req.URL.Path = strings.Replace(req.URL.Path, gatewayAPIPathV1, gatewayAPIPathV1Beta1, 1)
+	if req.URL.Path == gatewayAPIPathV1 || strings.HasPrefix(req.URL.Path, gatewayAPIPathV1) {
+		req.URL.Path = gatewayAPIPathV1Beta1 + req.URL.Path[len(gatewayAPIPathV1):]
 	}
 
 	return r.RoundTripper.RoundTrip(req)