first commit

connect/heartbeat.go

@@ -0,0 +1,88 @@
+package connect
+
+import (
+	"fmt"
+	"os"
+	"os/exec"
+	"sync"
+	"time"
+
+	"FireLeave_tool/logger"
+)
+
+var (
+	InferenceProcesses sync.Map // key: deviceUUID string, value: *exec.Cmd
+
+	// 记录最后一次收到心跳的时间
+	inferenceLastHeartbeat sync.Map // key: deviceUUID string, value: time.Time
+)
+
+// UpdateInferenceHeartbeat 每收到一帧 /video/post 就调用这个
+func UpdateInferenceHeartbeat(deviceUUID string) {
+	inferenceLastHeartbeat.Store(deviceUUID, time.Now())
+
+}
+
+// ShouldRestartInference 判断是否需要重启推理程序
+func ShouldRestartInference(deviceUUID string) bool {
+	val, ok := inferenceLastHeartbeat.Load(deviceUUID)
+	if !ok {
+		return true // 从未收到过心跳，肯定要启动
+	}
+	last := val.(time.Time)
+	return time.Since(last) > 85*time.Second
+}
+
+// StopInferenceProcess 停止推理进程（供 main 和 DeviceManager 调用）
+func StopInferenceProcess(deviceUUID string) {
+	if raw, ok := InferenceProcesses.Load(deviceUUID); ok {
+		if cmd, ok := raw.(*exec.Cmd); ok && cmd.Process != nil {
+			logger.Logger.Printf("正在停止推理进程 (DeviceUUID=%s, PID=%d)", deviceUUID, cmd.Process.Pid)
+			cmd.Process.Kill()
+			cmd.Wait()
+		}
+		InferenceProcesses.Delete(deviceUUID)
+	}
+}
+
+// StartInferenceProcess 重启推理程序（保持原端口！）
+func StartInferenceProcess(deviceData *DeviceData) {
+
+	StopInferenceProcess(deviceData.DeviceUUID)
+
+	port, err := GlobalPortManager.GetPort(deviceData.DeviceUUID)
+	if err != nil {
+		logger.Logger.Printf("获取端口失败，无法重启推理程序 (DeviceUUID=%s): %v", deviceData.DeviceUUID, err)
+		return
+	}
+
+	detectAreaStr := ProcessDetectAreaForInference(deviceData.DetectArea)
+
+	inferenceCmd := exec.Command(
+		"./yolov5",
+		"-s", deviceData.CameraRTSP,
+		"-m", "model",
+		"-c", deviceData.DeviceUUID,
+		"-p", fmt.Sprintf("%d", port),
+		"-w", detectAreaStr,
+		"-r", fmt.Sprintf("%d", deviceData.Confidence),
+	)
+
+	inferenceCmd.Stdout = os.Stdout
+	inferenceCmd.Stderr = os.Stderr
+
+	logger.Logger.Printf("正在重启推理程序 (使用原端口 %d): %v", port, inferenceCmd.Args)
+
+	if err := inferenceCmd.Start(); err != nil {
+		logger.Logger.Printf("推理程序重启失败 (DeviceUUID=%s): %v", deviceData.DeviceUUID, err)
+		return
+	}
+
+	// 保存进程对象
+	InferenceProcesses.Store(deviceData.DeviceUUID, inferenceCmd)
+	logger.Logger.Printf("推理程序已成功重启，PID=%d (DeviceUUID=%s, Port=%d)",
+		inferenceCmd.Process.Pid, deviceData.DeviceUUID, port)
+
+	// 重启后立即刷新一次心跳，防止 30s 检测又触发
+	UpdateInferenceHeartbeat(deviceData.DeviceUUID)
+}