Nemcio/NodeBB
1
0
Fork 0
mirror of https://github.com/NodeBB/NodeBB.git synced 2025-11-06 14:05:46 +01:00
Code Issues Packages Projects Releases Wiki Activity

adding cls support for ws

Browse Source
This commit is contained in:
akhoury
2016-02-12 12:20:21 -05:00
parent 852a1a178e
commit 4245cb2739
3 changed files with 27 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

15
src/socket.io/index.js
View File

@@ -11,6 +11,7 @@ var SocketIO = require('socket.io'),
user = require('../user'),
logger = require('../logger'),
ratelimit = require('../middleware/ratelimit'),
cls = require('../middleware/cls'),
Sockets = {},
Namespaces = {};
@@ -43,14 +44,20 @@ function onConnection(socket) {
logger.io_one(socket, socket.uid);
onConnect(socket);
cls.socket(socket, null, 'connection', function () {
onConnect(socket);
});
socket.on('disconnect', function(data) {
onDisconnect(socket, data);
socket.on('disconnect', function(payload) {
cls.socket(socket, payload, 'disconnect', function () {
onDisconnect(socket, payload);
});
});
socket.on('*', function(payload) {
onMessage(socket, payload);
cls.socket(socket, payload, null, function() {
onMessage(socket, payload);
});
});
}