使用 inotifywait 监控文件并自动重启 Go 应用的教程

#!/usr/bin/env bash

WATCH_DIR=$1
FILENAME=$2

function restart_goserver() {
  if go run $FILENAME
  then
    pkill -9 -f $FILENAME > /dev/null 2>&1
    pkill -9 -f a.out > /dev/null 2>&1
    go run $FILENAME &
    echo "started $FILENAME"
  else
    echo "server restart failed"
  fi
}

cd $WATCH_DIR
restart_goserver

echo "watching directory: $WATCH_DIR"
inotifywait -mrq -e close_write $WATCH_DIR | while read file
do
  if grep -E '^(.*\.go)|(.*\.html)$'
  then
    echo "--------------------"
    restart_goserver
  fi
done

# ...
inotifywait -mrq -e close_write $WATCH_DIR | while read file
do
  # 将文件路径通过 echo 传递给 grep 进行匹配
  if echo "$file" | grep -E '^(.*\.go)|(.*\.html)$' > /dev/null
  then
    echo "--------------------"
    restart_goserver
  fi
done

function restart_goserver() {
  # 尝试编译Go程序，如果编译失败则不重启
  echo "Attempting to build $FILENAME..."
  if go build -o app_server $FILENAME # 编译为可执行文件 app_server
  then
    # 查找并发送 SIGTERM 信号给之前启动的进程
    # 使用 pgrep 结合进程名或命令行参数来精确查找进程
    # 注意：pkill -f "$FILENAME" 可能会误杀其他同名进程，更推荐使用 pid 文件或更精确的 pgrep 模式
    # 这里为了演示，我们假设 go run $FILENAME 或 ./app_server 是唯一的

    # 尝试查找之前由 go run 启动的进程
    pkill -15 -f "go run $FILENAME" > /dev/null 2>&1
    # 尝试查找之前编译的 app_server 进程
    pkill -15 -f "./app_server" > /dev/null 2>&1

    # 等待一小段时间，让进程有机会清理并退出
    sleep 1

    echo "Previous server instances terminated (if any)."

    # 启动新的编译后的应用程序
    ./app_server & # 运行编译后的可执行文件
    echo "Started new instance of $FILENAME (as app_server)."
  else
    echo "Server build failed for $FILENAME. Not restarting."
  fi
}

#!/usr/bin/env bash

# 脚本用途：监控指定目录下的 .go 或 .html 文件变更，并自动重启 Go 应用程序。
# 用法：./gowatcher.sh <监控目录> <Go应用主文件>
# 示例：./gowatcher.sh /path/to/my/project main.go

WATCH_DIR=$1
FILENAME=$2
APP_EXEC_NAME="app_server" # 编译后的可执行文件名

# 检查参数是否提供
if [ -z "$WATCH_DIR" ] || [ -z "$FILENAME" ]; then
  echo "Usage: $0 <directory_to_watch> <golang_main_file>"
  exit 1
fi

# 确保监控目录存在
if [ ! -d "$WATCH_DIR" ]; then
  echo "Error: Directory '$WATCH_DIR' does not exist."
  exit 1
fi

# 确保Go应用主文件存在
if [ ! -f "$WATCH_DIR/$FILENAME" ]; then
  echo "Error: Go main file '$WATCH_DIR/$FILENAME' does not exist."
  exit 1
fi

# 函数：重启 Go 服务器
function restart_goserver() {
  echo "--------------------"
  echo "Attempting to build and restart $FILENAME..."

  # 1. 尝试编译Go程序
  # 将编译后的可执行文件放在 WATCH_DIR 目录下
  if go build -o "$WATCH_DIR/$APP_EXEC_NAME" "$WATCH_DIR/$FILENAME"
  then
    # 2. 优雅地终止之前运行的实例
    # 注意：这里使用 pkill -f 可能会终止所有包含 APP_EXEC_NAME 字符串的进程
    # 在生产环境或复杂场景下，建议使用 PID 文件或更精确的进程管理方法
    echo "Sending SIGTERM to previous instances of $APP_EXEC_NAME..."
    pkill -15 -f "$APP_EXEC_NAME" > /dev/null 2>&1

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

    # 3. 启动新的编译后的应用程序实例
    echo "Starting new instance of $APP_EXEC_NAME..."
    # 使用 nohup 和 & 确保脚本退出后应用也能继续运行，并重定向输出
    nohup "$WATCH_DIR/$APP_EXEC_NAME" > "$WATCH_DIR/app.log" 2>&1 &
    echo "New instance of $APP_EXEC_NAME started. Output redirected to $WATCH_DIR/app.log."
  else
    echo "Go application build failed for $FILENAME. Server not restarted."
  fi
}

# 切换到监控目录，方便 go build 和运行
cd "$WATCH_DIR" || { echo "Failed to change directory to $WATCH_DIR"; exit 1; }

# 首次启动服务器
restart_goserver

echo "Watching directory: $WATCH_DIR for changes in .go or .html files..."

# 使用 inotifywait 监控文件变更
inotifywait -mrq -e close_write "$WATCH_DIR" | while read -r event_path event_type filename
do
  # 检查文件名是否符合 .go 或 .html 模式
  if echo "$filename" | grep -E '\.(go|html)$' > /dev/null
  then
    echo "Detected change in: $filename (Event: $event_type)"
    restart_goserver
  fi
done