From 3a27c52317005fcf1eed9c3b28b1cd15d83d73d2 Mon Sep 17 00:00:00 2001 From: mikatong Date: Wed, 2 Nov 2022 16:38:49 +0800 Subject: [PATCH] feat: support set ReqClient from env --- tencentcloud/connectivity/transport.go | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/tencentcloud/connectivity/transport.go b/tencentcloud/connectivity/transport.go index cff0fedbca..dc91c1b419 100644 --- a/tencentcloud/connectivity/transport.go +++ b/tencentcloud/connectivity/transport.go @@ -7,9 +7,12 @@ import ( "io/ioutil" "log" "net/http" + "os" "time" ) +const REQUEST_CLIENT = "TENCENTCLOUD_API_REQUEST_CLIENT" + var ReqClient = "Terraform-latest" func SetReqClient(name string) { @@ -35,6 +38,11 @@ func (me *LogRoundTripper) RoundTrip(request *http.Request) (response *http.Resp return } var headName = "X-TC-Action" + + if envReqClient := os.Getenv(REQUEST_CLIENT); envReqClient != "" { + ReqClient = envReqClient + } + request.Header.Set("X-TC-RequestClient", ReqClient) inBytes = []byte(fmt.Sprintf("%s, request: ", request.Header[headName])) requestBody, errRet := ioutil.ReadAll(bodyReader)