From aed35f5b068c5798e55440348b0d85f9cf9b4f72 Mon Sep 17 00:00:00 2001 From: elmato Date: Mon, 16 Jan 2017 19:55:17 +0000 Subject: [PATCH] Add access to HTTP request headers in websocket_connection --- include/fc/network/http/websocket.hpp | 2 ++ src/network/http/websocket.cpp | 5 +++++ 2 files changed, 7 insertions(+) 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; };