322 lines
11 KiB
Python
322 lines
11 KiB
Python
#!/usr/bin/env python3
|
||
# -*- coding: utf-8 -*-
|
||
"""
|
||
部署执行器
|
||
负责执行具体的部署任务
|
||
"""
|
||
|
||
import os
|
||
import sys
|
||
import logging
|
||
import subprocess
|
||
import shutil
|
||
import glob
|
||
from pathlib import Path
|
||
|
||
|
||
class Deployer:
|
||
"""部署执行器"""
|
||
|
||
def __init__(self, config):
|
||
"""初始化部署器"""
|
||
self.config = config
|
||
self.logger = logging.getLogger('Deployer')
|
||
self.runtime_path = Path(config['main_repository']['runtime_path'])
|
||
self.main_repo_url = config['main_repository']['url']
|
||
self.main_repo_branch = config['main_repository']['branch']
|
||
|
||
def run_command(self, cmd, cwd=None, timeout=600):
|
||
"""执行命令"""
|
||
self.logger.info(f"执行命令: {cmd}")
|
||
try:
|
||
result = subprocess.run(
|
||
cmd,
|
||
shell=True,
|
||
cwd=cwd,
|
||
capture_output=True,
|
||
text=True,
|
||
timeout=timeout
|
||
)
|
||
|
||
if result.stdout:
|
||
self.logger.debug(f"输出: {result.stdout}")
|
||
|
||
if result.returncode != 0:
|
||
self.logger.error(f"命令执行失败: {result.stderr}")
|
||
return False
|
||
|
||
return True
|
||
except subprocess.TimeoutExpired:
|
||
self.logger.error(f"命令执行超时: {cmd}")
|
||
return False
|
||
except Exception as e:
|
||
self.logger.error(f"命令执行异常: {e}")
|
||
return False
|
||
|
||
def ensure_main_repo(self):
|
||
"""确保主仓库存在并是最新的"""
|
||
repo_path = self.runtime_path / 'a-cloud-all'
|
||
|
||
if not repo_path.exists():
|
||
self.logger.info("主仓库不存在，开始克隆...")
|
||
self.runtime_path.mkdir(parents=True, exist_ok=True)
|
||
|
||
cmd = f"git clone --recurse-submodules {self.main_repo_url} {repo_path}"
|
||
if not self.run_command(cmd, cwd=self.runtime_path):
|
||
self.logger.error("克隆主仓库失败")
|
||
return False
|
||
|
||
self.logger.info("主仓库克隆成功")
|
||
else:
|
||
self.logger.info("主仓库已存在，更新代码...")
|
||
|
||
# 切换到指定分支
|
||
if not self.run_command(f"git checkout {self.main_repo_branch}", cwd=repo_path):
|
||
return False
|
||
|
||
# 拉取最新代码
|
||
if not self.run_command("git pull", cwd=repo_path):
|
||
return False
|
||
|
||
# 更新所有子模块
|
||
if not self.run_command("git submodule update --init --recursive", cwd=repo_path):
|
||
return False
|
||
|
||
self.logger.info("主仓库更新成功")
|
||
|
||
return True
|
||
|
||
def update_submodule(self, repo_config):
|
||
"""更新指定的子模块"""
|
||
repo_path = self.runtime_path / 'a-cloud-all'
|
||
submodule_path = repo_path / repo_config['path']
|
||
|
||
self.logger.info(f"更新子模块: {repo_config['name']}")
|
||
|
||
# 进入子模块目录
|
||
if not submodule_path.exists():
|
||
self.logger.error(f"子模块目录不存在: {submodule_path}")
|
||
return False
|
||
|
||
# 切换到指定分支
|
||
branch = repo_config['branch']
|
||
if not self.run_command(f"git checkout {branch}", cwd=submodule_path):
|
||
return False
|
||
|
||
# 拉取最新代码
|
||
if not self.run_command("git pull origin " + branch, cwd=submodule_path):
|
||
return False
|
||
|
||
self.logger.info(f"子模块更新成功: {repo_config['name']}")
|
||
return True
|
||
|
||
def build_project(self, repo_config):
|
||
"""构建项目"""
|
||
repo_path = self.runtime_path / 'a-cloud-all'
|
||
submodule_path = repo_path / repo_config['path']
|
||
|
||
self.logger.info(f"开始构建: {repo_config['name']}")
|
||
|
||
# 执行构建命令
|
||
for cmd in repo_config['build_commands']:
|
||
self.logger.info(f"执行构建命令: {cmd}")
|
||
if not self.run_command(cmd, cwd=submodule_path, timeout=1800):
|
||
self.logger.error(f"构建失败: {cmd}")
|
||
return False
|
||
|
||
self.logger.info(f"构建成功: {repo_config['name']}")
|
||
return True
|
||
|
||
def copy_artifacts(self, repo_config):
|
||
"""复制构建产物到 docker 目录"""
|
||
repo_path = self.runtime_path / 'a-cloud-all'
|
||
submodule_path = repo_path / repo_config['path']
|
||
|
||
self.logger.info(f"复制构建产物: {repo_config['name']}")
|
||
|
||
# 获取构建产物路径
|
||
artifact_pattern = submodule_path / repo_config['artifact_path']
|
||
artifacts = glob.glob(str(artifact_pattern))
|
||
|
||
if not artifacts:
|
||
self.logger.error(f"未找到构建产物: {artifact_pattern}")
|
||
return False
|
||
|
||
# 目标目录
|
||
docker_path = repo_path / repo_config['docker_path']
|
||
docker_path.mkdir(parents=True, exist_ok=True)
|
||
|
||
# 复制文件
|
||
for artifact in artifacts:
|
||
artifact_path = Path(artifact)
|
||
if artifact_path.is_file():
|
||
dest = docker_path / artifact_path.name
|
||
shutil.copy2(artifact, dest)
|
||
self.logger.info(f"复制文件: {artifact_path.name}")
|
||
elif artifact_path.is_dir():
|
||
# 如果是目录（如 dist），清空目标目录后复制
|
||
if docker_path.exists():
|
||
for item in docker_path.iterdir():
|
||
if item.name != '.gitkeep':
|
||
if item.is_dir():
|
||
shutil.rmtree(item)
|
||
else:
|
||
item.unlink()
|
||
shutil.copytree(artifact, docker_path, dirs_exist_ok=True)
|
||
self.logger.info(f"复制目录: {artifact_path.name}")
|
||
|
||
self.logger.info("构建产物复制完成")
|
||
return True
|
||
|
||
def run_deploy_script(self, repo_config):
|
||
"""执行部署脚本"""
|
||
repo_path = self.runtime_path / 'a-cloud-all'
|
||
script_name = repo_config['deploy_script']
|
||
script_path = repo_path / '.devops' / 'scripts' / script_name
|
||
|
||
if not script_path.exists():
|
||
self.logger.error(f"部署脚本不存在: {script_path}")
|
||
return False
|
||
|
||
self.logger.info(f"执行部署脚本: {script_name}")
|
||
|
||
# 准备脚本参数
|
||
docker_service = repo_config.get('docker_service', '')
|
||
docker_compose_path = self.config['deploy']['docker_compose_path']
|
||
|
||
# 执行脚本
|
||
cmd = f"bash {script_path} {repo_config['name']} {docker_service} {docker_compose_path}"
|
||
|
||
if not self.run_command(cmd, cwd=repo_path, timeout=600):
|
||
self.logger.error("部署脚本执行失败")
|
||
return False
|
||
|
||
self.logger.info("部署脚本执行成功")
|
||
return True
|
||
|
||
def commit_submodule_update(self, repo_config):
|
||
"""提交子模块更新到主仓库"""
|
||
if not self.config['deploy'].get('auto_commit', False):
|
||
self.logger.info("自动提交已禁用，跳过")
|
||
return True
|
||
|
||
repo_path = self.runtime_path / 'a-cloud-all'
|
||
|
||
self.logger.info("提交子模块更新到主仓库")
|
||
|
||
# 添加子模块更改
|
||
submodule_path = repo_config['path']
|
||
if not self.run_command(f"git add {submodule_path}", cwd=repo_path):
|
||
return False
|
||
|
||
# 检查是否有更改
|
||
result = subprocess.run(
|
||
"git diff --cached --quiet",
|
||
shell=True,
|
||
cwd=repo_path
|
||
)
|
||
|
||
if result.returncode == 0:
|
||
self.logger.info("没有需要提交的更改")
|
||
return True
|
||
|
||
# 提交更改
|
||
commit_msg = self.config['deploy']['commit_message'].format(
|
||
repo_name=repo_config['name']
|
||
)
|
||
|
||
if not self.run_command(f'git commit -m "{commit_msg}"', cwd=repo_path):
|
||
return False
|
||
|
||
# 推送到远程
|
||
if not self.run_command(f"git push origin {self.main_repo_branch}", cwd=repo_path):
|
||
self.logger.warning("推送失败，但部署已完成")
|
||
return True
|
||
|
||
self.logger.info("子模块更新已提交并推送")
|
||
return True
|
||
|
||
def deploy(self, repo_config):
|
||
"""执行完整的部署流程"""
|
||
self.logger.info(f"=" * 60)
|
||
self.logger.info(f"开始部署: {repo_config['name']}")
|
||
self.logger.info(f"=" * 60)
|
||
|
||
try:
|
||
# 1. 确保主仓库存在
|
||
if not self.ensure_main_repo():
|
||
return False
|
||
|
||
# 2. 部署基础设施（首次部署）
|
||
if not self.deploy_infrastructure():
|
||
return False
|
||
|
||
# 3. 更新子模块
|
||
if not self.update_submodule(repo_config):
|
||
return False
|
||
|
||
# 3. 构建项目
|
||
if not self.build_project(repo_config):
|
||
return False
|
||
|
||
# 4. 复制构建产物
|
||
if not self.copy_artifacts(repo_config):
|
||
return False
|
||
|
||
# 5. 执行部署脚本
|
||
if not self.run_deploy_script(repo_config):
|
||
return False
|
||
|
||
# 6. 提交子模块更新
|
||
if not self.commit_submodule_update(repo_config):
|
||
return False
|
||
|
||
self.logger.info(f"部署完成: {repo_config['name']}")
|
||
return True
|
||
|
||
except Exception as e:
|
||
self.logger.error(f"部署过程中发生异常: {e}", exc_info=True)
|
||
return False
|
||
|
||
def deploy_infrastructure(self):
|
||
"""部署基础设施服务（只部署一次）"""
|
||
if 'infrastructure' not in self.config:
|
||
return True
|
||
|
||
repo_path = self.runtime_path / 'a-cloud-all'
|
||
|
||
for infra in self.config['infrastructure']:
|
||
name = infra['name']
|
||
deployed_flag = repo_path / infra['deployed_flag']
|
||
|
||
# 检查是否已部署
|
||
if deployed_flag.exists():
|
||
self.logger.info(f"基础设施 {name} 已部署，跳过")
|
||
continue
|
||
|
||
self.logger.info(f"部署基础设施: {name}")
|
||
|
||
# 执行预部署命令
|
||
if 'pre_deploy_commands' in infra:
|
||
for cmd in infra['pre_deploy_commands']:
|
||
if not self.run_command(cmd, cwd=repo_path):
|
||
self.logger.error(f"预部署命令失败: {cmd}")
|
||
return False
|
||
|
||
# 部署服务
|
||
docker_service = infra['docker_service']
|
||
docker_dir = repo_path / 'docker'
|
||
|
||
cmd = f"docker-compose build --no-cache {docker_service} && docker-compose up -d {docker_service}"
|
||
if not self.run_command(cmd, cwd=docker_dir, timeout=1800):
|
||
self.logger.error(f"部署失败: {name}")
|
||
return False
|
||
|
||
# 创建部署标记
|
||
deployed_flag.parent.mkdir(parents=True, exist_ok=True)
|
||
deployed_flag.touch()
|
||
|
||
self.logger.info(f"基础设施部署完成: {name}")
|
||
|
||
return True
|