50 lines
1.7 KiB
Go
50 lines
1.7 KiB
Go
package cli
|
|
|
|
import (
|
|
"context"
|
|
"errors"
|
|
"time"
|
|
|
|
"git.pcux.cn/xuans/Scert/backend/internal/config"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/database"
|
|
"git.pcux.cn/xuans/Scert/backend/internal/secretstore"
|
|
)
|
|
|
|
func validateBackupDriver(driver string) error {
|
|
switch driver {
|
|
case "sqlite", "mysql", "postgres":
|
|
return nil
|
|
default:
|
|
return errors.New("当前数据库类型不支持备份与恢复")
|
|
}
|
|
}
|
|
|
|
// 原请求取消后仍需尝试恢复旧数据，但不能让回滚无限等待。
|
|
func runDatabaseRollback(ctx context.Context, restore func(context.Context) error) error {
|
|
rollbackCtx, cancel := context.WithTimeout(context.WithoutCancel(ctx), 5*time.Minute)
|
|
defer cancel()
|
|
if err := restore(rollbackCtx); err != nil {
|
|
return errors.Join(errors.New("数据库自动回滚失败，请保留恢复前快照并停止服务，联系管理员手动恢复"), err)
|
|
}
|
|
return nil
|
|
}
|
|
|
|
// 返回值仅供本次维护操作连接数据库，禁止写入配置、备份清单或日志。
|
|
func resolveMaintenanceDatabase(ctx context.Context, cfg config.DatabaseConfig, directory string) (config.DatabaseConfig, error) {
|
|
if err := database.ValidateConfig(cfg); err != nil {
|
|
return config.DatabaseConfig{}, errors.New("数据库维护配置无效，请检查数据库类型与加密凭据引用")
|
|
}
|
|
if cfg.Driver == "sqlite" {
|
|
return cfg, nil
|
|
}
|
|
store, err := secretstore.Open(directory)
|
|
if err != nil {
|
|
return config.DatabaseConfig{}, errors.New("无法打开数据库维护凭据存储，请检查目录及主密钥权限")
|
|
}
|
|
resolved, err := database.ResolveConfig(ctx, cfg, store)
|
|
if err != nil {
|
|
return config.DatabaseConfig{}, errors.New("无法读取数据库维护账号，请检查加密凭据及主密钥")
|
|
}
|
|
return resolved, nil
|
|
}
|