feat(应用退出): 添加优雅退出功能防止操作中断

No-Github · No-Github · commit f18307efa529 · 2026-03-14T21:16:23.000+08:00
在应用关闭前检查是否有进行中的异步操作（部署/销毁/编排）
添加确认对话框提示用户强制退出可能导致资源状态不一致
为所有异步操作添加activeOps原子计数器进行跟踪
更新中英文国际化文本添加退出确认相关翻译
diff --git a/app.go b/app.go
@@ -7,6 +7,7 @@ import (
 	"os"
 	"path/filepath"
 	"sync"
+	"sync/atomic"
 	"time"
 
 	"red-cloud/i18n"
@@ -40,6 +41,7 @@ type App struct {
 	disableRightClick       bool
 	httpSrv                 *HTTPServer
 	wailsMode               bool // true when running inside Wails desktop
+	activeOps               atomic.Int32 // tracks in-flight async operations (apply/destroy/compose)
 }
 
 // NewApp creates a new App application struct
@@ -49,6 +51,29 @@ func NewApp() *App {
 	}
 }
 
+// HasActiveOperations returns true if any async operations (apply/destroy/compose) are in progress
+func (a *App) HasActiveOperations() bool {
+	return a.activeOps.Load() > 0
+}
+
+// beforeClose is called when the user tries to close the window
+func (a *App) beforeClose(ctx context.Context) bool {
+	if a.activeOps.Load() > 0 {
+		result, err := runtime.MessageDialog(ctx, runtime.MessageDialogOptions{
+			Type:          runtime.QuestionDialog,
+			Title:         i18n.T("app_quit_confirm_title"),
+			Message:       i18n.T("app_quit_confirm_message"),
+			Buttons:       []string{i18n.T("app_quit_btn_cancel"), i18n.T("app_quit_btn_confirm")},
+			DefaultButton: i18n.T("app_quit_btn_cancel"),
+			CancelButton:  i18n.T("app_quit_btn_cancel"),
+		})
+		if err != nil || result != i18n.T("app_quit_btn_confirm") {
+			return true // prevent close
+		}
+	}
+	return false // allow close
+}
+
 // startup is called when the app starts. The context is saved
 // so we can call the runtime methods
 func (a *App) startup(ctx context.Context) {
diff --git a/app_compose.go b/app_compose.go
@@ -91,6 +91,8 @@ func (a *App) ComposeUp(filePath string, profiles []string) error {
 	a.emitLog(i18n.Tf("app_compose_up_start", filePath))
 	a.emitEvent("compose-status", map[string]string{"action": "up", "phase": "running"})
 	go func() {
+		a.activeOps.Add(1)
+		defer a.activeOps.Add(-1)
 		defer a.emitRefresh()
 		if err := compose.RunComposeUp(opts); err != nil {
 			errMsg := i18n.Tf("app_compose_up_failed", err)
@@ -133,6 +135,8 @@ func (a *App) ComposeDown(filePath string, profiles []string) error {
 	a.emitLog(i18n.Tf("app_compose_down_start", filePath))
 	a.emitEvent("compose-status", map[string]string{"action": "down", "phase": "running"})
 	go func() {
+		a.activeOps.Add(1)
+		defer a.activeOps.Add(-1)
 		defer a.emitRefresh()
 		if err := compose.RunComposeDown(opts); err != nil {
 			errMsg := i18n.Tf("app_compose_down_failed", err)
diff --git a/app_scene.go b/app_scene.go
@@ -74,6 +74,8 @@ func (a *App) StartCase(caseID string) error {
 	a.emitLog(i18n.Tf("app_scene_prepare_start", caseName, casePath, caseState))
 
 	go func() {
+		a.activeOps.Add(1)
+		defer a.activeOps.Add(-1)
 		defer func() {
 			if r := recover(); r != nil {
 				a.emitLog(i18n.Tf("app_scene_start_error", r))
@@ -125,6 +127,8 @@ func (a *App) StopCase(caseID string) error {
 	}
 
 	go func() {
+		a.activeOps.Add(1)
+		defer a.activeOps.Add(-1)
 		defer func() {
 			if r := recover(); r != nil {
 				a.emitLog(i18n.Tf("app_scene_stop_error", r))
@@ -165,6 +169,8 @@ func (a *App) RemoveCase(caseID string) error {
 	}
 
 	go func() {
+		a.activeOps.Add(1)
+		defer a.activeOps.Add(-1)
 		defer func() {
 			if r := recover(); r != nil {
 				a.emitLog(i18n.Tf("app_scene_delete_error", r))
@@ -228,6 +234,8 @@ func (a *App) CreateAndRunCase(templateName string, name string, vars map[string
 	a.emitLog(i18n.Tf("app_creating_running_scene", name, templateName))
 
 	go func() {
+		a.activeOps.Add(1)
+		defer a.activeOps.Add(-1)
 		defer func() {
 			if r := recover(); r != nil {
 				a.emitLog(i18n.Tf("app_scene_init_error", r))
diff --git a/frontend/wailsjs/wailsjs/go/main/App.d.ts b/frontend/wailsjs/wailsjs/go/main/App.d.ts
@@ -177,6 +177,8 @@ export function GetVersion():Promise<string>;
 
 export function GetWebhookConfig():Promise<main.WebhookConfig>;
 
+export function HasActiveOperations():Promise<boolean>;
+
 export function ImportConfigTemplate(arg1:string,arg2:string):Promise<void>;
 
 export function ImportTemplates(arg1:string):Promise<Array<string>>;
diff --git a/frontend/wailsjs/wailsjs/go/main/App.js b/frontend/wailsjs/wailsjs/go/main/App.js
@@ -342,6 +342,10 @@ export function GetWebhookConfig() {
   return window['go']['main']['App']['GetWebhookConfig']();
 }
 
+export function HasActiveOperations() {
+  return window['go']['main']['App']['HasActiveOperations']();
+}
+
 export function ImportConfigTemplate(arg1, arg2) {
   return window['go']['main']['App']['ImportConfigTemplate'](arg1, arg2);
 }
diff --git a/i18n/en.go b/i18n/en.go
@@ -550,4 +550,10 @@ var enMessages = map[string]string{
 	"plugin_disable_short":   "Disable a plugin",
 	"plugin_update_short":    "Update a plugin (git pull)",
 	"plugin_info_short":      "Show plugin details",
+
+	// Graceful quit
+	"app_quit_confirm_title":   "Confirm Exit",
+	"app_quit_confirm_message": "There are active operations (deploy/destroy/compose). Are you sure you want to exit? Force quitting may leave resources in an inconsistent state.",
+	"app_quit_btn_confirm":     "Force Quit",
+	"app_quit_btn_cancel":      "Cancel",
 }
diff --git a/i18n/zh.go b/i18n/zh.go
@@ -550,4 +550,10 @@ var zhMessages = map[string]string{
 	"plugin_disable_short":   "禁用插件",
 	"plugin_update_short":    "更新插件 (git pull)",
 	"plugin_info_short":      "查看插件详情",
+
+	// Graceful quit
+	"app_quit_confirm_title":   "确认退出",
+	"app_quit_confirm_message": "有正在进行的操作（部署/销毁/编排），确定要退出吗？强制退出可能导致资源状态不一致。",
+	"app_quit_btn_confirm":     "强制退出",
+	"app_quit_btn_cancel":      "取消",
 }
diff --git a/main.go b/main.go
@@ -40,6 +40,7 @@ func main() {
 		},
 		BackgroundColour: &options.RGBA{R: 250, G: 251, B: 252, A: 1}, // 匹配 App.svelte 的 bg-[#fafbfc]
 		OnStartup:        app.startup,
+		OnBeforeClose:    app.beforeClose,
 		Bind: []interface{}{
 			app,
 		},
diff --git a/mod/flag.go b/mod/flag.go
@@ -8,5 +8,5 @@ var (
 	Domain        string
 	Domain2       string
 	Base64Command string
-	Version       = "v3.1.1"
+	Version       = "v3.1.2"
 )

Original file line number	Diff line number	Diff line change
`@@ -342,6 +342,10 @@ export function GetWebhookConfig() {`
`342`	`342`	`return window['go']['main']['App']['GetWebhookConfig']();`
`343`	`343`	`}`
`344`	`344`
	`345`	`+export function HasActiveOperations() {`
	`346`	`+ return window['go']['main']['App']['HasActiveOperations']();`
	`347`	`+}`
	`348`	`+`
`345`	`349`	`export function ImportConfigTemplate(arg1, arg2) {`
`346`	`350`	`return window['go']['main']['App']['ImportConfigTemplate'](arg1, arg2);`
`347`	`351`	`}`
Original file line number	Diff line number	Diff line change
`@@ -8,5 +8,5 @@ var (`
`8`	`8`	`Domain string`
`9`	`9`	`Domain2 string`
`10`	`10`	`Base64Command string`
`11`		`- Version = "v3.1.1"`
	`11`	`+ Version = "v3.1.2"`
`12`	`12`	`)`