From a47889c03ef51bc3e1a41108755a2caf447eb5d8 Mon Sep 17 00:00:00 2001
From: dpomier <dimitri.pomier@gmail.com>
Date: Fri, 30 Apr 2021 15:22:44 +0200
Subject: [PATCH] fix ws disconnection

---
 com/smartfoxserver/v2/bitswarm/BitSwarmClient.hx | 9 ++++-----
 1 file changed, 4 insertions(+), 5 deletions(-)

diff --git a/com/smartfoxserver/v2/bitswarm/BitSwarmClient.hx b/com/smartfoxserver/v2/bitswarm/BitSwarmClient.hx
index 1b4dc76..eff0d02 100644
--- a/com/smartfoxserver/v2/bitswarm/BitSwarmClient.hx
+++ b/com/smartfoxserver/v2/bitswarm/BitSwarmClient.hx
@@ -369,11 +369,10 @@ class BitSwarmClient extends EventDispatcher
 	{
 		if(_useBlueBox)
 			_bbClient.close();
-		else
-		{
-			if(socket.connected)
-				_socket.close();
-		}
+		else if(_wsClient.connected)
+			_wsClient.close();
+		else if(_socket.connected)
+			_socket.close();
 				
 		onSocketClose(new BitSwarmEvent(BitSwarmEvent.DISCONNECT, { reason:reason } ));
 	}