sashakoshka/go-gemini
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

examples/stream: Remove usage of Flusher

Browse Source
This commit is contained in:
Adnan Maolood 2021-02-22 20:07:15 -05:00
parent 118e019df0
commit c5b304216c
1 changed files with 2 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
examples/stream.go
View File

@ -38,9 +38,7 @@ func main() {
defer wg.Wait()
mux.HandleFunc("/shutdown", func(ctx context.Context, w gemini.ResponseWriter, r *gemini.Request) {
fmt.Fprintln(w, "Shutting down...")
if flusher, ok := w.(gemini.Flusher); ok {
flusher.Flush()
}
w.Flush()
go shutdownOnce.Do(func() {
server.Shutdown(context.Background())
wg.Done()
@ -54,12 +52,6 @@ func main() {
// stream writes an infinite stream to w.
func stream(ctx context.Context, w gemini.ResponseWriter, r *gemini.Request) {
flusher, ok := w.(gemini.Flusher)
if !ok {
w.WriteHeader(gemini.StatusTemporaryFailure, "Internal error")
return
}
ch := make(chan string)
ctx, cancel := context.WithCancel(ctx)
@ -84,7 +76,7 @@ func stream(ctx context.Context, w gemini.ResponseWriter, r *gemini.Request) {
break
}
fmt.Fprintln(w, s)
if err := flusher.Flush(); err != nil {
if err := w.Flush(); err != nil {
cancel()
return
}