From a96b85fa5b3be30cdf5447c8367cf86a2daddf86 Mon Sep 17 00:00:00 2001
From: calzoneman <calzoneman@gmail.com>
Date: Mon, 7 Mar 2016 20:25:32 -0800
Subject: [PATCH] proxyinterceptor: handle SocketDisconnectEvent

---
 src/backend/proxyinterceptor.js | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/backend/proxyinterceptor.js b/src/backend/proxyinterceptor.js
index 2f67e0fe..95119525 100644
--- a/src/backend/proxyinterceptor.js
+++ b/src/backend/proxyinterceptor.js
@@ -24,6 +24,7 @@ export default class ProxyInterceptor {
         socket.on('close', this.onFrontendDisconnect.bind(this, socket));
         socket.on('SocketConnectEvent', this.onSocketConnect.bind(this, socket));
         socket.on('SocketFrameEvent', this.onSocketFrame.bind(this, socket));
+        socket.on('SocketDisconnectEvent', this.onSocketDisconnect.bind(this, socket));
     }
 
     onFrontendDisconnect(socket) {
@@ -73,4 +74,17 @@ export default class ProxyInterceptor {
         const socket = socketMap[socketID];
         socket.onProxiedEventReceived.apply(socket, [event].concat(args));
     }
+
+    onSocketDisconnect(frontendConnection, socketID) {
+        const mapKey = frontendConnection.endpoint;
+        const socketMap = this.frontendProxiedSockets[mapKey];
+        if (!socketMap || !socketMap.hasOwnProperty(socketID)) {
+            logger.error(`Received SocketDisconnectEvent for nonexistent socket`,
+                    { socketID });
+            return;
+        }
+
+        const socket = socketMap[socketID];
+        socket.onProxiedEventReceived.apply(socket, ['disconnect']);
+    }
 }