fix sse shutdown

karngyan · karngyan · commit c98a0311aeab · 2025-04-09T20:34:01.000+05:30
diff --git a/server/sse.go b/server/sse.go
@@ -269,6 +269,8 @@ func (s *SSEServer) handleSSE(w http.ResponseWriter, r *http.Request) {
 		case <-r.Context().Done():
 			close(session.done)
 			return
+		case <-session.done:
+			return
 		}
 	}
 }

Original file line number	Diff line number	Diff line change
`@@ -269,6 +269,8 @@ func (s SSEServer) handleSSE(w http.ResponseWriter, r http.Request) {`
`269`	`269`	`case <-r.Context().Done():`
`270`	`270`	`close(session.done)`
`271`	`271`	`return`
	`272`	`+ case <-session.done:`
	`273`	`+ return`
`272`	`274`	`}`
`273`	`275`	`}`
`274`	`276`	`}`