fix sse shutdown (#128)

karngyan · web-flow · commit 2e4af4cf0c6d · 2025-04-22T15:47:36.000+03:00
diff --git a/server/sse.go b/server/sse.go
@@ -321,6 +321,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
`@@ -321,6 +321,8 @@ func (s SSEServer) handleSSE(w http.ResponseWriter, r http.Request) {`
`321`	`321`	`case <-r.Context().Done():`
`322`	`322`	`close(session.done)`
`323`	`323`	`return`
	`324`	`+ case <-session.done:`
	`325`	`+ return`
`324`	`326`	`}`
`325`	`327`	`}`
`326`	`328`	`}`