新增GPU部分

67-202508-雄安空能院/关停计划/启动脚本.sh

@@ -0,0 +1,186 @@
+#!/bin/bash
+#
+# Nginx及业务环境恢复脚本
+# Author: AI Assistant
+# Version: 1.0.0
+# License: MIT
+#
+# 全局常量定义
+readonly LOG_FILE="/root/wdd/ccc.log"
+readonly NGINX_CONF_BACKUP="/etc/nginx/conf.d/real-proxy.conf_back"
+readonly NGINX_CONF_LIVE="/etc/nginx/conf.d/real-proxy.conf"
+readonly OFFLINE_CONF_LIVE="/etc/nginx/conf.d/offline.conf"
+readonly OFFLINE_CONF_BACKUP="/etc/nginx/conf.d/offline.conf_back"
+readonly HARBOR_COMPOSE_FILE="/root/wdd/harbor/docker-compose.yml"
+readonly K8S_INGRESS_FILE="/root/wdd/install/k8s-ingress-nginx.yaml"
+readonly K8S_DEPLOYMENT_FILE="/root/wdd/all-deployment-xakny.yaml"
+
+# 依赖命令检查列表
+readonly REQUIRED_COMMANDS=("systemctl" "docker-compose" "kubectl" "mv" "sleep")
+
+# 初始化脚本执行环境
+set -euo pipefail
+trap 'log ERROR "脚本被中断"; exit 130' INT TERM
+
+###
+# 分级日志记录函数
+# @param level string 日志级别（DEBUG/INFO/WARN/ERROR）
+# @param message string 日志消息内容
+# @return 无返回值
+# @require 无外部依赖
+###
+log() {
+  local level="$1"
+  local message="$2"
+  local timestamp=$(date '+%Y-%m-%d %H:%M:%S')
+  local log_entry="[${timestamp}] ${level}: ${message}"
+
+  # 输出到控制台和日志文件
+  echo "${log_entry}" | tee -a "${LOG_FILE}" >&2
+}
+
+###
+# 检查命令依赖是否存在
+# @param 无参数
+# @return 0-检查通过 1-缺少依赖
+# @require 无外部依赖
+###
+check_dependencies() {
+  for cmd in "${REQUIRED_COMMANDS[@]}"; do
+    if ! command -v "${cmd}" &> /dev/null; then
+      log ERROR "缺少必要命令: ${cmd}"
+      return 1
+    fi
+  done
+  log DEBUG "所有依赖命令检查通过"
+  return 0
+}
+
+###
+# 恢复Nginx配置文件并重启服务
+# @param 无参数
+# @return 0-操作成功 1-操作失败
+# @require systemctl, mv命令
+###
+restore_nginx() {
+  log INFO "开始恢复Nginx配置"
+
+  # > 恢复主配置文件
+  if ! mv "${NGINX_CONF_BACKUP}" "${NGINX_CONF_LIVE}"; then
+    log ERROR "恢复real-proxy.conf失败"
+    return 1
+  fi
+  log DEBUG "成功恢复real-proxy.conf"
+
+  # > 备份离线配置文件
+  if ! mv "${OFFLINE_CONF_LIVE}" "${OFFLINE_CONF_BACKUP}"; then
+    log WARN "备份offline.conf失败（可能文件不存在）"
+  else
+    log DEBUG "成功备份offline.conf"
+  fi
+
+  # > 重启Nginx服务
+  if ! systemctl restart nginx; then
+    log ERROR "Nginx服务重启失败"
+    return 1
+  fi
+  log INFO "Nginx服务重启成功"
+  return 0
+}
+
+###
+# 启动Harbor容器服务
+# @param 无参数
+# @return 0-启动成功 1-启动失败
+# @require docker-compose, sleep命令
+###
+start_harbor() {
+  log INFO "开始启动Harbor服务"
+
+  if ! docker-compose -f "${HARBOR_COMPOSE_FILE}" up -d; then
+    log ERROR "Harbor启动失败"
+    return 1
+  fi
+
+  log INFO "Harbor启动完成，等待30秒初始化..."
+  sleep 30
+  log DEBUG "Harbor初始化等待完成"
+  return 0
+}
+
+###
+# 启用K8S Ingress暴露面
+# @param 无参数
+# @return 0-应用成功 1-应用失败
+# @require kubectl命令
+###
+enable_ingress() {
+  log INFO "开始应用K8S Ingress配置"
+
+  if ! kubectl apply -f "${K8S_INGRESS_FILE}"; then
+    log ERROR "Ingress配置应用失败"
+    return 1
+  fi
+
+  log INFO "Ingress配置应用成功"
+  return 0
+}
+
+###
+# 恢复业务Deployment配置
+# @param 无参数
+# @return 0-应用成功 1-应用失败
+# @require kubectl命令
+###
+restore_business() {
+  log INFO "开始恢复业务Deployment"
+
+  if ! kubectl apply -f "${K8S_DEPLOYMENT_FILE}"; then
+    log ERROR "业务Deployment应用失败"
+    return 1
+  fi
+
+  log INFO "业务Deployment恢复成功"
+  return 0
+}
+
+###
+# 主执行函数
+# @param 无参数
+# @return 0-全部成功 1-任意步骤失败
+# @require 所有子函数依赖
+###
+main() {
+  log INFO "===== 开始执行环境恢复脚本 ====="
+
+  # > 检查命令依赖
+  if ! check_dependencies; then
+    log ERROR "依赖检查失败，脚本终止"
+    return 1
+  fi
+
+  # 函数执行序列
+  local steps=(
+    restore_nginx
+    start_harbor
+    enable_ingress
+    restore_business
+  )
+
+  for step in "${steps[@]}"; do
+    if ! ${step}; then
+      log ERROR "执行步骤 ${step} 失败"
+      return 1
+    fi
+  done
+
+  log INFO "===== 环境恢复脚本执行完成 ====="
+  return 0
+}
+
+# 执行主函数并捕获退出状态
+if main; then
+  exit 0
+else
+  exit 1
+fi