sync gin #382

trofman · Mar 1, 2019 · 1ba5ef4 · 1ba5ef4
1 parent 871ad40
commit 1ba5ef4
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/realtime-chat/main.go b/realtime-chat/main.go
@@ -26,9 +26,15 @@ func stream(c *gin.Context) {
  listener := openListener(roomid)
  defer closeListener(roomid, listener)
 
+ clientGone := c.Writer.CloseNotify()
  c.Stream(func(w io.Writer) bool {
- c.SSEvent("message", <-listener)
- return true
+ select {
+ case <-clientGone:
+ return false
+ case message := <-listener:
+ c.SSEvent("message", message)
+ return true
+ }
  })
 }