129 lines
4.0 KiB
Python
129 lines
4.0 KiB
Python
|
#!/usr/bin/env python3
|
|||
|
|
# -*- coding: utf-8 -*-
|
|||
|
|
"""
|
|||
|
|
MySQL 初始化模块
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import os
|
|||
|
|
import subprocess
|
|||
|
|
import shutil
|
|||
|
|
from pathlib import Path
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
# 添加父目录到路径
|
|||
|
|
sys.path.insert(0, str(Path(__file__).parent.parent))
|
|||
|
|
from log import Logger
|
|||
|
|
|
|||
|
|
|
|||
|
|
def init_mysql(project_root, pre_deploy_commands=None):
|
|||
|
|
"""
|
|||
|
|
初始化 MySQL
|
|||
|
|
|
|||
|
|
参数：
|
|||
|
|
project_root: 项目根目录
|
|||
|
|
pre_deploy_commands: 预部署命令列表（可选）
|
|||
|
|
|
|||
|
|
返回：
|
|||
|
|
bool: 成功返回 True，失败返回 False
|
|||
|
|
"""
|
|||
|
|
try:
|
|||
|
|
project_root = Path(project_root).resolve()
|
|||
|
|
|
|||
|
|
Logger.separator()
|
|||
|
|
Logger.info("开始初始化 MySQL")
|
|||
|
|
Logger.separator()
|
|||
|
|
Logger.info(f"项目根目录: {project_root}")
|
|||
|
|
|
|||
|
|
# 定义路径
|
|||
|
|
sql_dir = project_root / "sql"
|
|||
|
|
target_dir = project_root / "docker" / "mysql" / "db"
|
|||
|
|
docker_dir = project_root / "docker"
|
|||
|
|
|
|||
|
|
# 复制 SQL 脚本
|
|||
|
|
Logger.info("复制 SQL 脚本到 MySQL 初始化目录")
|
|||
|
|
Logger.info(f"源目录: {sql_dir}")
|
|||
|
|
Logger.info(f"目标目录: {target_dir}")
|
|||
|
|
|
|||
|
|
target_dir.mkdir(parents=True, exist_ok=True)
|
|||
|
|
|
|||
|
|
# 如果提供了预部署命令，从中提取 SQL 文件名
|
|||
|
|
if pre_deploy_commands:
|
|||
|
|
for cmd in pre_deploy_commands:
|
|||
|
|
# 解析命令：cp sql/xxx.sql docker/mysql/db/
|
|||
|
|
if cmd.startswith('cp ') and '.sql' in cmd:
|
|||
|
|
parts = cmd.split()
|
|||
|
|
if len(parts) >= 2:
|
|||
|
|
source_path = parts[1] # sql/xxx.sql
|
|||
|
|
sql_file = Path(source_path).name
|
|||
|
|
source_file = project_root / source_path
|
|||
|
|
|
|||
|
|
if source_file.exists():
|
|||
|
|
Logger.info(f"复制文件: {sql_file}")
|
|||
|
|
shutil.copy2(source_file, target_dir / sql_file)
|
|||
|
|
else:
|
|||
|
|
Logger.warn(f"SQL 文件不存在: {source_file}")
|
|||
|
|
else:
|
|||
|
|
# 默认的 SQL 文件列表（向后兼容）
|
|||
|
|
sql_files = [
|
|||
|
|
"ry_20250523.sql",
|
|||
|
|
"ry_config_20250902.sql"
|
|||
|
|
]
|
|||
|
|
|
|||
|
|
for sql_file in sql_files:
|
|||
|
|
source_file = sql_dir / sql_file
|
|||
|
|
if source_file.exists():
|
|||
|
|
Logger.info(f"复制文件: {sql_file}")
|
|||
|
|
shutil.copy2(source_file, target_dir / sql_file)
|
|||
|
|
else:
|
|||
|
|
Logger.warn(f"SQL 文件不存在: {source_file}")
|
|||
|
|
|
|||
|
|
Logger.info("目标目录内容:")
|
|||
|
|
for item in target_dir.iterdir():
|
|||
|
|
Logger.info(f" - {item.name}")
|
|||
|
|
|
|||
|
|
# 构建并启动 MySQL 容器
|
|||
|
|
Logger.separator()
|
|||
|
|
Logger.info("构建 MySQL 镜像")
|
|||
|
|
Logger.separator()
|
|||
|
|
Logger.info(f"执行目录: {docker_dir}")
|
|||
|
|
Logger.info("执行命令: docker-compose build --no-cache ruoyi-mysql")
|
|||
|
|
|
|||
|
|
result = subprocess.run(
|
|||
|
|
"docker-compose build --no-cache ruoyi-mysql",
|
|||
|
|
shell=True,
|
|||
|
|
cwd=docker_dir,
|
|||
|
|
capture_output=True,
|
|||
|
|
text=True
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
if result.returncode != 0:
|
|||
|
|
Logger.error("MySQL 镜像构建失败")
|
|||
|
|
if result.stderr:
|
|||
|
|
Logger.error(f"错误信息: {result.stderr}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
Logger.info("MySQL 镜像构建成功")
|
|||
|
|
|
|||
|
|
Logger.info("执行命令: docker-compose up -d ruoyi-mysql")
|
|||
|
|
result = subprocess.run(
|
|||
|
|
"docker-compose up -d ruoyi-mysql",
|
|||
|
|
shell=True,
|
|||
|
|
cwd=docker_dir,
|
|||
|
|
capture_output=True,
|
|||
|
|
text=True
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
if result.returncode != 0:
|
|||
|
|
Logger.error("MySQL 容器启动失败")
|
|||
|
|
if result.stderr:
|
|||
|
|
Logger.error(f"错误信息: {result.stderr}")
|
|||
|
|
return False
|
|||
|
|
|
|||
|
|
Logger.info("MySQL 容器启动成功")
|
|||
|
|
Logger.info("MySQL 初始化完成")
|
|||
|
|
return True
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
Logger.error(f"MySQL 初始化异常: {e}")
|
|||
|
|
return False
|