Ensure connection is closed after group cancel for safety

zikaeroh created 5 years ago

Change summary

internal/server/server.go | 5 +++++
1 file changed, 5 insertions(+)

Detailed changes

internal/server/server.go

@@ -266,6 +266,11 @@ func (r *Room) HandleConn(playerID uuid.UUID, nickname string, c *websocket.Conn
 		r.sendAll()
 	}()
 
+	g.Go(func() error {
+		<-ctx.Done()
+		return c.Close(websocket.StatusGoingAway, "going away")
+	})
+
 	g.Go(func() error {
 		ticker := time.NewTicker(time.Minute)
 		defer ticker.Stop()