From 36bf5d1c032b7324fb8bc380f0b2e8d24c06ac95 Mon Sep 17 00:00:00 2001
From: JCash <mwesterdahl76@gmail.com>
Date: Sun, 27 Sep 2020 17:04:00 +0200
Subject: [PATCH] code cleanup

---
 websocket/src/websocket.cpp | 8 +++-----
 1 file changed, 3 insertions(+), 5 deletions(-)

diff --git a/websocket/src/websocket.cpp b/websocket/src/websocket.cpp
index a5ac74e..30215a2 100644
--- a/websocket/src/websocket.cpp
+++ b/websocket/src/websocket.cpp
@@ -504,13 +504,11 @@ static dmExtension::Result OnUpdate(dmExtension::Params* params)
             if (RESULT_OK != conn->m_Status)
             {
                 HandleCallback(conn, EVENT_ERROR, 0, conn->m_BufferSize);
-                HandleCallback(conn, EVENT_DISCONNECTED, 0, 0);
-            }
-            else
-            {
-                HandleCallback(conn, EVENT_DISCONNECTED, 0, conn->m_BufferSize);
+                conn->m_BufferSize = 0;
             }
 
+            HandleCallback(conn, EVENT_DISCONNECTED, 0, conn->m_BufferSize);
+
             g_Websocket.m_Connections.EraseSwap(i);
             --i;
             --size;