jyapayne/Simple-Web-Server
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added on_upgrade for cases where one wants to handle connection upgrades by another library, for instanace Simple-WebSocket-Server

Browse source
This commit is contained in:
eidheim 2017-01-01 14:34:42 +01:00
commit bfcb325472
1 changed files with 10 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

10
server_http.hpp
View file

@ -147,6 +147,8 @@ namespace SimpleWeb {
std::function<void(std::shared_ptr<typename ServerBase<socket_type>::Response>, std::shared_ptr<typename ServerBase<socket_type>::Request>)> > default_resource; std::function<void(std::shared_ptr<typename ServerBase<socket_type>::Response>, std::shared_ptr<typename ServerBase<socket_type>::Request>)> > default_resource;
std::function<void(std::shared_ptr<typename ServerBase<socket_type>::Request>, const boost::system::error_code&)> on_error; std::function<void(std::shared_ptr<typename ServerBase<socket_type>::Request>, const boost::system::error_code&)> on_error;
std::function<void(std::shared_ptr<socket_type> socket, std::shared_ptr<typename ServerBase<socket_type>::Request>)> on_upgrade;
private: private:
std::vector<std::pair<std::string, std::vector<std::pair<REGEX_NS::regex, std::vector<std::pair<std::string, std::vector<std::pair<REGEX_NS::regex,
@ -356,6 +358,14 @@ namespace SimpleWeb {
} }
void find_resource(const std::shared_ptr<socket_type> &socket, const std::shared_ptr<Request> &request) { void find_resource(const std::shared_ptr<socket_type> &socket, const std::shared_ptr<Request> &request) {
//Upgrade connection
if(on_upgrade) {
auto it_param=request->header.find("Upgrade");
if(it_param!=request->header.end()) {
on_upgrade(socket, request);
return;
}
}
//Find path- and method-match, and call write_response //Find path- and method-match, and call write_response
for(auto& res: opt_resource) { for(auto& res: opt_resource) {
if(request->method==res.first) { if(request->method==res.first) {