handle exit logic

2025-01-10 07:25:09 +08:00
parent ebffac7ba4
commit db10ed8378
4 changed files with 32 additions and 8 deletions
--- a/main.go
+++ b/main.go
@@ -55,7 +55,12 @@ func main() {
 		return nil
 	})

-	defer log.Println("Exit!")
+	app.OnTerminate().Add(func(e *core.TerminateEvent) error {
+		routes.Unregister()
+		log.Println("Exit!")
+		return nil
+	})
+
 	log.Printf("Visit the website: http://%s", httpFlag)

 	if err := app.Start(); err != nil {