60 lines
2.4 KiB
Python
60 lines
2.4 KiB
Python
from __future__ import annotations
|
||
|
||
import asyncio
|
||
import logging
|
||
from datetime import datetime
|
||
from pathlib import Path
|
||
|
||
from .compose_manager import ComposeManager
|
||
from .models import CommandResult
|
||
|
||
logger = logging.getLogger(__name__)
|
||
|
||
|
||
class MysqlBackupManager:
|
||
def __init__(self, *, enabled: bool, backup_dir: str, dump_command: str, timeout_seconds: int, compose_manager: ComposeManager) -> None:
|
||
self.enabled = enabled
|
||
self.backup_dir = Path(backup_dir)
|
||
self.dump_command = dump_command
|
||
self.timeout_seconds = timeout_seconds
|
||
self.compose_manager = compose_manager
|
||
if self.enabled:
|
||
self.backup_dir.mkdir(parents=True, exist_ok=True)
|
||
|
||
async def backup_before_upgrade(self, target_release: str) -> CommandResult:
|
||
if not self.enabled:
|
||
return CommandResult(success=True, stdout="mysql backup skipped", returncode=0)
|
||
|
||
timestamp = datetime.now().strftime("%Y%m%d-%H%M%S")
|
||
backup_file = self.backup_dir / f"mysql-backup-{target_release}-{timestamp}.sql"
|
||
command = self.dump_command.replace("{backup_file}", str(backup_file))
|
||
logger.info("开始执行MySQL备份，目标文件: %s", backup_file)
|
||
result = await self._run_shell_command(command)
|
||
if result.success:
|
||
logger.info("MySQL备份完成: %s", backup_file)
|
||
result.stdout = str(backup_file)
|
||
else:
|
||
logger.error("MySQL备份失败: %s", result.stderr or result.stdout)
|
||
return result
|
||
|
||
async def _run_shell_command(self, command: str) -> CommandResult:
|
||
process = await asyncio.create_subprocess_shell(
|
||
command,
|
||
cwd=str(self.compose_manager.working_dir),
|
||
stdout=asyncio.subprocess.PIPE,
|
||
stderr=asyncio.subprocess.PIPE,
|
||
)
|
||
try:
|
||
stdout, stderr = await asyncio.wait_for(process.communicate(), timeout=self.timeout_seconds)
|
||
except asyncio.TimeoutError:
|
||
process.kill()
|
||
await process.wait()
|
||
return CommandResult(success=False, stderr=f"mysql backup timeout after {self.timeout_seconds}s", returncode=-1)
|
||
|
||
return CommandResult(
|
||
success=process.returncode == 0,
|
||
stdout=stdout.decode("utf-8", errors="ignore"),
|
||
stderr=stderr.decode("utf-8", errors="ignore"),
|
||
returncode=process.returncode or 0,
|
||
)
|