bug修复

backend-single/deploy.py

@@ -2,14 +2,13 @@
 # -*- coding: utf-8 -*-
 """
 情绪博物馆后端服务部署脚本
-支持本地部署和远程部署到服务器
+部署到远程服务器 101.200.208.45
 使用系统自带的ssh/scp命令，无需额外依赖
 """
 
 import os
 import sys
 import subprocess
-import time
 from pathlib import Path
 
 # 配置变量
@@ -21,11 +20,6 @@ SPRING_PROFILE = "test"
 # 本地路径
 SCRIPT_DIR = Path(__file__).parent.absolute()
 JAR_PATH = SCRIPT_DIR / "target" / JAR_NAME
-LOG_DIR = SCRIPT_DIR / "logs"
-PID_FILE = Path(f"/tmp/{APP_NAME}.pid")
-
-# Java配置
-JAVA_OPTS = "-Xms512m -Xmx1024m -XX:+UseG1GC -XX:+HeapDumpOnOutOfMemoryError"
 
 # 远程服务器配置
 REMOTE_HOST = "101.200.208.45"
@@ -61,13 +55,7 @@ def run_command(cmd, cwd=None, shell=True, capture=True):
     """执行本地命令"""
     try:
         if capture:
-            result = subprocess.run(
-                cmd,
-                cwd=cwd,
-                shell=shell,
-                capture_output=True,
-                text=True
-            )
+            result = subprocess.run(cmd, cwd=cwd, shell=shell, capture_output=True, text=True)
             return result.returncode == 0, result.stdout, result.stderr
         else:
             result = subprocess.run(cmd, cwd=cwd, shell=shell)
@@ -95,23 +83,19 @@ def build_project():
     """构建项目"""
     log_info("开始构建项目...")
     
-    # 检查Maven
     success, _, _ = run_command("mvn --version")
     if not success:
         log_error("未找到Maven命令，请确保已安装Maven")
         sys.exit(1)
     
-    # 执行Maven构建
     log_info("执行: mvn clean package -DskipTests")
     os.chdir(SCRIPT_DIR)
     
-    # 不捕获输出，直接显示构建过程
     success, _, _ = run_command("mvn clean package -DskipTests", capture=False)
     if not success:
         log_error("项目构建失败")
         sys.exit(1)
     
-    # 检查JAR文件
     if not JAR_PATH.exists():
         log_error(f"项目构建失败，未找到JAR文件: {JAR_PATH}")
         sys.exit(1)
@@ -130,27 +114,24 @@ def check_jar():
     log_info(f"JAR文件检查通过: {JAR_PATH}")
 
 
-def deploy_to_remote(upload_script=None):
+def deploy(upload_script=None):
     """
-    远程部署到服务器
+    部署到远程服务器
     
     Args:
         upload_script: 可选，指定要上传的额外文件路径（如 deploy-server.sh）
     """
-    log_info(f"开始远程部署到 {REMOTE_HOST}...")
+    log_info(f"开始部署到 {REMOTE_HOST}...")
     
-    # 构建项目
     build_project()
-    
-    # 检查JAR文件
     check_jar()
     
-    # 1. 创建远程目录
+    # 创建远程目录
     log_info("创建远程目录...")
     exec_ssh_cmd(f"mkdir -p {REMOTE_DIR}")
     exec_ssh_cmd(f"mkdir -p {REMOTE_LOG_DIR}")
     
-    # 2. 上传JAR文件
+    # 上传JAR文件
     log_info("上传JAR文件到远程服务器...")
     log_info(f"本地文件: {JAR_PATH}")
     log_info(f"远程路径: {REMOTE_USER}@{REMOTE_HOST}:{REMOTE_DIR}/{REMOTE_JAR_NAME}")
@@ -160,7 +141,7 @@ def deploy_to_remote(upload_script=None):
         sys.exit(1)
     log_info("✅ JAR文件上传成功")
     
-    # 3. 验证远程文件
+    # 验证远程文件
     log_info("验证远程文件...")
     success, output, _ = exec_ssh_cmd(f"ls -lh {REMOTE_DIR}/{REMOTE_JAR_NAME}")
     if not success:
@@ -168,26 +149,22 @@ def deploy_to_remote(upload_script=None):
         sys.exit(1)
     log_info(output)
     
-    # 4. 如果指定了额外文件，则上传
+    # 上传额外文件
     if upload_script:
         script_path = SCRIPT_DIR / upload_script
         if script_path.exists():
