fix Ctrl+C

2026-09-08 17:26:35 +08:00
parent 620da470ee
commit b41dc9a079
1 changed files with 10 additions and 10 deletions
--- a/main.go
+++ b/main.go
@@ -68,16 +68,16 @@ func main() {
 	signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)

 	go func() {
-		<-quit
-		fmt.Print("\nProceed to exit? (y/n): ")
-		var input string
-		fmt.Scanln(&input)
-		if input == "y" {
-			db.Save()
-			app.Shutdown()
-		} else {
-			signal.Reset(syscall.SIGINT, syscall.SIGTERM)
-			signal.Notify(quit, syscall.SIGINT, syscall.SIGTERM)
+		for {
+			<-quit
+			fmt.Print("\nProceed to exit? (y/n): ")
+			var input string
+			fmt.Scanln(&input)
+			if input == "y" {
+				db.Save()
+				app.Shutdown()
+				return
+			}
 		}
 	}()