diff options
Diffstat (limited to 'internal/net/server.go')
| -rw-r--r-- | internal/net/server.go | 9 |
1 files changed, 8 insertions, 1 deletions
diff --git a/internal/net/server.go b/internal/net/server.go index 0854426..d1a0492 100644 --- a/internal/net/server.go +++ b/internal/net/server.go @@ -570,7 +570,14 @@ func (h *Hub) RekeyRooms(idMap map[int]int) { moved := make(map[int]map[*Session]bool, len(h.rooms)) for id, set := range h.rooms { if newID, ok := idMap[id]; ok && newID != id { - moved[newID] = set + if moved[newID] == nil { + moved[newID] = make(map[*Session]bool) + } + // Merge rather than assign: a many-to-one idMap (A->C, B->C) + // would otherwise clobber the first source bucket's sessions. + for s := range set { + moved[newID][s] = true + } delete(h.rooms, id) } } |