room stream mutex

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

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/handler/handler.go b/handler/handler.go
@@ -5,14 +5,16 @@ import (
 	"github.com/PulseyTeam/game-server/jwt"
 	pb "github.com/PulseyTeam/game-server/proto"
 	"go.mongodb.org/mongo-driver/mongo"
+	"sync"
 )
 
 type MultiplayerHandler struct {
 	pb.UnimplementedMultiplayerServiceServer
-	rooms      map[string]map[string]*pb.Player
-	jwtManager *jwt.Manager
-	mongoDB    *mongo.Client
-	cfg        *config.Config
+	rooms         map[string]map[string]*pb.Player
+	roomsMapMutex sync.RWMutex
+	jwtManager    *jwt.Manager
+	mongoDB       *mongo.Client
+	cfg           *config.Config
 }
 
 type AuthHandler struct {

diff --git a/handler/room.go b/handler/room.go
@@ -73,7 +73,9 @@ func (h *MultiplayerHandler) RoomStream(stream pb.MultiplayerService_RoomStreamS
 			log.Trace().Msg(request.String())
 		}
 
+		h.roomsMapMutex.Lock()
 		h.rooms[request.GetRoomId()][request.GetPlayer().GetId()] = request.GetPlayer()
+		h.roomsMapMutex.Unlock()
 
 		currentPlayers := make([]*pb.Player, 0, len(h.rooms[request.GetRoomId()]))
 		if len(h.rooms[request.GetRoomId()]) > 2 {