mayfly-go/server/internal/db/application/db_restore.go

package application

import (
	"context"
	"mayfly-go/internal/db/domain/entity"
	"mayfly-go/internal/db/domain/repository"
	"mayfly-go/pkg/model"
)

func newDbRestoreApp(repositories *repository.Repositories, dbApp Db, scheduler *dbScheduler) (*DbRestoreApp, error) {
	var jobs []*entity.DbRestore
	if err := repositories.Restore.ListToDo(&jobs); err != nil {
		return nil, err
	}
	if err := scheduler.AddJob(context.Background(), false, entity.DbJobTypeRestore, jobs); err != nil {
		return nil, err
	}
	app := &DbRestoreApp{
		restoreRepo:        repositories.Restore,
		instanceRepo:       repositories.Instance,
		backupHistoryRepo:  repositories.BackupHistory,
		restoreHistoryRepo: repositories.RestoreHistory,
		binlogHistoryRepo:  repositories.BinlogHistory,
		dbApp:              dbApp,
		scheduler:          scheduler,
	}
	return app, nil
}

type DbRestoreApp struct {
	restoreRepo        repository.DbRestore
	instanceRepo       repository.Instance
	backupHistoryRepo  repository.DbBackupHistory
	restoreHistoryRepo repository.DbRestoreHistory
	binlogHistoryRepo  repository.DbBinlogHistory
	dbApp              Db
	scheduler          *dbScheduler
}

func (app *DbRestoreApp) Close() {
	app.scheduler.Close()
}

func (app *DbRestoreApp) Create(ctx context.Context, job *entity.DbRestore) error {
	return app.scheduler.AddJob(ctx, true /* 保存到数据库 */, entity.DbJobTypeRestore, job)
}

func (app *DbRestoreApp) Update(ctx context.Context, job *entity.DbRestore) error {
	return app.scheduler.UpdateJob(ctx, job)
}

func (app *DbRestoreApp) Delete(ctx context.Context, jobId uint64) error {
	// todo: 删除数据库恢复历史文件
	return app.scheduler.RemoveJob(ctx, entity.DbJobTypeRestore, jobId)
}

func (app *DbRestoreApp) Enable(ctx context.Context, jobId uint64) error {
	return app.scheduler.EnableJob(ctx, entity.DbJobTypeRestore, jobId)
}

func (app *DbRestoreApp) Disable(ctx context.Context, jobId uint64) error {
	return app.scheduler.DisableJob(ctx, entity.DbJobTypeRestore, jobId)
}

// GetPageList 分页获取数据库恢复任务
func (app *DbRestoreApp) GetPageList(condition *entity.DbJobQuery, pageParam *model.PageParam, toEntity any, orderBy ...string) (*model.PageResult[any], error) {
	return app.restoreRepo.GetPageList(condition, pageParam, toEntity, orderBy...)
}

// GetDbNamesWithoutRestore 获取未配置定时恢复的数据库名称
func (app *DbRestoreApp) GetDbNamesWithoutRestore(instanceId uint64, dbNames []string) ([]string, error) {
	return app.restoreRepo.GetDbNamesWithoutRestore(instanceId, dbNames)
}

// GetHistoryPageList 分页获取数据库备份历史
func (app *DbRestoreApp) GetHistoryPageList(condition *entity.DbRestoreHistoryQuery, pageParam *model.PageParam, toEntity any, orderBy ...string) (*model.PageResult[any], error) {
	return app.restoreHistoryRepo.GetDbRestoreHistories(condition, pageParam, toEntity, orderBy...)
}