feat(v2): add realtime command and release tooling

2026-07-22 21:28:57 +08:00
parent a98005e631
commit 536396f373
9 changed files with 367 additions and 10 deletions
@@ -0,0 +1,181 @@
+package main
+
+import (
+	"context"
+	"crypto/sha256"
+	"flag"
+	"fmt"
+	"io"
+	"log"
+	"os"
+	"path/filepath"
+	"strings"
+	"sync"
+	"time"
+
+	"silverpose/v2/internal/alert"
+	"silverpose/v2/internal/config"
+	"silverpose/v2/internal/fall"
+	"silverpose/v2/internal/monitor"
+	"silverpose/v2/internal/source"
+	"silverpose/v2/internal/spike"
+	"silverpose/v2/internal/ui"
+)
+
+func main() {
+	configPath := flag.String("config", "config.local.json", "local V2 JSON configuration without RTSP credentials")
+	flag.Parse()
+	cfg, err := config.Load(*configPath)
+	if err != nil {
+		log.Fatal("配置无效：", err)
+	}
+	if err := validateStartup(cfg); err != nil {
+		log.Fatal("启动预检失败：", err)
+	}
+	controller := &controller{config: cfg}
+	window, err := ui.NewWindow(settingsFor(cfg), ui.Commands{Start: controller.Start, Stop: controller.Stop})
+	if err != nil {
+		log.Fatal("创建操作窗口失败：", err)
+	}
+	controller.window = window
+	window.Run()
+	controller.Stop()
+}
+
+type controller struct {
+	config config.Config
+	window *ui.Window
+	mu     sync.Mutex
+	cancel func()
+	runID  uint64
+}
+
+func (controller *controller) Start() {
+	controller.mu.Lock()
+	if controller.cancel != nil {
+		controller.mu.Unlock()
+		return
+	}
+	ctx, cancel := context.WithCancel(context.Background())
+	controller.cancel = cancel
+	controller.runID++
+	runID := controller.runID
+	controller.mu.Unlock()
+	go controller.run(ctx, cancel, runID)
+}
+
+func (controller *controller) Stop() {
+	controller.mu.Lock()
+	cancel := controller.cancel
+	controller.cancel = nil
+	controller.mu.Unlock()
+	if cancel != nil {
+		cancel()
+	}
+}
+
+func (controller *controller) run(ctx context.Context, cancel func(), runID uint64) {
+	defer func() {
+		controller.mu.Lock()
+		if controller.runID == runID {
+			controller.cancel = nil
+		}
+		controller.mu.Unlock()
+		cancel()
+	}()
+	runtime, err := spike.OpenRuntime(controller.config.Model.ONNXPath, controller.config.Model.ONNXRuntimeDLLPath)
+	if err != nil {
+		controller.window.ReportIssue("无法初始化 ONNX Runtime")
+		return
+	}
+	engine, err := fall.NewEngine(fall.EngineConfig{
+		KeypointConfidenceThreshold:     controller.config.Event.KeypointConfidenceThreshold,
+		SuspectWindowSeconds:            controller.config.Event.SuspectWindowSeconds,
+		ConfirmWindowSeconds:            controller.config.Event.ConfirmWindowSeconds,
+		RecoveryWindowSeconds:           controller.config.Event.RecoveryWindowSeconds,
+		CooldownSeconds:                 controller.config.Event.CooldownSeconds,
+		RequireRapidDrop:                controller.config.Event.RequireRapidDrop,
+		RequireLowerBody:                controller.config.Event.RequireLowerBody,
+		HorizontalAngleThresholdDegrees: controller.config.Event.HorizontalAngleThresholdDegrees,
+		ConfigVersion:                   controller.config.RuntimeConfigVersion,
+		SessionID:                       time.Now().UTC().Format("20060102-150405"),
+	})
+	if err != nil {
+		runtime.Close()
+		controller.window.ReportIssue("无法初始化摔倒事件引擎")
+		return
+	}
+	stream := source.Start(ctx, source.Config{
+		SourceURL: controller.config.Source.URL, FFmpegPath: controller.config.Tools.FFmpegPath,
+		FFprobePath: controller.config.Tools.FFprobePath, Transport: controller.config.Source.Transport,
+		Timeout: controller.config.Source.Timeout, LowLatency: controller.config.Source.LowLatency,
+	}, source.Dependencies{})
+	monitored, err := monitor.New(
+		stream, runtime, engine, controller.config.Model.ConfidenceThreshold,
+		alert.NewDispatcher(controller.config.Artifacts.EventDirectory, controller.config.Source.ID), time.Now,
+	)
+	if err != nil {
+		runtime.Close()
+		controller.window.ReportIssue("无法启动监控管线")
+		return
+	}
+	go monitored.Run(ctx)
+	updates, alerts := monitored.Updates(), monitored.Alerts()
+	for updates != nil || alerts != nil {
+		select {
+		case update, open := <-updates:
+			if !open {
+				updates = nil
+				continue
+			}
+			controller.window.Present(update)
+		case record, open := <-alerts:
+			if !open {
+				alerts = nil
+				continue
+			}
+			controller.window.PresentAlert(record)
+		}
+	}
+}
+
+func validateStartup(cfg config.Config) error {
+	for _, path := range []string{cfg.Model.ONNXPath, cfg.Model.ONNXRuntimeDLLPath, cfg.Tools.FFmpegPath, cfg.Tools.FFprobePath} {
+		info, err := os.Stat(path)
+		if err != nil || info.IsDir() {
+			return fmt.Errorf("必需运行文件缺失或不可读")
+		}
+	}
+	actual, err := sha256File(cfg.Model.ONNXPath)
+	if err != nil {
+		return fmt.Errorf("无法校验 ONNX 模型")
+	}
+	if !strings.EqualFold(actual, cfg.Model.SHA256) {
+		return fmt.Errorf("ONNX 模型哈希与配置不匹配")
+	}
+	return nil
+}
+
+func sha256File(path string) (string, error) {
+	file, err := os.Open(filepath.Clean(path))
+	if err != nil {
+		return "", err
+	}
+	defer file.Close()
+	digest := sha256.New()
+	if _, err := io.Copy(digest, file); err != nil {
+		return "", err
+	}
+	return fmt.Sprintf("%x", digest.Sum(nil)), nil
+}
+
+func settingsFor(cfg config.Config) ui.Settings {
+	return ui.Settings{
+		SourceEnvironment: cfg.Source.RTSPURLEnv,
+		ModelSHA256:       cfg.Model.SHA256,
+		RuntimeSummary:    "ONNX Runtime、FFmpeg 和 FFprobe 已通过启动预检",
+		EventSummary: fmt.Sprintf("确认 %.1f 秒；水平角 %.0f°；快速下移=%t；下肢=%t",
+			cfg.Event.ConfirmWindowSeconds, cfg.Event.HorizontalAngleThresholdDegrees,
+			cfg.Event.RequireRapidDrop, cfg.Event.RequireLowerBody),
+	}
+}
@@ -0,0 +1,22 @@
+package main
+
+import (
+	"strings"
+	"testing"
+
+	"silverpose/v2/internal/config"
+)
+
+func TestValidateStartupRejectsMissingRuntimeWithoutLeakingSource(t *testing.T) {
+	err := validateStartup(config.Config{
+		Source: config.SourceConfig{URL: "rtsp://operator:secret@example/Streaming/Channels/101"},
+		Model:  config.ModelConfig{ONNXPath: "missing-model.onnx", SHA256: strings.Repeat("a", 64), ONNXRuntimeDLLPath: "missing-runtime.dll"},
+		Tools:  config.ToolConfig{FFmpegPath: "missing-ffmpeg.exe", FFprobePath: "missing-ffprobe.exe"},
+	})
+	if err == nil {
+		t.Fatal("expected missing runtime error")
+	}
+	if strings.Contains(err.Error(), "rtsp://") || strings.Contains(err.Error(), "secret") {
+		t.Fatalf("startup error leaked source: %v", err)
+	}
+}