diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/Client.hx | 1 | ||||
| -rw-r--r-- | src/server/Main.hx | 15 |
2 files changed, 16 insertions, 0 deletions
diff --git a/src/Client.hx b/src/Client.hx index 708cbaf..c7bc256 100644 --- a/src/Client.hx +++ b/src/Client.hx @@ -23,6 +23,7 @@ class Client { public final ws:WebSocket; public final id:Int; public final req:IncomingMessage; + public var isAlive = true; #end public var name:String; public var group:EnumFlags<ClientGroup>; 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); |