use Request.Context().Done() instead (gin-gonic#72)

djwbyte · Apr 22, 2022 · be3d1f7 · be3d1f7
1 parent 6c5e6ca
commit be3d1f7
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/realtime-chat/main.go b/realtime-chat/main.go
@@ -5,6 +5,7 @@ import (
  "io"
  "math/rand"
  "net/http"
+
  "github.com/gin-gonic/gin"
 )
 
@@ -28,7 +29,7 @@ func stream(c *gin.Context) {
  listener := roomManager.OpenListener(roomid)
  defer roomManager.CloseListener(roomid, listener)
 
- clientGone := c.Writer.CloseNotify()
+ clientGone := c.Request.Context().Done()
  c.Stream(func(w io.Writer) bool {
  select {
  case <-clientGone: