From: Steinar H. Gunderson <sgunderson@bigfoot.com>
Date: Tue, 11 Nov 2014 20:08:32 +0000 (+0100)
Subject: Remove a client from the sleeping list immediately if it closes its socket prematurely.
X-Git-Url: https://git.sesse.net/?p=remoteglot;a=commitdiff_plain;h=425bc089202a6160970805df00dbc76b2f74b43d;hp=405e4d1cabe8e21b8bc6291f4258da35d51f5958

Remove a client from the sleeping list immediately if it closes its socket prematurely.
---

diff --git a/www/serve-analysis.js b/www/serve-analysis.js
index 5778dc9..13ba06f 100644
--- a/www/serve-analysis.js
+++ b/www/serve-analysis.js
@@ -128,7 +128,8 @@ var count_viewers = function() {
 fs.watch(path.dirname(json_filename), reread_file);
 reread_file(null, path.basename(json_filename));
 
-http.createServer(function(request, response) {
+var server = http.createServer();
+server.on('request', function(request, response) {
 	var u = url.parse(request.url, true);
 	var ims = (u.query)['ims'];
 	var unique = (u.query)['unique'];
@@ -168,4 +169,16 @@ http.createServer(function(request, response) {
 	client.accept_gzip = accept_gzip;
 	client.unique = unique;
 	sleeping_clients[request_id++] = client;
-}).listen(5000);
+
+	request.socket.client = client;
+});
+server.on('connection', function(socket) {
+	socket.on('close', function() {
+		var client = socket.client;
+		if (client) {
+			mark_recently_seen(client.unique);
+			delete sleeping_clients[client.request_id];
+		}
+	});
+});
+server.listen(5000);