Global-sales/wai-server/core/websocket/services/session.js

const generateId = require('../../../utils/generateId.util');

module.exports = () => {
  const sessions = new Map();

  const createSession = ws => {
    const sessionId = generateId();
    sessions.set(ws, sessionId);
    return { sessionId, url: ws.url };
  };

  const getSession = sessionId => {
    for (const [ws, storedSessionId] of sessions) {
      if (storedSessionId === sessionId) {
        return ws;
      }
    }
    return null;
  };

  const removeSession = ws => {
    sessions.delete(ws);
  };

  return {
    createSession,
    getSession,
    removeSession,
    sessions,
  };
};