#2486

various fixes for socket.io cluster
display user presence correctly

src/socket.io/index.js

@@ -12,6 +12,7 @@ var	SocketIO = require('socket.io'),
 	user = require('../user'),
 	logger = require('../logger'),
 	ratelimit = require('../middleware/ratelimit'),
+	rooms = require('./rooms'),
 
 	Sockets = {},
 	Namespaces = {};
@@ -63,8 +64,8 @@ function onConnection(socket) {
 
 function onConnect(socket) {
 	if (socket.uid) {
-		socket.join('uid_' + socket.uid);
-		socket.join('online_users');
+		rooms.enter(socket, 'uid_' + socket.uid);
+		rooms.enter(socket, 'online_users');
 
 		user.getUserFields(socket.uid, ['status'], function(err, userData) {
 			if (err || !userData) {
@@ -77,7 +78,7 @@ function onConnect(socket) {
 			}
 		});
 	} else {
-		socket.join('online_guests');
+		rooms.enter(socket, 'online_guests');
 		socket.emit('event:connect');
 	}
 }
@@ -85,7 +86,7 @@ function onConnect(socket) {
 function onDisconnect(socket, data) {
 	if (socket.uid) {
 		var socketCount = Sockets.getUserSocketCount(socket.uid);
-		if (socketCount <= 0) {
+		if (socketCount <= 1) {
 			socket.broadcast.emit('event:user_status_change', {uid: socket.uid, status: 'offline'});
 		}
 
@@ -96,6 +97,7 @@ function onDisconnect(socket, data) {
 			}
 		});
 	}
+	rooms.leaveAll(socket, data.rooms);
 }
 
 function onMessage(socket, payload) {
@@ -183,7 +185,7 @@ function authorize(socket, callback) {
 					socket.uid = parseInt(sessionData.passport.user, 10);
 				} else {
 					socket.uid = 0;
-				}	
+				}
 				next();
 			});
 		}
@@ -218,27 +220,25 @@ Sockets.in = function(room) {
 };
 
 Sockets.getSocketCount = function() {
-	// TODO: io.sockets.adapter.sids is local to this worker
-	// use redis-adapter
-
-	var clients = Object.keys(io.sockets.adapter.sids || {});
-	return Array.isArray(clients) ? clients.length : 0;
+	return rooms.socketCount();
 };
 
 Sockets.getUserSocketCount = function(uid) {
-	// TODO: io.sockets.adapter.rooms is local to this worker
-	// use .clients('uid_' + uid, fn)
+	return rooms.clients('uid_' + uid).length;
+};
 
