Mirrors/cloud-game
1
0
Fork 0
mirror of https://github.com/giongto35/cloud-game.git synced 2026-07-17 16:50:31 +00:00
Code Issues Projects Releases Packages Wiki Activity

Cache peer slice to avoid allocs during map iterations

Browse source
This commit is contained in:
sergystepanov 2026-07-06 16:06:17 +03:00
parent 6823adbb9c
commit 4b23798904
1 changed files with 53 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

60
pkg/worker/room/room.go
View file

@ -50,6 +50,9 @@ type Room[T Session] struct {
media MediaPipe
users SessionManager[T]
peers []T
peersMu sync.RWMutex
closed bool
HandleClose func()
}
@ -71,13 +74,19 @@ func (r *Room[T]) InitMedia() {
}
func (r *Room[T]) sendAudio(data []byte, dur time.Duration) {
for u := range r.users.Values() {
r.peersMu.RLock()
peers := r.peers
r.peersMu.RUnlock()
for _, u := range peers {
u.SendAudio(data, dur)
}
}
func (r *Room[T]) sendVideo(data []byte, dur time.Duration) {
for u := range r.users.Values() {
r.peersMu.RLock()
peers := r.peers
r.peersMu.RUnlock()
for _, u := range peers {
u.SendVideo(data, dur)
}
}
@ -88,11 +97,28 @@ func (r *Room[T]) SetApp(app app.App) { r.app = app }
func (r *Room[T]) SetMedia(m MediaPipe) { r.media = m }
func (r *Room[T]) StartApp() { r.app.Start() }
func (r *Room[T]) Send(data []byte) {
for u := range r.users.Values() {
r.peersMu.RLock()
peers := r.peers
r.peersMu.RUnlock()
for _, u := range peers {
u.SendData(data)
}
}
// SyncPeers rebuilds the cached peer slice from the session manager.
// Must be called after any add/remove/reset of sessions.
func (r *Room[T]) SyncPeers() {
if r.users == nil {
return
}
r.peersMu.Lock()
r.peers = r.peers[:0]
for u := range r.users.Values() {
r.peers = append(r.peers, u)
}
r.peersMu.Unlock()
}
func (r *Room[T]) Close() {
if r == nil || r.closed {
return
@ -127,13 +153,27 @@ func (r *Router[T]) FindRoom(id string) *Room[T] {
return nil
}
func (r *Router[T]) AddUser(user T) { r.mu.Lock(); r.users.Add(user); r.mu.Unlock() }
func (r *Router[T]) AddUser(user T) {
r.mu.Lock()
r.users.Add(user)
r.mu.Unlock()
if r.room != nil {
r.room.SyncPeers()
}
}
func (r *Router[T]) FindUser(uid string) T { return r.users.Find(uid) }
func (r *Router[T]) Users() SessionManager[T] { return r.users }
func (r *Router[T]) Room() *Room[T] { r.mu.Lock(); defer r.mu.Unlock(); return r.room }
func (r *Router[T]) SetRoom(room *Room[T]) { r.mu.Lock(); r.room = room; r.mu.Unlock() }
func (r *Router[T]) HasRoom() bool { r.mu.Lock(); defer r.mu.Unlock(); return r.room != nil }
func (r *Router[T]) Close() { r.mu.Lock(); r.close(); r.mu.Unlock() }
func (r *Router[T]) SetRoom(room *Room[T]) {
r.mu.Lock()
r.room = room
r.mu.Unlock()
if room != nil {
room.SyncPeers()
}
}
func (r *Router[T]) HasRoom() bool { r.mu.Lock(); defer r.mu.Unlock(); return r.room != nil }
func (r *Router[T]) Close() { r.mu.Lock(); r.close(); r.mu.Unlock() }
func (r *Router[T]) Reset() {
r.mu.Lock()
r.close()
@ -141,12 +181,18 @@ func (r *Router[T]) Reset() {
u.Disconnect()
}
r.users.Reset()
if r.room != nil {
r.room.SyncPeers()
}
r.mu.Unlock()
}
func (r *Router[T]) Remove(user T) {
r.mu.Lock()
defer r.mu.Unlock()
r.users.Remove(user)
if r.room != nil {
r.room.SyncPeers()
}
if r.users.Empty() {
r.close()
}