更新版本到2.0

start.sh

@@ -0,0 +1,176 @@
+#!/bin/bash
+#  启动/停止/重启脚本
+
+# ===================== 配置区 =====================
+# 程序路径
+AGENT_PATH="./dns-server"
+# 日志文件路径
+LOG_FILE="./server.log"
+# PID文件路径（记录进程ID）
+PID_FILE="./server.pid"
+# 启动参数（根据实际需求调整）
+START_ARGS=""
+# 工作目录
+WORK_DIR="."
+# ==================== 配置区结束 ====================
+
+# 检查程序文件是否存在
+check_agent_exists() {
+    if [ ! -f "${AGENT_PATH}" ]; then
+        echo "错误：程序文件 ${AGENT_PATH} 不存在！"
+        exit 1
+    fi
+    if [ ! -x "${AGENT_PATH}" ]; then
+        echo "错误：程序文件 ${AGENT_PATH} 没有执行权限，正在尝试添加..."
+        chmod +x "${AGENT_PATH}"
+        if [ $? -ne 0 ]; then
+            echo "错误：添加执行权限失败，请手动执行 chmod +x ${AGENT_PATH}"
+            exit 1
+        fi
+    fi
+}
+
+# 检查进程是否运行
+check_running() {
+    if [ -f "${PID_FILE}" ]; then
+        PID=$(cat "${PID_FILE}")
+        if ps -p "${PID}" > /dev/null 2>&1; then
+            return 0  # 运行中
+        else
+            rm -f "${PID_FILE}"  # PID文件存在但进程已死，清理PID文件
+        fi
+    fi
+    return 1  # 未运行
+}
+
+# 启动程序
+start_agent() {
+    if check_running; then
+        echo "✅ dns-server 已在运行（PID: $(cat ${PID_FILE})）"
+        return 0
+    fi
+
+    echo "🚀 正在启动 dns-server（工作目录：${WORK_DIR}）..."
+    
+    # 新增：检查并切换工作目录
+    if [ ! -d "${WORK_DIR}" ]; then
+        echo "⚠️ 工作目录 ${WORK_DIR} 不存在，正在创建..."
+        mkdir -p "${WORK_DIR}"
+        if [ $? -ne 0 ]; then
+            echo "❌ 创建工作目录 ${WORK_DIR} 失败！"
+            exit 1
+        fi
+    fi
+    
+    # 切换到工作目录（关键：程序将在此目录下运行）
+    cd "${WORK_DIR}" || {
+        echo "❌ 切换到工作目录 ${WORK_DIR} 失败！"
+        exit 1
+    }
+
+    # 创建日志目录
+    mkdir -p "$(dirname ${LOG_FILE})"
+    # 后台启动程序（注意：cd仅影响当前子进程，需在同一行执行）
+    nohup "${AGENT_PATH}" ${START_ARGS} > "${LOG_FILE}" 2>&1 &
+    AGENT_PID=$!
+    echo "${AGENT_PID}" > "${PID_FILE}"
+
+    # 等待检查启动状态
+    sleep 2
+    if check_running; then
+        echo "✅ dns-server 启动成功（PID: ${AGENT_PID}，工作目录：${WORK_DIR}）"
+        echo "日志文件：${LOG_FILE}"
+    else
+        echo "❌ dns-server 启动失败！请查看日志：${LOG_FILE}"
+        rm -f "${PID_FILE}"
+        exit 1
+    fi
+}
+# 停止程序
+stop_agent() {
+    if ! check_running; then
+        echo "ℹ️ dns-server 未运行"
+        return 0
+    fi
+
+    PID=$(cat "${PID_FILE}")
+    echo "🛑 正在停止 dns-server（PID: ${PID}）..."
+    # 优雅停止（先尝试TERM信号，失败则强制KILL）
+    kill "${PID}" > /dev/null 2>&1
+    sleep 3
+
+    if ps -p "${PID}" > /dev/null 2>&1; then
+        echo "⚠️ 优雅停止失败，强制杀死进程..."
+        kill -9 "${PID}" > /dev/null 2>&1
+        sleep 1
+    fi
+
+    # 清理PID文件
+    rm -f "${PID_FILE}"
+    echo "✅ dns-server 已停止"
+}
+
+# 查看状态
+status_agent() {
+    if check_running; then
+        echo "✅ dns-server 运行中（PID: $(cat ${PID_FILE})）"
+    else
+        echo "ℹ️ dns-server 未运行"
+    fi
+}
+
+# 重启程序
+restart_agent() {
+    echo "🔄 正在重启 dns-server..."
+    stop_agent
+    sleep 2
+    start_agent
+}
+
+# 帮助信息
+show_help() {
+    echo "使用方法：$0 [start|stop|restart|status|help]"
+    echo "  start   - 启动 dns-server"
+    echo "  stop    - 停止 dns-server"
+    echo "  restart - 重启 dns-server"
+    echo "  status  - 查看 dns-server 运行状态"
+    echo "  help    - 显示帮助信息"
+}
+
+# 主逻辑
+main() {
+    # 检查是否为root用户（可选，根据需求调整）
+    if [ "$(id -u)" -ne 0 ]; then
+        echo "警告：建议使用root用户运行此脚本（当前用户：$(whoami)）"
+        # exit 1  # 如果强制要求root，取消注释
+    fi
+
+    check_agent_exists
+
+    case "$1" in
+        start)
+            start_agent
+            ;;
+        stop)
+            stop_agent
+            ;;
+        restart)
+            restart_agent
+            ;;
+        status)
+            status_agent
+            ;;
+        help|--help|-h)
+            show_help
+            ;;
+        *)
+            echo "错误：无效参数 '$1'"
+            show_help
+            exit 1
+            ;;
+    esac
+}
+
+# 执行主逻辑
+main "$@"
+