-	var roomClients = Object.keys(io.sockets.adapter.rooms['uid_' + uid] || {});
-	return Array.isArray(roomClients) ? roomClients.length : 0;
+Sockets.getOnlineUserCount = function() {
+	var count = 0;
+	Object.keys(rooms.roomClients()).forEach(function(roomName) {
+		if (roomName.startsWith('uid_')) {
+			++ count;
+		}
+	});
+	return count;
 };
 
 Sockets.getOnlineAnonCount = function () {
-	// TODO: io.sockets.adapter.rooms is local to this worker
-	// use .clients()
-
-	var guestSocketIds = Object.keys(io.sockets.adapter.rooms.online_guests || {});
-	return Array.isArray(guestSocketIds) ? guestSocketIds.length : 0;
+	return rooms.clients('online_guests').length;
 };
 
 Sockets.reqFromSocket = function(socket) {
@@ -258,9 +258,7 @@ Sockets.reqFromSocket = function(socket) {
 };
 
 Sockets.isUserOnline = function(uid) {
-	// TODO: io.sockets.adapter.rooms is local to this worker
-	// use .clients('uid_' + uid, fn)
-	return io ? !!io.sockets.adapter.rooms['uid_' + uid] : false;
+	return !!rooms.clients('uid_' + uid).length;
 };
 
 Sockets.isUsersOnline = function(uids, callback) {
@@ -301,26 +299,29 @@ Sockets.getUsersInRoom = function (uid, roomName, callback) {
 
 Sockets.getUidsInRoom = function(roomName, callback) {
 	callback = callback || function() {};
-	// TODO : doesnt work in cluster
 
 	var uids = [];
 
-	var socketids = Object.keys(io.sockets.adapter.rooms[roomName] || {});
+	var socketids = rooms.clients(roomName);
 	if (!Array.isArray(socketids) || !socketids.length) {
 		callback(null, []);
 		return [];
 	}
 
 	for(var i=0; i<socketids.length; ++i) {
-		var socketRooms = Object.keys(io.sockets.adapter.sids[socketids[i]]);
+		var socketRooms = rooms.clientRooms(socketids[i]);
 		if (Array.isArray(socketRooms)) {
 			socketRooms.forEach(function(roomName) {
-				if (roomName.indexOf('uid_') === 0 ) {
-					uids.push(roomName.split('_')[1]);
+				if (roomName.startsWith('uid_')) {
+					var uid = roomName.split('_')[1];
+					if (uids.indexOf(uid) === -1) {
+						uids.push(uid);
+					}
 				}
 			});
 		}
 	}
+
 	callback(null, uids);
 	return uids;
 };

src/socket.io/meta.js

@@ -12,6 +12,7 @@ var	nconf = require('nconf'),
 	logger = require('../logger'),
 	plugins = require('../plugins'),
 	emitter = require('../emitter'),
+	rooms = require('./rooms'),
 
 	websockets = require('./'),
 
@@ -58,7 +59,7 @@ SocketMeta.rooms.enter = function(socket, data, callback) {
 	}
 
 	if (socket.currentRoom) {
-		socket.leave(socket.currentRoom);
+		rooms.leave(socket, socket.currentRoom);
 		if (socket.currentRoom.indexOf('topic') !== -1) {
 			websockets.in(socket.currentRoom).emit('event:user_leave', socket.uid);
 		}
@@ -66,7 +67,7 @@ SocketMeta.rooms.enter = function(socket, data, callback) {
 	}
 
 	if (data.enter) {
-		socket.join(data.enter);
+		rooms.enter(socket, data.enter);
 		socket.currentRoom = data.enter;
 		if (data.enter.indexOf('topic') !== -1) {
 			data.uid = socket.uid;
@@ -80,69 +81,64 @@ SocketMeta.rooms.enter = function(socket, data, callback) {
 };
 
 SocketMeta.rooms.getAll = function(socket, data, callback) {
-	var now = Date.now();
-	db.sortedSetCount('users:online', now - 300000, now, function(err, onlineRegisteredCount) {
+	var roomClients = rooms.roomClients();
+	var socketData = {
+			onlineGuestCount: websockets.getOnlineAnonCount(),
+			onlineRegisteredCount: websockets.getOnlineUserCount(),
+			socketCount: websockets.getSocketCount(),
+			users: {
+				categories: roomClients.categories ? roomClients.categories.length : 0,
+				topics: 0,
+				category: 0
+			},
+			topics: {}
+		};
+
+	var scores = {},
+		topTenTopics = [],
+		tid;
+
+	for (var room in roomClients) {
+		if (roomClients.hasOwnProperty(room)) {
+			tid = room.match(/^topic_(\d+)/);
+			if (tid) {
+				var length = roomClients[room].length;
+				socketData.users.topics += length;
+
+				if (scores[length]) {
+					scores[length].push(tid[1]);
+				} else {
+					scores[length] = [tid[1]];
+				}
+			} else if (room.match(/^category/)) {
+				socketData.users.category += roomClients[room].length;
+			}
+		}
+	}
+
+	var scoreKeys = Object.keys(scores),
+		mostActive = scoreKeys.sort();
+
+	while(topTenTopics.length < 10 && mostActive.length > 0) {
+		topTenTopics = topTenTopics.concat(scores[mostActive.pop()]);
+	}
+
+	topTenTopics = topTenTopics.slice(0, 10);
+
+	topics.getTopicsFields(topTenTopics, ['title'], function(err, titles) {
 		if (err) {
 			return callback(err);
 		}
-
-		var rooms = {}; // TODO: websockets.server.sockets.manager.rooms; doesnt work in socket.io 1.x
-		var socketData = {
-				onlineGuestCount: websockets.getOnlineAnonCount(),
-				onlineRegisteredCount: onlineRegisteredCount,
-				socketCount: websockets.getSocketCount(),
-				users: {
-					home: rooms['/home'] ? rooms['/home'].length : 0,
-					topics: 0,
-					category: 0
-				},
-				topics: {}
+		topTenTopics.forEach(function(tid, id) {
+			socketData.topics[tid] = {
+				value: Array.isArray(roomClients['topic_' + tid]) ? roomClients['topic_' + tid].length : 0,
+				title: validator.escape(titles[id].title)
 			};
-
-		var scores = {},
-			topTenTopics = [],
-			tid;
-
-		for (var room in rooms) {
-			if (rooms.hasOwnProperty(room)) {
-				if (tid = room.match(/^\/topic_(\d+)/)) {
-					var length = rooms[room].length;
-					socketData.users.topics += length;
-
-					if (scores[length]) {
-						scores[length].push(tid[1]);
-					} else {
-						scores[length] = [tid[1]];
-					}
-				} else if (room.match(/^\/category/)) {
-					socketData.users.category += rooms[room].length;
-				}
-			}
-		}
-
-		var scoreKeys = Object.keys(scores),
-			mostActive = scoreKeys.sort();
-
-		while(topTenTopics.length < 10 && mostActive.length > 0) {
-			topTenTopics = topTenTopics.concat(scores[mostActive.pop()]);
-		}
-
-		topTenTopics = topTenTopics.slice(0, 10);
-
-		topics.getTopicsFields(topTenTopics, ['title'], function(err, titles) {
-			if (err) {
-				return callback(err);
-			}
-			topTenTopics.forEach(function(tid, id) {
-				socketData.topics[tid] = {
-					value: rooms['/topic_' + tid].length,
-					title: validator.escape(titles[id].title)
-				};
-			});
-
-			callback(null, socketData);
 		});
+
+		callback(null, socketData);
 	});
+
 };
 
 /* Exports */

src/socket.io/rooms.js

@@ -0,0 +1,95 @@
+'use strict';
+
+
+// Temp solution until
+//   https://github.com/NodeBB/NodeBB/issues/2486
+// and
+//   https://github.com/Automattic/socket.io/issues/1945
+// are closed.
+// Once they are closed switch to .clients() and async calls
+
+
+var pubsub = require('../pubsub');
+
+var rooms = {};
+
+var clientRooms = {};
+var roomClients = {};
+
+rooms.enter = function(socket, room) {
+	socket.join(room);
+	pubsub.publish('socket:join', {id: socket.id, room: room});
+};
+
+rooms.leave = function(socket, room) {
+	socket.leave(room);
+	pubsub.publish('socket:leave', {id: socket.id, room: room});
+};
+
+rooms.leaveAll = function(socket, roomsToLeave) {
+	roomsToLeave.forEach(function(room) {
+		rooms.leave(socket, room);
+	});
+};
+
+pubsub.on('socket:join', onSocketJoin);
+pubsub.on('socket:leave', onSocketLeave);
+
+function onSocketJoin(data) {
+	clientRooms[data.id] = clientRooms[data.id] || [];
+	if (clientRooms[data.id].indexOf(data.room) === -1) {
+		clientRooms[data.id].push(data.room);
+	}
+
+	roomClients[data.room] = roomClients[data.room] || [];
+	if (roomClients[data.room].indexOf(data.id) === -1) {
+		roomClients[data.room].push(data.id);
+	}
+}
+
+
+function onSocketLeave(data) {
+	var index;
+	if (Array.isArray(clientRooms[data.id])) {
+		index = clientRooms[data.id].indexOf(data.room);
+		if (index !== -1) {
+			clientRooms[data.id].splice(index, 1);
+			if (!clientRooms[data.id].length) {
+				delete clientRooms[data.id];
+			}
+		}
+	}
+
+	if (Array.isArray(roomClients[data.room])) {
+		index = roomClients[data.room].indexOf(data.id);
+		if (index !== -1) {
+			roomClients[data.room].splice(index, 1);
+			if (!roomClients[data.room].length) {
+				delete roomClients[data.room];
+			}
+		}
+	}
+}
+
+
+rooms.clients = function(room) {
+	return Array.isArray(roomClients[room]) ? roomClients[room] : [];
+};
+
+rooms.clientRooms = function(id) {
+	return Array.isArray(clientRooms[id]) ? clientRooms[id] : [];
+};
+
+rooms.socketCount = function() {
+	return Object.keys(clientRooms || {}).length;
+};
+
+rooms.roomClients = function() {
+	return roomClients;
+};
+
+
+
+
+module.exports = rooms;
+

src/views/admin/general/dashboard.tpl

@@ -1,6 +1,5 @@
 <div class="row dashboard">
-	<!-- Override for now, until the right sidebar graphs are fixed (pending socket.io resolution) -->
-	<div class="col-lg-12">
+	<div class="col-lg-9">
 		<div class="panel panel-default">
 			<div class="panel-heading">Forum Traffic</div>
 			<div class="panel-body">
@@ -88,8 +87,8 @@
 			</div>
 		</div>
 	</div>
-	<!-- Override for now, until the right sidebar graphs are fixed (pending socket.io resolution) -->
-	<div class="col-lg-3 hide">
+
+	<div class="col-lg-3">
 		<div class="panel panel-default">
 			<div class="panel-heading">Anonymous vs Registered Users</div>
 			<div class="panel-body">
@@ -108,7 +107,7 @@
 			<div class="panel-body">
 				<div class="graph-container pie-chart legend-up">
 					<ul class="graph-legend">
-						<li><div class="on-homepage"></div><span>On Homepage</span></li>
+						<li><div class="on-categories"></div><span>On categories list</span></li>
 						<li><div class="reading-posts"></div><span>Reading posts</span></li>
 						<li><div class="browsing-topics"></div><span>Browsing topics</span></li>
 						<li><div class="idle"></div><span>Idle</span></li>