-            log_info(f"上传指定文件到远程服务器: {upload_script}")
+            log_info(f"上传文件到远程服务器: {upload_script}")
             if not scp_upload(script_path, f"{REMOTE_DIR}/{upload_script}"):
                 log_error(f"上传文件失败: {upload_script}")
                 sys.exit(1)
             log_info(f"✅ 文件上传成功: {upload_script}")
-            
-            # 如果是脚本文件，设置执行权限
             if upload_script.endswith('.sh'):
                 exec_ssh_cmd(f"chmod +x {REMOTE_DIR}/{upload_script}")
         else:
             log_error(f"指定的文件不存在: {script_path}")
             sys.exit(1)
-    else:
-        log_info("跳过部署脚本上传（服务器已存在）")
     
-    # 5. 在远程服务器上执行部署
+    # 执行远程部署
     log_info("在远程服务器上执行部署...")
     success, output, error = exec_ssh_cmd(f"cd {REMOTE_DIR} && ./deploy-server.sh {SPRING_PROFILE}")
     if output:
@@ -199,155 +176,24 @@ def deploy_to_remote(upload_script=None):
         log_error("远程部署脚本执行失败")
         sys.exit(1)
     
-    log_info("✅ 远程部署完成！")
-    show_remote_status()
+    log_info("✅ 部署完成！")
+    show_status()
 
 
-def show_remote_status():
+def show_status():
     """显示远程服务状态"""
-    log_info("=== 远程服务信息 ===")
+    log_info("=== 服务信息 ===")
     log_info(f"服务器地址: {REMOTE_HOST}")
     log_info(f"部署目录: {REMOTE_DIR}")
     log_info(f"日志目录: {REMOTE_LOG_DIR}")
     log_info(f"Spring Profile: {SPRING_PROFILE}")
     
-    log_info("检查远程服务状态...")
+    log_info("检查服务状态...")
     success, output, _ = exec_ssh_cmd(f"ps aux | grep {REMOTE_JAR_NAME} | grep -v grep")
     if output:
-        log_info(f"远程服务运行中:\n{output}")
+        log_info(f"服务运行中:\n{output}")
     else:
-        log_info("远程服务未运行")
-
-
-def local_deploy():
-    """本地部署"""
-    log_info(f"开始本地部署 {APP_NAME} 服务...")
-    
-    # 构建项目
-    build_project()
-    
-    # 检查JAR文件
-    check_jar()
-    
-    # 创建日志目录
-    LOG_DIR.mkdir(parents=True, exist_ok=True)
-    
-    # 停止旧服务
-    stop_local_service()
-    
-    # 启动新服务
-    start_local_service()
-    
-    # 等待启动
-    if wait_for_startup():
-        log_info("✅ 本地部署成功！")
-        show_local_status()
-    else:
-        log_error("本地部署失败！")
-        sys.exit(1)
-
-
-def stop_local_service():
-    """停止本地服务"""
-    if PID_FILE.exists():
-        pid = PID_FILE.read_text().strip()
-        success, _, _ = run_command(f"ps -p {pid}")
-        if success:
-            log_info(f"停止旧服务 (PID: {pid})")
-            run_command(f"kill {pid}")
-            
-            # 等待服务停止
-            for _ in range(30):
-                success, _, _ = run_command(f"ps -p {pid}")
-                if not success:
-                    log_info("服务已停止")
-                    break
-                time.sleep(1)
-            else:
-                log_warn(f"强制停止服务 (PID: {pid})")
-                run_command(f"kill -9 {pid}")
-        else:
-            log_warn("PID文件存在但进程不存在，清理PID文件")
-        
-        PID_FILE.unlink()
-    else:
-        log_info("没有找到PID文件，服务可能未运行")
-
-
-def start_local_service():
-    """启动本地服务"""
-    log_info("启动本地服务...")
-    
-    startup_log = LOG_DIR / "startup.log"
-    app_log = LOG_DIR / "application.log"
-    
-    cmd = (
-        f"nohup java {JAVA_OPTS} "
-        f"-Dspring.profiles.active={SPRING_PROFILE} "
-        f"-Dlogging.file.path={LOG_DIR} "
-        f"-Dlogging.file.name={app_log} "
-        f"-jar {JAR_PATH} "
-        f"> {startup_log} 2>&1 &"
-    )
-    
-    # 使用subprocess启动后台进程
-    subprocess.Popen(
-        cmd,
-        shell=True,
-        cwd=SCRIPT_DIR,
-        stdout=subprocess.DEVNULL,
-        stderr=subprocess.DEVNULL
-    )
-    
-    # 获取实际的Java进程PID
-    time.sleep(2)
-    success, output, _ = run_command(f"pgrep -f {JAR_NAME}")
-    if success and output:
-        pid = output.strip().split('\n')[0]
-        PID_FILE.write_text(pid)
-        log_info(f"服务启动中，PID: {pid}")
-    else:
-        log_warn("无法获取服务PID")
-    
-    log_info(f"启动日志: {startup_log}")
-    log_info(f"应用日志: {app_log}")
-
-
-def wait_for_startup():
-    """等待服务启动"""
-    log_info("等待服务启动...")
-    
-    for _ in range(60):
-        # 检查端口是否监听
-        success, output, _ = run_command("netstat -tlnp 2>/dev/null | grep ':19089.*LISTEN'")
-        if success and output:
-            log_info("服务启动成功！")
-            return True
-        time.sleep(2)
-    
-    log_error(f"服务启动超时，请检查日志: {LOG_DIR}/startup.log")
-    return False
-
-
-def show_local_status():
-    """显示本地服务状态"""
-    log_info("=== 本地服务信息 ===")
-    log_info(f"应用名称: {APP_NAME}")
-    log_info(f"JAR文件: {JAR_PATH}")
-    log_info(f"日志目录: {LOG_DIR}")
-    log_info(f"PID文件: {PID_FILE}")
-    log_info(f"Java参数: {JAVA_OPTS}")
-    log_info(f"Spring Profile: {SPRING_PROFILE}")
-    
-    if PID_FILE.exists():
-        pid = PID_FILE.read_text().strip()
-        success, _, _ = run_command(f"ps -p {pid}")
-        if success:
-            log_info(f"服务状态: 运行中 (PID: {pid})")
-        else:
-            log_info("服务状态: 未运行")
-    else:
-        log_info("服务状态: 未运行")
+        log_info("服务未运行")
 
 
 def print_usage():
