diff --git a/src/socket/SocketServer.js b/src/socket/SocketServer.js index 13fd1b9..e4107ed 100644 --- a/src/socket/SocketServer.js +++ b/src/socket/SocketServer.js @@ -87,7 +87,7 @@ class SocketServer extends WebSocketEvents { cheapDetector(user.ip); ws.send(OnlineCounter.dehydrate({ - online: this.wss.clients.size || 0, + online: ipCounter.amount() || 0, })); const ip = getIPFromRequest(req); @@ -282,7 +282,7 @@ class SocketServer extends WebSocketEvents { } onlineCounterBroadcast() { - const online = this.wss.clients.size || 0; + const online = ipCounter.amount() || 0; webSockets.broadcastOnlineCounter(online); } diff --git a/src/utils/Counter.js b/src/utils/Counter.js index d07b533..578eff1 100644 --- a/src/utils/Counter.js +++ b/src/utils/Counter.js @@ -11,6 +11,10 @@ export default class Counter { this.map = new Map(); } + amount(): number { + return this.map.size; + } + get(item: T): number { const count = this.map.get(item) || 0; return count;