Initial commit: TaskPool React panel
- React frontend with route-level code splitting - Backend rebranded from Baihu to TaskPool - DB brand migration script and local compatibility
This commit is contained in:
@@ -0,0 +1,64 @@
|
||||
package restore
|
||||
|
||||
import (
|
||||
"flag"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
|
||||
"github.com/engigu/taskpool/cmd/clibase"
|
||||
"github.com/engigu/taskpool/internal/services"
|
||||
)
|
||||
|
||||
func printHelp() {
|
||||
clibase.PrintSubCommandUsage("任务池系统数据恢复工具", "taskpool restore <备份文件.zip>", " taskpool restore backup_20231027.zip", nil)
|
||||
}
|
||||
|
||||
func Run(args []string) {
|
||||
if len(args) > 0 && (args[0] == "-h" || args[0] == "--help") {
|
||||
printHelp()
|
||||
return
|
||||
}
|
||||
|
||||
fs := flag.NewFlagSet("restore", flag.ExitOnError)
|
||||
fs.Usage = printHelp
|
||||
|
||||
if err := fs.Parse(args); err != nil {
|
||||
return
|
||||
}
|
||||
|
||||
parsedArgs := fs.Args()
|
||||
if len(parsedArgs) < 1 {
|
||||
fmt.Fprintf(os.Stderr, "错误: 必须提供备份文件路径\n")
|
||||
fs.Usage()
|
||||
return
|
||||
}
|
||||
|
||||
backupFile := parsedArgs[0]
|
||||
absPath, err := filepath.Abs(backupFile)
|
||||
if err != nil {
|
||||
fmt.Printf("文件路径解析失败: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
if _, err := os.Stat(absPath); os.IsNotExist(err) {
|
||||
fmt.Printf("错误: 备份文件 '%s' 不存在\n", absPath)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
// 必须初始化环境与数据库才能恢复数据
|
||||
clibase.InitContext(false)
|
||||
|
||||
backupService := services.NewBackupService()
|
||||
fmt.Printf("正在从 '%s' 恢复系统数据,请勿强制中断...\n", absPath)
|
||||
err = backupService.Restore(absPath)
|
||||
if err != nil {
|
||||
fmt.Printf("恢复备份失败: %v\n", err)
|
||||
os.Exit(1)
|
||||
}
|
||||
|
||||
fmt.Println("--------------------------------------------------")
|
||||
fmt.Println("系统备份恢复成功!")
|
||||
fmt.Println("注意:部分设定可能需要重启后台服务后才能完全生效。")
|
||||
fmt.Println("--------------------------------------------------")
|
||||
}
|
||||
Reference in New Issue
Block a user