From aa661972cc3b52feb20388a0e5a09e9a033b6c4f Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E8=80=97=E5=AD=90?= Date: Tue, 4 Feb 2025 20:34:44 +0800 Subject: [PATCH] fix: lint --- internal/data/backup.go | 12 ++++++++++-- 1 file changed, 10 insertions(+), 2 deletions(-) diff --git a/internal/data/backup.go b/internal/data/backup.go index 30959cac..92d0d473 100644 --- a/internal/data/backup.go +++ b/internal/data/backup.go @@ -418,12 +418,13 @@ func (r *backupRepo) restoreMySQL(backup, target string) error { return err } + clean := false if !strings.HasSuffix(backup, ".sql") { backup, err = r.autoUnCompressSQL(backup) if err != nil { return err } - defer io.Remove(filepath.Dir(backup)) // 删除解压目录 + clean = true } if _, err = shell.Execf(`mysql -u root '%s' < '%s'`, target, backup); err != nil { @@ -432,6 +433,9 @@ func (r *backupRepo) restoreMySQL(backup, target string) error { if err = os.Unsetenv("MYSQL_PWD"); err != nil { return err } + if clean { + _ = io.Remove(filepath.Dir(backup)) + } return nil } @@ -451,17 +455,21 @@ func (r *backupRepo) restorePostgres(backup, target string) error { return fmt.Errorf("数据库不存在：%s", target) } + clean := false if !strings.HasSuffix(backup, ".sql") { backup, err = r.autoUnCompressSQL(backup) if err != nil { return err } - defer io.Remove(filepath.Dir(backup)) // 删除解压目录 + clean = true } if _, err = shell.Execf(`su - postgres -c "psql '%s'" < '%s'`, target, backup); err != nil { return err } + if clean { + _ = io.Remove(filepath.Dir(backup)) + } return nil }