stop: use graceful shutdown

2024-09-26 22:58:55 +02:00 · 2024-09-26 22:58:55 +02:00 · c0fea38790
commit c0fea38790
parent b98513ad36
2 changed files with 7 additions and 0 deletions
--- a/handlers.go
+++ b/handlers.go
@ -58,6 +58,7 @@ func FooControllerHandler(ctx *framework.Context) error {
 }

 func UserLoginController(ctx *framework.Context) error {
+	time.Sleep(10 * time.Second)
 	ctx.WriteJSON(http.StatusOK, "ok")
 	return nil
 }
--- a/main.go
+++ b/main.go
@ -1,7 +1,9 @@
 package main

 import (
+	"context"
 	"fmt"
+	"log"
 	"net/http"
 	"os"
 	"os/signal"
@ -30,4 +32,8 @@ func main() {

 	<-quit
 	fmt.Println("YOLO")
+
+	if err := server.Shutdown(context.Background()); err != nil {
+		log.Fatal("server shutdown: ", err)
+	}
 }