From d7f985a33bbeec8e304e668ac91884891980fa5f Mon Sep 17 00:00:00 2001
From: wb-hx510875 <wb-hx510875@alibaba-inc.com>
Date: Mon, 14 Sep 2020 13:51:10 +0800
Subject: [PATCH] Adding some request header when constructing request

---
 core/src/RpcServiceRequest.cc     | 1 +
 core/src/ServiceRequest.cc        | 4 +++-
 test/core/rpcservicerequest_ut.cc | 1 +
 test/core/servicerequest_ut.cc    | 1 +
 4 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/core/src/RpcServiceRequest.cc b/core/src/RpcServiceRequest.cc
index faf05b286..e49faf211 100644
--- a/core/src/RpcServiceRequest.cc
+++ b/core/src/RpcServiceRequest.cc
@@ -23,6 +23,7 @@ RpcServiceRequest::RpcServiceRequest(const std::string &product,
                                      const std::string &action)
     : ServiceRequest(product, version) {
   setActionName(action);
+  setHeader(std::string("x-acs-action"), action);
 }
 
 RpcServiceRequest::~RpcServiceRequest() {}
diff --git a/core/src/ServiceRequest.cc b/core/src/ServiceRequest.cc
index 9b06757f3..c6c79afe5 100644
--- a/core/src/ServiceRequest.cc
+++ b/core/src/ServiceRequest.cc
@@ -25,7 +25,9 @@ ServiceRequest::ServiceRequest(const std::string &product,
     : content_(nullptr), contentSize_(0), params_(), product_(product),
       resourcePath_("/"), version_(version), scheme_("https"),
       connectTimeout_(kInvalidTimeout), readTimeout_(kInvalidTimeout),
-      method_(HttpRequest::Method::Get) {}
+      method_(HttpRequest::Method::Get) {
+  setHeader(std::string("x-acs-version"), version);
+}
 
 ServiceRequest::ServiceRequest(const ServiceRequest &other)
     : content_(nullptr), contentSize_(other.contentSize_),
diff --git a/test/core/rpcservicerequest_ut.cc b/test/core/rpcservicerequest_ut.cc
index eaaa949e2..c979e9c3c 100644
--- a/test/core/rpcservicerequest_ut.cc
+++ b/test/core/rpcservicerequest_ut.cc
@@ -15,4 +15,5 @@ TEST(test_RpcServiceRequest, basic) {
   EXPECT_TRUE(rpc.actionName() == action);
   EXPECT_TRUE(rpc.product() == product);
   EXPECT_TRUE(rpc.version() == version);
+  EXPECT_EQ(string("testAction"), rpc.getHeader(string("x-acs-action")));
 }
diff --git a/test/core/servicerequest_ut.cc b/test/core/servicerequest_ut.cc
index 242290831..0fdf111cb 100644
--- a/test/core/servicerequest_ut.cc
+++ b/test/core/servicerequest_ut.cc
@@ -40,6 +40,7 @@ namespace {
 
     TestServiceRequest sr1("ECS", "1.0");
     TestServiceRequest sr2(sr1);
+    EXPECT_EQ(string("1.0"), sr1.getHeader(string("x-acs-version")));
 
     sr1.addParameter("k1", "v1");
     sr2.addParameter("k2", "v2");