利用inotifywait实现Go项目自动重载与热部署

#!/usr/bin/env bash

WATCH_DIR=$1
FILENAME=$2 # 通常是Go主源文件，例如 main.go

function restart_goserver() {
  echo "尝试重启 $FILENAME..."
  # 潜在问题1：这里应该先停止旧服务，再启动新服务。
  # 并且 if go run $FILENAME 这样的判断是错误的，它会阻塞直到服务退出。
  if go run "$FILENAME" # 错误用法，会阻塞
  then
    pkill -9 -f "$FILENAME" > /dev/null 2>&1 # 潜在问题2：直接使用 kill -9
    pkill -9 -f a.out > /dev/null 2>&1 # 潜在问题3：a.out 通常不适用于 go run
    go run "$FILENAME" & # 启动服务到后台
    echo "已启动 $FILENAME"
  else
    echo "服务器重启失败"
  fi
}

cd "$WATCH_DIR" || { echo "无法切换到目录 $WATCH_DIR"; exit 1; }
restart_goserver # 首次启动服务

echo "正在监控目录: $WATCH_DIR"
inotifywait -mrq -e close_write "$WATCH_DIR" | while read -r event_path event_name
do
  # 潜在问题4：grep 没有输入
  if grep -E '^(.*\.go)|(.*\.html)$'
  then
    echo "--------------------"
    echo "检测到文件变化: $event_name"
    restart_goserver
  fi
done

# 原始错误
# if grep -E '^(.*\.go)|(.*\.html)$'

# 修正后的代码
if echo "$event_name" | grep -E '\.(go|html)$' > /dev/null
then
  # ... 执行重启逻辑
fi

# 优雅终止进程函数
function kill_existing_server() {
  local target_filename="$1"
  echo "尝试优雅关闭旧进程 ($target_filename)..."

  # 尝试发送 SIGTERM (默认信号)
  pkill -f "$target_filename"

  # 等待一段时间，给进程清理的机会
  sleep 1

  # 检查进程是否仍在运行，如果仍在运行则强制杀死
  if pgrep -f "$target_filename" > /dev/null; then
    echo "进程仍在运行，强制关闭 ($target_filename)..."
    pkill -9 -f "$target_filename"
    sleep 1 # 再次等待，确保进程终止
  fi
}

function restart_goserver() {
  local filename_to_run="$1"
  echo "--------------------"
  echo "尝试重启服务: $filename_to_run"

  # 1. 停止旧进程
  kill_existing_server "$filename_to_run"

  # 2. 启动新进程
  # go run 命令通常会将标准输出和标准错误输出到控制台，
  # 如果需要更安静的后台运行，可以重定向输出。
  # 这里为了调试方便，暂时不重定向。
  go run "$filename_to_run" & # 在后台启动服务

  # 检查新服务是否成功启动 (通过检查进程是否存在)
  sleep 0.5 # 给予Go程序一些时间来启动
  if pgrep -f "$filename_to_run" > /dev/null; then
    echo "服务 $filename_to_run 已成功启动。"
  else
    echo "错误: 服务 $filename_to_run 启动失败。请检查Go程序日志。"
  fi
}

#!/usr/bin/env bash

# 检查参数
if [ -z "$1" ] || [ -z "$2" ]; then
  echo "用法: $0 <监控目录> <Go主源文件>"
  echo "示例: $0 /path/to/my/project main.go"
  exit 1
fi

WATCH_DIR="$1"
FILENAME="$2" # 例如: main.go

# 确保监控目录存在
if [ ! -d "$WATCH_DIR" ]; then
  echo "错误: 监控目录 '$WATCH_DIR' 不存在。"
  exit 1
fi

# 确保Go主源文件存在于监控目录中
if [ ! -f "$WATCH_DIR/$FILENAME" ]; then
  echo "错误: Go主源文件 '$FILENAME' 在目录 '$WATCH_DIR' 中不存在。"
  exit 1
fi

# 优雅终止进程函数
function kill_existing_server() {
  local target_filename="$1"
  echo "尝试优雅关闭旧进程 ($target_filename)..."

  # 尝试发送 SIGTERM (默认信号)
  pkill -f "$target_filename"

  # 等待一段时间，给进程清理的机会
  sleep 1

  # 检查进程是否仍在运行，如果仍在运行则强制杀死
  if pgrep -f "$target_filename" > /dev/null; then
    echo "进程仍在运行，强制关闭 ($target_filename)..."
    pkill -9 -f "$target_filename"
    sleep 1 # 再次等待，确保进程终止
  fi
}

# 重启Go服务器函数
function restart_goserver() {
  local filename_to_run="$1"
  echo "--------------------"
  echo "尝试重启服务: $filename_to_run"

  # 1. 停止旧进程
  kill_existing_server "$filename_to_run"

  # 2. 启动新进程
  # 注意: go run 命令会在当前目录执行，所以需要先cd到WATCH_DIR
  # 将Go程序的标准输出和标准错误重定向到 /dev/null，以保持终端整洁。
  # 如果需要查看Go程序的输出，可以重定向到日志文件。
  (cd "$WATCH_DIR" && go run "$filename_to_run" &> /dev/null &)

  # 检查新服务是否成功启动 (通过检查进程是否存在)
  sleep 0.5 # 给予Go程序一些时间来启动
  if pgrep -f "$filename_to_run" > /dev/null; then
    echo "服务 $filename_to_run 已成功启动。"
  else
    echo "错误: 服务 $filename_to_run 启动失败。请检查Go程序日志或手动运行调试。"
  fi
}

# 首次启动服务
restart_goserver "$FILENAME"

echo "正在监控目录: $WATCH_DIR"
# inotifywait -mrq -e close_write 监控目录及其子目录下的文件写入关闭事件
inotifywait -mrq -e close_write "$WATCH_DIR" | while read -r event_path event_name
do
  # 过滤 .go 或 .html 文件
  if echo "$event_name" | grep -E '\.(go|html)$' > /dev/null
  then
    echo "检测到文件变化: $event_path$event_name"
    restart_goserver "$FILENAME"
  fi
done