Reject duplicate client IDs

This commit is contained in:
zikaeroh 2020-06-14 12:23:33 -07:00
parent d6517bf0d3
commit bee5fe4af3
1 changed files with 28 additions and 15 deletions

View File

@ -245,26 +245,39 @@ func (r *Room) HandleConn(ctx context.Context, playerID uuid.UUID, nickname stri
g, ctx := errgroup.WithContext(ctx) g, ctx := errgroup.WithContext(ctx)
r.mu.Lock() setup := func() (success bool) {
r.players[playerID] = func(s protocol.ServerNote) { r.mu.Lock()
if ctx.Err() != nil { defer r.mu.Unlock()
return
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 r.players[playerID] = func(s protocol.ServerNote) {
// goroutine and hope that errors here will be reflected later via ping/receive failures. if ctx.Err() != nil {
go func() {
ctx, cancel := context.WithTimeout(ctx, time.Second)
defer cancel()
if err := wsjson.Write(ctx, c, &s); err != nil {
return 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() { defer func() {
r.mu.Lock() r.mu.Lock()