!96 删除数据库备份和恢复历史

* feat: 删除数据库备份历史 * refactor dbScheduler * feat: 从数据库备份历史中恢复数据库 * feat: 删除数据库恢复历史记录 * refactor dbScheuler
2026-01-02 04:36:35 +08:00 · 2024-01-30 13:12:43 +00:00
parent fc1b9ef35d
commit 3f828cc5b0
33 changed files with 1101 additions and 378 deletions
--- a/server/internal/db/application/db_backup.go
+++ b/server/internal/db/application/db_backup.go
@@ -3,9 +3,14 @@ package application
 import (
 	"context"
 	"encoding/binary"
+	"errors"
+	"fmt"
+	"gorm.io/gorm"
 	"mayfly-go/internal/db/domain/entity"
 	"mayfly-go/internal/db/domain/repository"
+	"mayfly-go/pkg/logx"
 	"mayfly-go/pkg/model"
+	"sync"

 	"github.com/google/uuid"
 )
@@ -14,6 +19,9 @@ type DbBackupApp struct {
 	scheduler         *dbScheduler               `inject:"DbScheduler"`
 	backupRepo        repository.DbBackup        `inject:"DbBackupRepo"`
 	backupHistoryRepo repository.DbBackupHistory `inject:"DbBackupHistoryRepo"`
+	restoreRepo       repository.DbRestore       `inject:"DbRestoreRepo"`
+	dbApp             Db                         `inject:"DbApp"`
+	mutex             sync.Mutex
 }

 func (app *DbBackupApp) Init() error {
@@ -21,7 +29,7 @@ func (app *DbBackupApp) Init() error {
 	if err := app.backupRepo.ListToDo(&jobs); err != nil {
 		return err
 	}
-	if err := app.scheduler.AddJob(context.Background(), false, entity.DbJobTypeBackup, jobs); err != nil {
+	if err := app.scheduler.AddJob(context.Background(), jobs); err != nil {
 		return err
 	}
 	return nil
@@ -32,32 +40,111 @@ func (app *DbBackupApp) Close() {
 }

 func (app *DbBackupApp) Create(ctx context.Context, jobs []*entity.DbBackup) error {
-	return app.scheduler.AddJob(ctx, true /* 保存到数据库 */, entity.DbJobTypeBackup, jobs)
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	if err := app.backupRepo.AddJob(ctx, jobs); err != nil {
+		return err
+	}
+	return app.scheduler.AddJob(ctx, jobs)
 }

 func (app *DbBackupApp) Update(ctx context.Context, job *entity.DbBackup) error {
-	return app.scheduler.UpdateJob(ctx, job)
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	if err := app.backupRepo.UpdateById(ctx, job); err != nil {
+		return err
+	}
+	_ = app.scheduler.UpdateJob(ctx, job)
+	return nil
 }

 func (app *DbBackupApp) Delete(ctx context.Context, jobId uint64) error {
 	// todo: 删除数据库备份历史文件
-	return app.scheduler.RemoveJob(ctx, entity.DbJobTypeBackup, jobId)
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	if err := app.scheduler.RemoveJob(ctx, entity.DbJobTypeBackup, jobId); err != nil {
+		return err
+	}
+	history := &entity.DbBackupHistory{
+		DbBackupId: jobId,
+	}
+	err := app.backupHistoryRepo.GetBy(history, "name")
+	switch {
+	default:
+		return err
+	case err == nil:
+		return fmt.Errorf("数据库备份存在历史记录【%s】，无法删除该任务", history.Name)
+	case errors.Is(err, gorm.ErrRecordNotFound):
+	}
+	if err := app.backupRepo.DeleteById(ctx, jobId); err != nil {
+		return err
+	}
+	return nil
 }

 func (app *DbBackupApp) Enable(ctx context.Context, jobId uint64) error {
-	return app.scheduler.EnableJob(ctx, entity.DbJobTypeBackup, jobId)
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	repo := app.backupRepo
+	job := &entity.DbBackup{}
+	if err := repo.GetById(job, jobId); err != nil {
+		return err
+	}
+	if job.IsEnabled() {
+		return nil
+	}
+	if job.IsExpired() {
+		return errors.New("任务已过期")
+	}
+	_ = app.scheduler.EnableJob(ctx, job)
+	if err := repo.UpdateEnabled(ctx, jobId, true); err != nil {
+		logx.Errorf("数据库备份任务已启用( jobId: %d )，任务状态保存失败: %v", jobId, err)
+		return err
+	}
+	return nil
 }

 func (app *DbBackupApp) Disable(ctx context.Context, jobId uint64) error {
-	return app.scheduler.DisableJob(ctx, entity.DbJobTypeBackup, jobId)
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	repo := app.backupRepo
+	job := &entity.DbBackup{}
+	if err := repo.GetById(job, jobId); err != nil {
+		return err
+	}
+	if !job.IsEnabled() {
+		return nil
+	}
+	_ = app.scheduler.DisableJob(ctx, entity.DbJobTypeBackup, jobId)
+	if err := repo.UpdateEnabled(ctx, jobId, false); err != nil {
+		logx.Errorf("数据库恢复任务已禁用( jobId: %d )，任务状态保存失败: %v", jobId, err)
+		return err
+	}
+	return nil
 }

-func (app *DbBackupApp) Start(ctx context.Context, jobId uint64) error {
-	return app.scheduler.StartJobNow(ctx, entity.DbJobTypeBackup, jobId)
+func (app *DbBackupApp) StartNow(ctx context.Context, jobId uint64) error {
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	job := &entity.DbBackup{}
+	if err := app.backupRepo.GetById(job, jobId); err != nil {
+		return err
+	}
+	if !job.IsEnabled() {
+		return errors.New("任务未启用")
+	}
+	_ = app.scheduler.StartJobNow(ctx, job)
+	return nil
 }

 // GetPageList 分页获取数据库备份任务
-func (app *DbBackupApp) GetPageList(condition *entity.DbJobQuery, pageParam *model.PageParam, toEntity any, orderBy ...string) (*model.PageResult[any], error) {
+func (app *DbBackupApp) GetPageList(condition *entity.DbBackupQuery, pageParam *model.PageParam, toEntity any, orderBy ...string) (*model.PageResult[any], error) {
 	return app.backupRepo.GetPageList(condition, pageParam, toEntity, orderBy...)
 }

@@ -68,7 +155,11 @@ func (app *DbBackupApp) GetDbNamesWithoutBackup(instanceId uint64, dbNames []str

 // GetHistoryPageList 分页获取数据库备份历史
 func (app *DbBackupApp) GetHistoryPageList(condition *entity.DbBackupHistoryQuery, pageParam *model.PageParam, toEntity any, orderBy ...string) (*model.PageResult[any], error) {
-	return app.backupHistoryRepo.GetHistories(condition, pageParam, toEntity, orderBy...)
+	return app.backupHistoryRepo.GetPageList(condition, pageParam, toEntity, orderBy...)
+}
+
+func (app *DbBackupApp) GetHistories(backupHistoryIds []uint64, toEntity any) error {
+	return app.backupHistoryRepo.GetHistories(backupHistoryIds, toEntity)
 }

 func NewIncUUID() (uuid.UUID, error) {
@@ -91,3 +182,41 @@ func NewIncUUID() (uuid.UUID, error) {

 	return uid, nil
 }
+
+func (app *DbBackupApp) DeleteHistory(ctx context.Context, historyId uint64) (retErr error) {
+	// todo: 删除数据库备份历史文件
+	app.mutex.Lock()
+	defer app.mutex.Unlock()
+
+	ok, err := app.backupHistoryRepo.UpdateDeleting(true, historyId)
+	if err != nil {
+		return err
+	}
+	defer func() {
+		_, err = app.backupHistoryRepo.UpdateDeleting(false, historyId)
+		if err == nil {
+			return
+		}
+		if retErr == nil {
+			retErr = err
+			return
+		}
+		retErr = fmt.Errorf("%w, %w", retErr, err)
+	}()
+	if !ok {
+		return errors.New("正在从备份历史中恢复数据库")
+	}
+	job := &entity.DbBackupHistory{}
+	if err := app.backupHistoryRepo.GetById(job, historyId); err != nil {
+		return err
+	}
+	conn, err := app.dbApp.GetDbConnByInstanceId(job.DbInstanceId)
+	if err != nil {
+		return err
+	}
+	dbProgram := conn.GetDialect().GetDbProgram()
+	if err := dbProgram.RemoveBackupHistory(ctx, job.DbBackupId, job.Uuid); err != nil {
+		return err
+	}
+	return app.backupHistoryRepo.DeleteById(ctx, historyId)
+}