Allow to AddOrReplace headers from THttpResponse

1 files changed, 11 insertions, 0 deletions

diff --git a/library/cpp/http/server/response.h b/library/cpp/http/server/response.h
index a75cb85605..8b7819de1e 100644
--- a/library/cpp/http/server/response.h
+++ b/library/cpp/http/server/response.h
@@ -32,6 +32,17 @@ public:
         return *this;
     }
 
+    template <typename ValueType>
+    THttpResponse& AddOrReplaceHeader(const TString& name, const ValueType& value) {
+        return AddOrReplaceHeader(THttpInputHeader(name, ToString(value)));
+    }
+
+    THttpResponse& AddOrReplaceHeader(const THttpInputHeader& header) {
+        Headers.AddOrReplaceHeader(header);
+
+        return *this;
+    }
+
     THttpResponse& AddMultipleHeaders(const THttpHeaders& headers);
 
     const THttpHeaders& GetHeaders() const {