room stream mutex current player

PulseyTeam · Jun 10, 2021 · 1a0a5e0 · 1a0a5e0
1 parent 8af64c4
commit 1a0a5e0
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 5 deletions.
diff --git a/handler/handler.go b/handler/handler.go
@@ -10,11 +10,12 @@ import (
 
 type MultiplayerHandler struct {
 	pb.UnimplementedMultiplayerServiceServer
-	rooms         map[string]map[string]*pb.Player
-	roomsMapMutex sync.RWMutex
-	jwtManager    *jwt.Manager
-	mongoDB       *mongo.Client
-	cfg           *config.Config
+	rooms               map[string]map[string]*pb.Player
+	roomsMapMutex       sync.RWMutex
+	currentPlayersMutex sync.RWMutex
+	jwtManager          *jwt.Manager
+	mongoDB             *mongo.Client
+	cfg                 *config.Config
 }
 
 type AuthHandler struct {

diff --git a/handler/room.go b/handler/room.go
@@ -79,9 +79,11 @@ func (h *MultiplayerHandler) RoomStream(stream pb.MultiplayerService_RoomStreamS
 
 		currentPlayers := make([]*pb.Player, 0, len(h.rooms[request.GetRoomId()]))
 		if len(h.rooms[request.GetRoomId()]) > 2 {
+			h.currentPlayersMutex.Lock()
 			for _, player := range h.rooms[request.GetRoomId()] {
 				currentPlayers = append(currentPlayers, player)
 			}
+			h.currentPlayersMutex.Unlock()
 		}
 
 		err = stream.Send(&pb.RoomStreamResponse{Players: currentPlayers})