close tikv#4490: fix transport and change component key

Signed-off-by: Cabinfever_B <cabinfeveroier@gmail.com>
CabinfeverB · Dec 28, 2021 · 746df30 · 746df30
1 parent 5511ebc
commit 746df30
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 2 deletions.
diff --git a/pkg/apiutil/apiutil.go b/pkg/apiutil/apiutil.go
@@ -30,7 +30,7 @@ import (
 var (
 	// ComponentSignatureKey is used for http request header key
 	// to identify component signature
-	ComponentSignatureKey = "ti-component"
+	ComponentSignatureKey = "component"
 	// ComponentAnonymousValue identify anonymous request source
 	ComponentAnonymousValue = "anonymous"
 )

diff --git a/tools/pd-ctl/pdctl/command/global.go b/tools/pd-ctl/pdctl/command/global.go
@@ -30,7 +30,12 @@ import (
 )
 
 var (
-	dialClient = &http.Client{}
+	dialClient = &http.Client{
+		Transport: &apiutil.UserSignatureRoundTripper{
+			Component: "pdctl",
+			Proxied:   http.DefaultTransport,
+		},
+	}
 	pingPrefix = "pd/api/v1/ping"
 )