Reject duplicate client IDs
This commit is contained in:
parent
d6517bf0d3
commit
bee5fe4af3
|
@ -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()
|
||||||
|
|
Loading…
Reference in New Issue