From c55e236a61ca3b7adc2d1a14af985914cebbbb11 Mon Sep 17 00:00:00 2001
From: Mitchell McCaffrey <mitchemmc@gmail.com>
Date: Sat, 20 Feb 2021 16:23:37 +1100
Subject: [PATCH] Added socket disconnect on game unmount

---
 src/network/Session.js | 4 ++++
 src/routes/Game.js     | 4 ++++
 2 files changed, 8 insertions(+)

diff --git a/src/network/Session.js b/src/network/Session.js
index c01f5e5..bdde321 100644
--- a/src/network/Session.js
+++ b/src/network/Session.js
@@ -105,6 +105,10 @@ class Session extends EventEmitter {
     }
   }
 
+  disconnect() {
+    this.socket.disconnect();
+  }
+
   /**
    * Send data to a single peer
    *
diff --git a/src/routes/Game.js b/src/routes/Game.js
index 500c05d..210c3fb 100644
--- a/src/routes/Game.js
+++ b/src/routes/Game.js
@@ -33,6 +33,10 @@ function Game() {
       await session.connect();
     }
     connect();
+
+    return () => {
+      session.disconnect();
+    };
   }, [session]);
 
   // Handle session errors