增加了通过网络主动上报图像的支持
增加了指令维护通道的支持
This commit is contained in:
@@ -1,40 +1,48 @@
|
||||
using NetMQ;
|
||||
using NetMQ.Sockets;
|
||||
using Newtonsoft.Json;
|
||||
using System;
|
||||
using Newtonsoft.Json.Linq;
|
||||
using System.Collections.Concurrent;
|
||||
using System.Diagnostics;
|
||||
using System.Text;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace SHH.CameraDashboard.Services;
|
||||
namespace SHH.CameraDashboard;
|
||||
|
||||
/// <summary>
|
||||
/// [Dashboard端] 指令控制服务
|
||||
/// 职责:双向通信通道。接收 Service 心跳/响应,向 Service 发送控制指令。
|
||||
/// 核心模式:ROUTER (Dashboard) <--> DEALER (Service)
|
||||
/// 职责:监听 6001 端口,接收 CameraService 的注册/心跳,并下发控制指令。
|
||||
/// </summary>
|
||||
public class CommandServer : IDisposable
|
||||
{
|
||||
// 单例模式
|
||||
public static CommandServer Instance { get; } = new CommandServer();
|
||||
|
||||
// 事件:收到消息时触发 (ServiceId, MessageContent)
|
||||
// =================================================================
|
||||
// 事件定义
|
||||
// =================================================================
|
||||
|
||||
// 当有新设备注册成功时触发 (UI 可以订阅这个来刷新列表)
|
||||
public event Action<ConnectedClient>? OnClientRegistered;
|
||||
|
||||
// 当收到通用业务消息时触发
|
||||
public event Action<string, string>? OnMessageReceived;
|
||||
|
||||
// =================================================================
|
||||
// 内部成员
|
||||
// =================================================================
|
||||
private RouterSocket? _routerSocket;
|
||||
private NetMQPoller? _poller;
|
||||
|
||||
// 【关键新增】发送队列:用于解决跨线程发送的安全问题
|
||||
// UI线程 -> Enqueue -> Poller线程 -> Socket.Send
|
||||
private NetMQQueue<CommandPacket>? _sendQueue;
|
||||
|
||||
// 在线设备表 (Key: Identity/AppId)
|
||||
// 线程安全字典,存储客户端的详细信息(包括视频地址)
|
||||
private readonly ConcurrentDictionary<string, ConnectedClient> _clients = new();
|
||||
|
||||
public int ListenPort { get; private set; }
|
||||
public bool IsRunning => _poller != null && _poller.IsRunning;
|
||||
|
||||
// 在线设备表 (可选,用于记录谁在线)
|
||||
// Key: ServiceId (Identity字符串)
|
||||
private readonly ConcurrentDictionary<string, DateTime> _onlineClients = new();
|
||||
// 获取当前所有在线客户端的副本
|
||||
public List<ConnectedClient> GetClients() => _clients.Values.ToList();
|
||||
|
||||
private CommandServer() { }
|
||||
|
||||
@@ -47,17 +55,15 @@ public class CommandServer : IDisposable
|
||||
{
|
||||
// 1. 初始化 Router Socket
|
||||
_routerSocket = new RouterSocket();
|
||||
_routerSocket.Bind($"tcp://*:{ListenPort}");
|
||||
_routerSocket.Bind($"tcp://*:{ListenPort}"); // 监听所有网卡
|
||||
_routerSocket.ReceiveReady += OnSocketReady;
|
||||
|
||||
// 2. 初始化发送队列
|
||||
// 2. 初始化发送队列 (确保 UI 线程可以安全发送)
|
||||
_sendQueue = new NetMQQueue<CommandPacket>();
|
||||
_sendQueue.ReceiveReady += OnQueueReady;
|
||||
|
||||
// 3. 启动 Poller (同时监听 Socket 接收 和 队列发送)
|
||||
// 3. 启动 Poller
|
||||
_poller = new NetMQPoller { _routerSocket, _sendQueue };
|
||||
|
||||
// RunAsync 会自动开启后台线程
|
||||
_poller.RunAsync();
|
||||
|
||||
Console.WriteLine($"[Dashboard] 指令服务启动,监听: tcp://*:{ListenPort}");
|
||||
@@ -65,33 +71,37 @@ public class CommandServer : IDisposable
|
||||
catch (Exception ex)
|
||||
{
|
||||
Console.WriteLine($"[Dashboard] 指令端口绑定失败: {ex.Message}");
|
||||
throw; // 必须抛出,让 App 感知
|
||||
throw;
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 处理来自 Service 的网络消息 (运行在 Poller 线程)
|
||||
/// [Poller线程] 处理网络接收
|
||||
/// </summary>
|
||||
private void OnSocketReady(object? sender, NetMQSocketEventArgs e)
|
||||
{
|
||||
try
|
||||
{
|
||||
// 1. 读取身份帧 (Identity)
|
||||
// 只要 Service 端 DealerSocket 设置了 Identity,这里收到就是那个 ID
|
||||
// Router 接收逻辑:
|
||||
// Frame 1: 发送者的 Identity (NetMQ 自动处理)
|
||||
// Frame 2: 真实数据
|
||||
|
||||
// 1. 读取身份 (Identity)
|
||||
var identityBytes = e.Socket.ReceiveFrameBytes();
|
||||
string serviceId = Encoding.UTF8.GetString(identityBytes);
|
||||
string serviceId = Encoding.UTF8.GetString(identityBytes); // e.g., "CameraApp_01"
|
||||
|
||||
// 2. 读取内容帧 (假设 Dealer 直接发内容,中间无空帧)
|
||||
// 如果你使用了 REQ/REP 模式,中间可能会有空帧,需注意兼容
|
||||
// 2. 读取消息内容
|
||||
// 兼容性处理:有些 Dealer 实现可能会发空帧,这里做个简单尝试
|
||||
// 如果发现在 Identity 后紧跟的是空帧,则再读一帧
|
||||
// 但在我们目前的 Dealer 实现中,是直接发的 JSON
|
||||
string message = e.Socket.ReceiveFrameString();
|
||||
if (string.IsNullOrWhiteSpace(message))
|
||||
{
|
||||
if (e.Socket.HasIn) message = e.Socket.ReceiveFrameString();
|
||||
}
|
||||
|
||||
// 3. 简单的心跳保活逻辑
|
||||
_onlineClients[serviceId] = DateTime.Now;
|
||||
|
||||
// 4. 触发业务事件
|
||||
// 注意:这依然在 Poller 线程,UI 处理时需 Invoke
|
||||
Console.WriteLine($"[指令] From {serviceId}: {message}");
|
||||
OnMessageReceived?.Invoke(serviceId, message);
|
||||
// 3. 协议解析与业务分发
|
||||
ProcessMessage(serviceId, message, identityBytes);
|
||||
}
|
||||
catch (Exception ex)
|
||||
{
|
||||
@@ -100,37 +110,99 @@ public class CommandServer : IDisposable
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 处理发送队列 (运行在 Poller 线程)
|
||||
/// 核心业务逻辑处理
|
||||
/// </summary>
|
||||
private void OnQueueReady(object? sender, NetMQQueueEventArgs<CommandPacket> e)
|
||||
private void ProcessMessage(string serviceId, string json, byte[] identityBytes)
|
||||
{
|
||||
try
|
||||
{
|
||||
if (_routerSocket == null) return;
|
||||
// 尝试解析基础结构
|
||||
var jObj = JObject.Parse(json);
|
||||
string action = jObj["Action"]?.ToString() ?? "Unknown";
|
||||
|
||||
// 从队列取出一个包
|
||||
if (e.Queue.TryDequeue(out var packet, TimeSpan.Zero))
|
||||
// 更新最后心跳时间 (如果已存在)
|
||||
if (_clients.TryGetValue(serviceId, out var existingClient))
|
||||
{
|
||||
// Router 发送标准三步走:
|
||||
// 1. 发送目标 Identity (More = true)
|
||||
// 2. 发送空帧 (可选,取决于协议约定,Router-Dealer 直连通常不需要空帧)
|
||||
// 3. 发送数据 (More = false)
|
||||
existingClient.LastHeartbeat = DateTime.Now;
|
||||
}
|
||||
|
||||
// 这里我们采用最简协议:[Identity][Data]
|
||||
_routerSocket.SendMoreFrame(packet.TargetId)
|
||||
.SendFrame(packet.JsonData);
|
||||
|
||||
Console.WriteLine($"[指令] To {packet.TargetId}: {packet.JsonData}");
|
||||
// ★★★ 处理注册握手 ★★★
|
||||
if (action == "Register")
|
||||
{
|
||||
HandleRegistration(serviceId, jObj, identityBytes);
|
||||
}
|
||||
else
|
||||
{
|
||||
// 其他业务消息,透传给上层
|
||||
Console.WriteLine($"[指令] From {serviceId}: {json}");
|
||||
OnMessageReceived?.Invoke(serviceId, json);
|
||||
}
|
||||
}
|
||||
catch (Exception ex)
|
||||
catch (JsonException)
|
||||
{
|
||||
Debug.WriteLine($"[Command Send Error] {ex.Message}");
|
||||
Console.WriteLine($"[指令] 收到非 JSON 消息 From {serviceId}: {json}");
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 发送指令 (线程安全,可由 UI 线程调用)
|
||||
/// 处理注册逻辑
|
||||
/// </summary>
|
||||
private void HandleRegistration(string serviceId, JObject jObj, byte[] identityBytes)
|
||||
{
|
||||
var payload = jObj["Payload"];
|
||||
if (payload == null) return;
|
||||
|
||||
// 1. 提取客户端信息
|
||||
var client = new ConnectedClient
|
||||
{
|
||||
ServiceId = serviceId,
|
||||
Ip = payload["Ip"]?.ToString() ?? "Unknown",
|
||||
|
||||
// ★★★ 解析新字段 ★★★
|
||||
WebPort = payload["WebPort"]?.Value<int>() ?? 5000,
|
||||
Version = payload["Version"]?.ToString() ?? "Unknown",
|
||||
Pid = payload["Pid"]?.Value<int>() ?? 0,
|
||||
|
||||
TargetVideoNodes = payload["TargetVideoNodes"]?.ToObject<List<string>>() ?? new List<string>(),
|
||||
LastHeartbeat = DateTime.Now
|
||||
};
|
||||
|
||||
// 2. 存入内存表 (Add or Update)
|
||||
_clients.AddOrUpdate(serviceId, client, (key, old) => client);
|
||||
|
||||
Console.WriteLine($"[注册成功] {serviceId}");
|
||||
|
||||
// 3. 回复 ACK (握手确认)
|
||||
// 告诉客户端:我收到你的注册了,连接建立成功
|
||||
var ackPacket = new { Action = "ACK", Message = $"Registered {serviceId}", Time = DateTime.Now };
|
||||
string ackJson = JsonConvert.SerializeObject(ackPacket);
|
||||
|
||||
// 直接在 Poller 线程发回,不需要走 Queue (因为拥有 Socket 所有权)
|
||||
_routerSocket?.SendMoreFrame(identityBytes).SendFrame(ackJson);
|
||||
|
||||
// 4. 通知 UI 更新列表
|
||||
OnClientRegistered?.Invoke(client);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// [Poller线程] 处理发送队列
|
||||
/// </summary>
|
||||
private void OnQueueReady(object? sender, NetMQQueueEventArgs<CommandPacket> e)
|
||||
{
|
||||
if (_routerSocket == null) return;
|
||||
|
||||
if (e.Queue.TryDequeue(out var packet, TimeSpan.Zero))
|
||||
{
|
||||
// Router 发送:[Identity] [Data]
|
||||
_routerSocket.SendMoreFrame(packet.TargetId)
|
||||
.SendFrame(packet.JsonData);
|
||||
|
||||
Console.WriteLine($"[发送] To {packet.TargetId}: {packet.JsonData}");
|
||||
}
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// [公共API] 向指定 Service 发送指令
|
||||
/// </summary>
|
||||
public void SendCommand(string targetServiceId, object commandData)
|
||||
{
|
||||
@@ -138,7 +210,6 @@ public class CommandServer : IDisposable
|
||||
|
||||
var json = JsonConvert.SerializeObject(commandData);
|
||||
|
||||
// ★★★ 核心修复:不直接操作 Socket,而是入队 ★★★
|
||||
_sendQueue.Enqueue(new CommandPacket
|
||||
{
|
||||
TargetId = targetServiceId,
|
||||
@@ -152,13 +223,12 @@ public class CommandServer : IDisposable
|
||||
_poller?.Dispose();
|
||||
_routerSocket?.Dispose();
|
||||
_sendQueue?.Dispose();
|
||||
|
||||
_poller = null;
|
||||
_routerSocket = null;
|
||||
_sendQueue = null;
|
||||
}
|
||||
|
||||
// 内部数据包结构
|
||||
// =============================================================
|
||||
// 数据模型
|
||||
// =============================================================
|
||||
|
||||
private class CommandPacket
|
||||
{
|
||||
public string TargetId { get; set; } = "";
|
||||
|
||||
Reference in New Issue
Block a user