commit

go_backend/monitor_service.sh

@@ -0,0 +1,124 @@
+#!/bin/bash
+
+# 服务监控脚本 - 用于外部监控服务状态
+# 可以配合cron定时任务使用
+
+# 配置
+SERVICE_NAME="AI小红书服务"
+ALERT_PHONE="15707023967"
+HEARTBEAT_FILE="/tmp/ai_xhs_service_heartbeat.json"
+CHECK_INTERVAL=120  # 检查间隔（秒），心跳超过这个时间未更新则认为服务宕机
+MONITOR_DIR="$(cd "$(dirname "$0")" && pwd)"
+
+# 颜色输出
+RED='\033[0;31m'
+GREEN='\033[0;32m'
+YELLOW='\033[1;33m'
+NC='\033[0m' # No Color
+
+log_info() {
+    echo -e "${GREEN}[INFO]${NC} $(date '+%Y-%m-%d %H:%M:%S') $1"
+}
+
+log_warn() {
+    echo -e "${YELLOW}[WARN]${NC} $(date '+%Y-%m-%d %H:%M:%S') $1"
+}
+
+log_error() {
+    echo -e "${RED}[ERROR]${NC} $(date '+%Y-%m-%d %H:%M:%S') $1"
+}
+
+# 检查心跳文件是否存在
+check_heartbeat_file() {
+    if [ ! -f "$HEARTBEAT_FILE" ]; then
+        log_error "心跳文件不存在: $HEARTBEAT_FILE"
+        return 1
+    fi
+    return 0
+}
+
+# 获取最后心跳时间
+get_last_heartbeat() {
+    if ! check_heartbeat_file; then
+        echo "0"
+        return
+    fi
+    
+    # 从JSON文件中提取last_heartbeat时间
+    last_heartbeat=$(grep -o '"last_heartbeat":"[^"]*"' "$HEARTBEAT_FILE" | cut -d'"' -f4)
+    
+    if [ -z "$last_heartbeat" ]; then
+        echo "0"
+        return
+    fi
+    
+    # 转换为Unix时间戳
+    heartbeat_timestamp=$(date -d "$last_heartbeat" +%s 2>/dev/null)
+    if [ $? -ne 0 ]; then
+        echo "0"
+        return
+    fi
+    
+    echo "$heartbeat_timestamp"
+}
+
+# 检查服务是否运行
+check_service_status() {
+    log_info "开始检查服务状态..."
+    
+    last_heartbeat_ts=$(get_last_heartbeat)
+    
+    if [ "$last_heartbeat_ts" = "0" ]; then
+        log_error "无法获取心跳信息"
+        return 1
+    fi
+    
+    current_ts=$(date +%s)
+    time_diff=$((current_ts - last_heartbeat_ts))
+    
+    log_info "距离上次心跳: ${time_diff}秒"
+    
+    if [ $time_diff -gt $CHECK_INTERVAL ]; then
+        log_error "服务可能已宕机（超过${CHECK_INTERVAL}秒未更新心跳）"
+        return 1
+    else
+        log_info "服务运行正常"
+        return 0
+    fi
+}
+
+# 发送宕机通知
+send_alert() {
+    log_warn "尝试发送宕机通知..."
+    
+    # 调用Go程序发送通知
+    cd "$MONITOR_DIR"
+    go run test_service_alert.go
+    
+    if [ $? -eq 0 ]; then
+        log_info "宕机通知发送成功"
+        return 0
+    else
+        log_error "宕机通知发送失败"
+        return 1
+    fi
+}
+
+# 主函数
+main() {
+    echo "========================================"
+    echo "服务监控检查 - $(date '+%Y-%m-%d %H:%M:%S')"
+    echo "========================================"
+    
+    if ! check_service_status; then
+        log_error "检测到服务异常"
+        send_alert
+        exit 1
+    fi
+    
+    log_info "服务状态正常"
+    exit 0
+}
+
+# 运行主函数
+main