85 lines
2.9 KiB
PowerShell
85 lines
2.9 KiB
PowerShell
[CmdletBinding()]
|
|
param(
|
|
[Parameter(Mandatory = $true)]
|
|
[string]$BackupDirectory,
|
|
[string]$ComposeFile = "",
|
|
[string]$PostgresDatabase = $env:POSTGRES_DB,
|
|
[string]$PostgresUser = $env:POSTGRES_USER,
|
|
[string]$Confirm = $env:RESTORE_CONFIRM,
|
|
[string]$BackupHelperImage = "alpine:3.21",
|
|
[switch]$SkipApplicationServices
|
|
)
|
|
|
|
$ErrorActionPreference = "Stop"
|
|
if ($Confirm -ne "restore-ai-ticket-platform") {
|
|
throw "恢复会替换当前 PostgreSQL 与 MinIO 数据；请传入 -Confirm restore-ai-ticket-platform。"
|
|
}
|
|
|
|
$scriptDirectory = Split-Path -Parent $MyInvocation.MyCommand.Path
|
|
$infraDirectory = [IO.Path]::GetFullPath((Join-Path $scriptDirectory ".."))
|
|
if (-not $ComposeFile) {
|
|
$ComposeFile = Join-Path $infraDirectory "docker-compose.prod.yml"
|
|
}
|
|
$ComposeFile = (Resolve-Path -LiteralPath $ComposeFile).Path
|
|
$target = (Resolve-Path -LiteralPath $BackupDirectory).Path
|
|
if (-not $PostgresDatabase) { $PostgresDatabase = "ai_ticket_platform" }
|
|
if (-not $PostgresUser) { $PostgresUser = "ai_ticket" }
|
|
|
|
& (Join-Path $scriptDirectory "verify-backup.ps1") -BackupDirectory $target
|
|
|
|
$postgresTemp = "/tmp/ai-ticket-restore-$PID.dump"
|
|
$minioContainer = & docker compose -f $ComposeFile ps -q minio
|
|
if ($LASTEXITCODE -ne 0 -or -not "$minioContainer".Trim()) {
|
|
throw "未找到 MinIO 容器"
|
|
}
|
|
$minioContainer = "$minioContainer".Trim()
|
|
|
|
function Invoke-Compose {
|
|
param([Parameter(Mandatory = $true)][string[]]$Arguments)
|
|
& docker compose -f $ComposeFile @Arguments
|
|
if ($LASTEXITCODE -ne 0) {
|
|
throw "docker compose 执行失败：$($Arguments -join ' ')"
|
|
}
|
|
}
|
|
|
|
if (-not $SkipApplicationServices) {
|
|
Invoke-Compose @("stop", "backend", "worker", "beat")
|
|
}
|
|
Invoke-Compose @("stop", "minio")
|
|
|
|
try {
|
|
Invoke-Compose @("cp", (Join-Path $target "postgres.dump"), "postgres:$postgresTemp")
|
|
Invoke-Compose @(
|
|
"exec", "-T", "postgres",
|
|
"dropdb", "-U", $PostgresUser, "--if-exists", "--force", $PostgresDatabase
|
|
)
|
|
Invoke-Compose @(
|
|
"exec", "-T", "postgres",
|
|
"createdb", "-U", $PostgresUser, $PostgresDatabase
|
|
)
|
|
Invoke-Compose @(
|
|
"exec", "-T", "postgres",
|
|
"pg_restore", "-U", $PostgresUser, "-d", $PostgresDatabase,
|
|
"--no-owner", "--exit-on-error", $postgresTemp
|
|
)
|
|
|
|
& docker run --rm `
|
|
--volumes-from $minioContainer `
|
|
-v "${target}:/backup:ro" `
|
|
$BackupHelperImage `
|
|
sh -c "find /data -mindepth 1 -maxdepth 1 -exec rm -rf -- {} + && tar -C /data -xzf /backup/minio-data.tar.gz"
|
|
if ($LASTEXITCODE -ne 0) {
|
|
throw "MinIO 数据恢复失败"
|
|
}
|
|
Invoke-Compose @("up", "-d", "--wait", "minio")
|
|
}
|
|
finally {
|
|
& docker compose -f $ComposeFile exec -T postgres rm -f $postgresTemp 2>$null
|
|
}
|
|
|
|
if (-not $SkipApplicationServices) {
|
|
Invoke-Compose @("start", "backend", "worker", "beat")
|
|
}
|
|
|
|
Write-Output "恢复完成。请立即检查 /api/system/health/ready 并执行核心业务冒烟测试。"
|