vinchent/go-web
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

stop: use graceful shutdown for 5s

Browse Source
This commit is contained in:
Muyao CHEN 2024-09-26 23:05:05 +02:00
parent c0fea38790
commit c5ab1debdb
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
main.go
View File

@ -8,6 +8,7 @@ import (
"os" "os"
"os/signal" "os/signal"
"syscall" "syscall"
"time"
"git.vinchent.xyz/vinchent/go-web/framework" "git.vinchent.xyz/vinchent/go-web/framework"
) )
@ -33,7 +34,9 @@ func main() {
<-quit <-quit
fmt.Println("YOLO") fmt.Println("YOLO")
if err := server.Shutdown(context.Background()); err != nil { ctx, cancel := context.WithTimeout(context.Background(), 5*time.Second)
defer cancel()
if err := server.Shutdown(ctx); err != nil {
log.Fatal("server shutdown: ", err) log.Fatal("server shutdown: ", err)
} }
} }