From c21768fae92e1ca6bf2c3ec23f8700af52b8b356 Mon Sep 17 00:00:00 2001
From: eidheim <eidheim@gmail.com>
Date: Tue, 18 Jul 2017 18:25:44 +0200
Subject: [PATCH] Server: on session upgrade, remove connection from
 connections

---
 server_http.hpp | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/server_http.hpp b/server_http.hpp
index 0a83c9b..48b223e 100644
--- a/server_http.hpp
+++ b/server_http.hpp
@@ -538,6 +538,14 @@ namespace SimpleWeb {
       if(on_upgrade) {
         auto it = session->request->header.find("Upgrade");
         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);
           return;
         }