Correctly cleaning room in pusher even if an exception occurs

Closes #1646 (again)
This commit is contained in:
David Négrier 2021-12-20 15:33:47 +01:00
parent e0aab9c0ce
commit d792d35d09
2 changed files with 13 additions and 5 deletions

View File

@ -273,7 +273,7 @@ export class IoSocketController {
rejected: true,
message: err?.response?.data.message,
status: err?.response?.status,
room,
roomId,
},
websocketKey,
websocketProtocol,
@ -368,6 +368,7 @@ export class IoSocketController {
rejected: true,
reason: e instanceof InvalidTokenError ? tokenInvalidException : null,
message: e.message,
roomId,
},
websocketKey,
websocketProtocol,
@ -380,6 +381,7 @@ export class IoSocketController {
rejected: true,
reason: null,
message: "500 Internal Server Error",
roomId,
},
websocketKey,
websocketProtocol,
@ -394,9 +396,8 @@ export class IoSocketController {
open: (ws) => {
if (ws.rejected === true) {
// If there is a room in the error, let's check if we need to clean it.
if (ws.room) {
const room = ws.room as PusherRoom;
socketManager.deleteRoomIfEmpty(room);
if (ws.roomId) {
socketManager.deleteRoomIfEmptyFromId(ws.roomId as string);
}
//FIX ME to use status code

View File

@ -370,7 +370,7 @@ export class SocketManager implements ZoneEventListener {
}
}
public deleteRoomIfEmpty(room: PusherRoom): void {
private deleteRoomIfEmpty(room: PusherRoom): void {
if (room.isEmpty()) {
room.close();
this.rooms.delete(room.roomUrl);
@ -378,6 +378,13 @@ export class SocketManager implements ZoneEventListener {
}
}
public deleteRoomIfEmptyFromId(roomUrl: string): void {
const room = this.rooms.get(roomUrl);
if (room) {
this.deleteRoomIfEmpty(room);
}
}
async getOrCreateRoom(roomUrl: string): Promise<PusherRoom> {
//check and create new world for a room
let room = this.rooms.get(roomUrl);