From 807e3a9d7e28e92249713653db32cb34b5850b09 Mon Sep 17 00:00:00 2001
From: Aziz Khoury <bentael@gmail.com>
Date: Fri, 15 Apr 2016 16:45:04 -0400
Subject: [PATCH] adding cls back in socketio index

---
 src/socket.io/index.js | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/src/socket.io/index.js b/src/socket.io/index.js
index 809f21ea11..419b083478 100644
--- a/src/socket.io/index.js
+++ b/src/socket.io/index.js
@@ -10,6 +10,7 @@ var winston = require('winston');
 var db = require('../database');
 var logger = require('../logger');
 var ratelimit = require('../middleware/ratelimit');
+var cls = require('../middleware/cls');
 
 var Sockets = {};
 var Namespaces = {};
@@ -29,6 +30,7 @@ Sockets.init = function(server) {
 	io.use(authorize);
 
 	io.on('connection', onConnection);
+	io.on('disconnect', onDisconnect);
 
 	io.listen(server, {
 		transports: nconf.get('socket.io:transports')
@@ -42,10 +44,14 @@ function onConnection(socket) {
 
 	logger.io_one(socket, socket.uid);
 
-	onConnect(socket);
+	cls.socket(socket, null, 'connection', function() {
+		onConnect(socket);
+	});
 
 	socket.on('*', function(payload) {
-		onMessage(socket, payload);
+		cls.socket(socket, payload, null, function() {
+			onMessage(socket, payload);
+		});
 	});
 }
 
@@ -58,6 +64,10 @@ function onConnect(socket) {
 	}
 }
 
+function onDisconnect(socket) {
+	cls.socket(socket, null, 'disconnect', function() {});
+}
+
 
 function onMessage(socket, payload) {
 	if (!payload.data.length) {