diff --git a/response.cpp b/response.cpp index ccd5dd6..12342d7 100644 --- a/response.cpp +++ b/response.cpp @@ -18,6 +18,10 @@ void Response::SetPayload(std::vector data) { m_payload = data; } +void Response::SetStatusCode(statuscode::statusCode statuscode) { + m_statusCode = statuscode; +} + void Response::SetPayload(std::string data) { m_headers.insert(std::pair( "content-length", std::to_string(std::strlen(data.data())))); diff --git a/response.hpp b/response.hpp index 5210ae2..402bbea 100644 --- a/response.hpp +++ b/response.hpp @@ -20,6 +20,7 @@ public: void SetPayload(std::vector data); void SetPayload(std::string data); void SetContentType(const std::string type); + void SetStatusCode(const statuscode::statusCode statuscode); void Send(int clientSocket); void Print(); };