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

Server: on session upgrade, remove connection from connections

Browse source
This commit is contained in:
eidheim 2017-07-18 18:25:44 +02:00
commit c21768fae9
1 changed files with 8 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

8
server_http.hpp
View file

@ -538,6 +538,14 @@ namespace SimpleWeb {
if(on_upgrade) { if(on_upgrade) {
auto it = session->request->header.find("Upgrade"); auto it = session->request->header.find("Upgrade");
if(it != session->request->header.end()) { if(it != session->request->header.end()) {
// remove connection from connections
{
std::unique_lock<std::mutex> lock(*connections_mutex);
auto it = connections->find(session->connection.get());
if(it != connections->end())
connections->erase(it);
}
on_upgrade(session->connection->socket, session->request); on_upgrade(session->connection->socket, session->request);
return; return;
} }