cleanup broken streams

WardCunningham · Dec 29, 2012 · 04b5d65 · 04b5d65
1 parent 5b3f72d
commit 04b5d65
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 4 deletions.
diff --git a/client/plugins/logwatch/logwatch.coffee b/client/plugins/logwatch/logwatch.coffee
@@ -11,7 +11,7 @@ window.plugins.logwatch =
 
     socket.onmessage = (e) ->
       msg = JSON.parse e.data
-      print wiki.resolveLinks("[[#{msg.title}]]")
+      print wiki.resolveLinks("[[#{msg.title}]] #{msg.listeners}")
 
     socket.onclose = ->
       print "WebSocket Connection Closed."
diff --git a/client/plugins/logwatch/logwatch.js b/client/plugins/logwatch/logwatch.js
diff --git a/server/express/lib/server.coffee b/server/express/lib/server.coffee
@@ -101,13 +101,15 @@ module.exports = exports = (argv) ->
     )
   )
   logWatchSocket.on('connection', (ws) ->
-    logWatchSocket.on('fetch', (page) ->
+    logWatchSocket.on('fetch', fetchListener = (page) ->
       reference =
         title: page.title
+        listeners: logWatchSocket.listeners('fetch').length
       ws.send JSON.stringify(reference), (e) ->
         if e
           log 'unable to send ws message: ', e
-          return
+          logWatchSocket.removeListener 'fetch', fetchListener
+          ws.close()
     )
     ws.on('message', (message) ->
       log 'logWatch message from client:', message