examples/server: Shutdown on interrupt signal
This commit is contained in:
parent
35f7958083
commit
9aebcd362e
|
@ -8,6 +8,7 @@ import (
|
||||||
"context"
|
"context"
|
||||||
"log"
|
"log"
|
||||||
"os"
|
"os"
|
||||||
|
"os/signal"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
"git.sr.ht/~adnano/go-gemini"
|
"git.sr.ht/~adnano/go-gemini"
|
||||||
|
@ -31,7 +32,26 @@ func main() {
|
||||||
GetCertificate: certificates.GetCertificate,
|
GetCertificate: certificates.GetCertificate,
|
||||||
}
|
}
|
||||||
|
|
||||||
if err := server.ListenAndServe(context.Background()); err != nil {
|
// Listen for interrupt signal
|
||||||
|
c := make(chan os.Signal, 1)
|
||||||
|
signal.Notify(c, os.Interrupt)
|
||||||
|
|
||||||
|
errch := make(chan error)
|
||||||
|
go func() {
|
||||||
|
ctx := context.Background()
|
||||||
|
errch <- server.ListenAndServe(ctx)
|
||||||
|
}()
|
||||||
|
|
||||||
|
select {
|
||||||
|
case err := <-errch:
|
||||||
log.Fatal(err)
|
log.Fatal(err)
|
||||||
|
case <-c:
|
||||||
|
// Shutdown the server
|
||||||
|
log.Println("Shutting down...")
|
||||||
|
ctx, _ := context.WithTimeout(context.Background(), 30*time.Second)
|
||||||
|
err := server.Shutdown(ctx)
|
||||||
|
if err != nil {
|
||||||
|
log.Fatal(err)
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in New Issue
Block a user