diff --git a/include/fc/network/http/websocket.hpp b/include/fc/network/http/websocket.hpp index 210211e..f56b4be 100644 --- a/include/fc/network/http/websocket.hpp +++ b/include/fc/network/http/websocket.hpp @@ -29,6 +29,8 @@ namespace fc { namespace http { void set_session_data( fc::any d ){ _session_data = std::move(d); } fc::any& get_session_data() { return _session_data; } + virtual std::string get_request_header(const std::string& key) = 0; + fc::signal closed; private: fc::any _session_data; diff --git a/src/network/http/websocket.cpp b/src/network/http/websocket.cpp index 1f92f77..7f26b4d 100644 --- a/src/network/http/websocket.cpp +++ b/src/network/http/websocket.cpp @@ -168,6 +168,11 @@ namespace fc { namespace http { _ws_connection->close(code,reason); } + virtual std::string get_request_header(const std::string& key)override + { + return _ws_connection->get_request_header(key); + } + T _ws_connection; };