From f70382462d69db15e11e642b5f8c5baa9357133a Mon Sep 17 00:00:00 2001
From: RblSb <msrblsb@gmail.com>
Date: Wed, 3 Jun 2020 14:56:44 +0300
Subject: Detect broken connections

---
 src/server/Main.hx | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

(limited to 'src/server')

diff --git a/src/server/Main.hx b/src/server/Main.hx
index 6c7753c..ee4d478 100644
--- a/src/server/Main.hx
+++ b/src/server/Main.hx
@@ -101,6 +101,20 @@ class Main {
 		server.listen(port);
 		wss = new WSServer({server: server});
 		wss.on("connection", onConnect);
+
+		new Timer(25000).run = () -> {
+			for (client in clients) {
+				if (client.isAlive) {
+					client.isAlive = false;
+					client.ws.ping();
+					continue;
+				}
+				client.ws.terminate();
+				onMessage(client, {
+					type: Disconnected
+				}, true);
+			}
+		};
 	}
 
 	public function exit():Void {
@@ -276,6 +290,7 @@ class Main {
 		final client = new Client(ws, req, id, name, 0);
 		client.isAdmin = isAdmin;
 		clients.push(client);
+		ws.on("pong", () -> client.isAlive = true);
 		onMessage(client, {
 			type: Connected
 		}, true);
-- 
cgit v1.2.3