Update websocket.js; Add function to get the current session amount

Author: simonwep

config/config.json

@@ -22,7 +22,7 @@
  "port": 8080,
  "storagePath": "./_storage",
  "uploadSizeLimitPerFile": 5000000000,
- "totalStorageLimitPerUser": -1,
+ "totalStorageLimitPerUser": 10000000000,
  "mediaStreamChunckSize": 4096000
  },
  "mongodb": {

src/app.js

@@ -33,7 +33,7 @@ app.use(cors());
 app.use('/api', require('./api/api'));
 
 // Spawn Websocket
-require('./websocket')(server);
+require('./websocket').launch(server);
 
 // Add express http-server
 server.on('request', app);

src/websocket.js

@@ -1,9 +1,13 @@
 const userModel = require('./models/user');
 const WebSocket = require('ws');
+const userMap = {};
 
-module.exports = server => {
+/**
+ * Launches the websocket broadcast server
+ * @param server
+ */
+module.exports.launch = server => {
  const wss = new WebSocket.Server({server: server});
- const userMap = {};
 
  wss.on('connection', ws => {
  let user;
@@ -100,3 +104,13 @@ module.exports = server => {
  });
  });
 };
+
+/**
+ * Returns the amount of currenty connected user
+ * @param userid
+ * @returns {number}
+ */
+module.exports.getSessionsBy = userid => {
+ const user = userMap[userid];
+ return ((user && user.websockets) || []).length;
+};