From c0fea387908ca9db96d4b63968a3c9047d1c2260 Mon Sep 17 00:00:00 2001
From: Muyao CHEN <muyao.chen@intersec.com>
Date: Thu, 26 Sep 2024 22:58:55 +0200
Subject: [PATCH] stop: use graceful shutdown

---
 handlers.go | 1 +
 main.go     | 6 ++++++
 2 files changed, 7 insertions(+)

diff --git a/handlers.go b/handlers.go
index cd03123..6dd3244 100644
--- 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
 }
diff --git a/main.go b/main.go
index b0f1847..49356f2 100644
--- 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)
+	}
 }