Reject duplicate client IDs

zikaeroh created

Change summary

internal/server/server.go | 43 ++++++++++++++++++++++++++--------------
1 file changed, 28 insertions(+), 15 deletions(-)

Detailed changes

internal/server/server.go 🔗

@@ -245,26 +245,39 @@ func (r *Room) HandleConn(ctx context.Context, playerID uuid.UUID, nickname stri
 
 	g, ctx := errgroup.WithContext(ctx)
 
-	r.mu.Lock()
-	r.players[playerID] = func(s protocol.ServerNote) {
-		if ctx.Err() != nil {
-			return
+	setup := func() (success bool) {
+		r.mu.Lock()
+		defer r.mu.Unlock()
+
+		if _, ok := r.players[playerID]; ok {
+			ctxlog.Warn(ctx, "client with this ID already exists")
+			return false
 		}
 
-		// It's not safe to start more group goroutines concurrently; just use a regular
-		// goroutine and hope that errors here will be reflected later via ping/receive failures.
-		go func() {
-			ctx, cancel := context.WithTimeout(ctx, time.Second)
-			defer cancel()
-			if err := wsjson.Write(ctx, c, &s); err != nil {
+		r.players[playerID] = func(s protocol.ServerNote) {
+			if ctx.Err() != nil {
 				return
 			}
-			metricSent.Inc()
-		}()
+
+			// It's not safe to start more group goroutines concurrently; just use a regular
+			// goroutine and hope that errors here will be reflected later via ping/receive failures.
+			go func() {
+				ctx, cancel := context.WithTimeout(ctx, time.Second)
+				defer cancel()
+				if err := wsjson.Write(ctx, c, &s); err != nil {
+					return
+				}
+				metricSent.Inc()
+			}()
+		}
+		r.room.AddPlayer(playerID, nickname)
+		r.sendAll()
+		return true
+	}
+
+	if !setup() {
+		return
 	}
-	r.room.AddPlayer(playerID, nickname)
-	r.sendAll()
-	r.mu.Unlock()
 
 	defer func() {
 		r.mu.Lock()