@@ -356,57 +202,40 @@ def print_usage():
 用法: python deploy.py [命令] [参数]
 
 命令:
-  deploy              - 本地部署服务（默认）
-  remote              - 远程部署到服务器（仅上传JAR）
-  remote [文件名]     - 远程部署并上传指定文件（如 deploy-server.sh）
-  build               - 构建项目
-  start               - 启动本地服务
-  stop                - 停止本地服务
-  restart             - 重启本地服务
-  status              - 查看本地服务状态
-  remote-status       - 查看远程服务状态
+  deploy              - 部署到远程服务器（默认）
+  deploy [文件名]     - 部署并上传指定文件（如 deploy-server.sh）
+  build               - 仅构建项目
+  status              - 查看远程服务状态
 
 示例:
-  python deploy.py remote                    # 仅上传JAR并部署
-  python deploy.py remote deploy-server.sh  # 同时上传部署脚本
+  python deploy.py                         # 部署到远程服务器
+  python deploy.py deploy-server.sh        # 同时上传部署脚本
+  python deploy.py build                   # 仅构建项目
+  python deploy.py status                  # 查看服务状态
 """)
 
 
 def main():
     """主函数"""
-    command = sys.argv[1] if len(sys.argv) > 1 else "deploy"
+    if len(sys.argv) < 2:
+        deploy()
+        return
     
-    if command == "deploy":
-        local_deploy()
-    elif command == "remote":
-        # 检查是否有额外的文件参数
-        upload_script = sys.argv[2] if len(sys.argv) > 2 else None
-        deploy_to_remote(upload_script)
-    elif command == "build":
+    command = sys.argv[1]
+    
+    if command == "build":
         build_project()
-    elif command == "start":
-        build_project()
-        check_jar()
-        LOG_DIR.mkdir(parents=True, exist_ok=True)
-        start_local_service()
-        wait_for_startup()
-    elif command == "stop":
-        stop_local_service()
-    elif command == "restart":
-        stop_local_service()
-        time.sleep(2)
-        build_project()
-        check_jar()
-        LOG_DIR.mkdir(parents=True, exist_ok=True)
-        start_local_service()
-        wait_for_startup()
     elif command == "status":
-        show_local_status()
-    elif command == "remote-status":
-        show_remote_status()
-    else:
+        show_status()
+    elif command == "help" or command == "-h" or command == "--help":
         print_usage()
-        sys.exit(1)
+    elif command.endswith('.sh'):
+        # 如果第一个参数是文件名，则上传该文件
+        deploy(command)
+    else:
+        # 其他情况视为部署命令
+        upload_script = sys.argv[2] if len(sys.argv) > 2 else None
+        deploy(upload_script)
 
 
 if __name__ == "__main__":