diff --git a/src/client/state/Notifications.js b/src/client/state/Notifications.js index 1f7be3e..8ede169 100644 --- a/src/client/state/Notifications.js +++ b/src/client/state/Notifications.js @@ -117,7 +117,7 @@ class Notifications extends EventEmitter { if (addT < 0 || addH < 0) return; addNoti(roomId, addT, addH); - const allParentSpaces = this.roomList.getParentSpaces(roomId); + const allParentSpaces = this.roomList.getAllParentSpaces(roomId); allParentSpaces.forEach((spaceId) => { addNoti(spaceId, addT, addH, roomId); }); @@ -149,7 +149,7 @@ class Notifications extends EventEmitter { }; removeNoti(roomId, total, highlight); - const allParentSpaces = this.roomList.getParentSpaces(roomId); + const allParentSpaces = this.roomList.getAllParentSpaces(roomId); allParentSpaces.forEach((spaceId) => { removeNoti(spaceId, total, highlight, roomId); }); diff --git a/src/client/state/RoomList.js b/src/client/state/RoomList.js index e200aac..ec3c060 100644 --- a/src/client/state/RoomList.js +++ b/src/client/state/RoomList.js @@ -91,21 +91,20 @@ class RoomList extends EventEmitter { if (parents.size === 0) this.roomIdToParents.delete(roomId); } - getParentSpaces(roomId) { + getAllParentSpaces(roomId) { const allParents = new Set(); - const processed = new Set(); + const addAllParentIds = (rId) => { - if (processed.has(rId)) return; - processed.add(rId); + if (allParents.has(rId)) return; + allParents.add(rId); const parents = this.roomIdToParents.get(rId); if (parents === undefined) return; - parents.forEach((id) => allParents.add(id)); parents.forEach((id) => addAllParentIds(id)); }; addAllParentIds(roomId); - processed.clear(); + allParents.delete(roomId); return allParents; }