examples/stream: Remove /shutdown handler
This commit is contained in:
parent
9aebcd362e
commit
3b8b5d6557
@ -8,7 +8,6 @@ import (
|
|||||||
"context"
|
"context"
|
||||||
"fmt"
|
"fmt"
|
||||||
"log"
|
"log"
|
||||||
"sync"
|
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"git.sr.ht/~adnano/go-gemini"
|
"git.sr.ht/~adnano/go-gemini"
|
||||||
@ -32,21 +31,9 @@ func main() {
|
|||||||
GetCertificate: certificates.GetCertificate,
|
GetCertificate: certificates.GetCertificate,
|
||||||
}
|
}
|
||||||
|
|
||||||
var shutdownOnce sync.Once
|
ctx := context.Background()
|
||||||
var wg sync.WaitGroup
|
if err := server.ListenAndServe(ctx); err != nil {
|
||||||
wg.Add(1)
|
log.Fatal(err)
|
||||||
defer wg.Wait()
|
|
||||||
mux.HandleFunc("/shutdown", func(ctx context.Context, w gemini.ResponseWriter, r *gemini.Request) {
|
|
||||||
fmt.Fprintln(w, "Shutting down...")
|
|
||||||
w.Flush()
|
|
||||||
go shutdownOnce.Do(func() {
|
|
||||||
server.Shutdown(context.Background())
|
|
||||||
wg.Done()
|
|
||||||
})
|
|
||||||
})
|
|
||||||
|
|
||||||
if err := server.ListenAndServe(context.Background()); err != nil {
|
|
||||||
log.Println(err)
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user