71 lines
2.4 KiB
C#
71 lines
2.4 KiB
C#
using System;
|
||
using NetMQ;
|
||
using NetMQ.Sockets;
|
||
using SHH.Contracts;
|
||
|
||
namespace SHH.NetMQ
|
||
{
|
||
/// <summary>
|
||
/// 视频分发服务端 (Publisher)
|
||
/// 特性：非阻塞、防内存溢出
|
||
/// </summary>
|
||
public class DistributorServer : IDisposable
|
||
{
|
||
private PublisherSocket _pubSocket;
|
||
private readonly object _lock = new object();
|
||
|
||
// 配置：高水位限制 (HWM)
|
||
// 假设 25fps，设置 50 意味着内存只缓存 2 秒的视频。
|
||
// 如果断网超过 2 秒，新来的视频帧直接丢弃，优先保证恢复后的实时性。
|
||
private const int HWM_LIMIT = 50;
|
||
|
||
public DistributorServer(string connectionString)
|
||
{
|
||
_pubSocket = new PublisherSocket();
|
||
|
||
// 1. 设置发送缓冲区大小 (防爆内存关键)
|
||
_pubSocket.Options.SendHighWatermark = HWM_LIMIT;
|
||
|
||
// 2. 绑定地址 (如 tcp://*:5555)
|
||
_pubSocket.Bind(connectionString);
|
||
}
|
||
|
||
public void Broadcast(VideoPayload payload)
|
||
{
|
||
if (payload == null) return;
|
||
|
||
// 补充发送时间
|
||
payload.DispatchTime = DateTime.Now;
|
||
|
||
// 准备数据帧
|
||
string jsonMeta = payload.GetMetadataJson();
|
||
byte[] rawBytes = payload.OriginalImageBytes ?? new byte[0];
|
||
byte[] targetBytes = payload.TargetImageBytes ?? new byte[0];
|
||
|
||
// 使用 NetMQMessage 封装多帧消息
|
||
// 这样比手动调三次 Send 更容易管理原子性
|
||
var msg = new NetMQMessage();
|
||
msg.Append(jsonMeta); // 第1帧
|
||
msg.Append(rawBytes); // 第2帧
|
||
msg.Append(targetBytes); // 第3帧
|
||
|
||
lock (_lock)
|
||
{
|
||
// 3. 非阻塞发送 (核心防卡死代码)
|
||
// TimeSpan.Zero 表示：如果缓冲区满了或者发不出去，立即放弃，不等待，返回 false
|
||
// 这样你的主线程（海康SDK回调）永远不会被卡住
|
||
bool sent = _pubSocket.TrySendMultipartMessage(TimeSpan.Zero, msg);
|
||
|
||
if (!sent)
|
||
{
|
||
// 这里可以打个日志：Console.WriteLine("警告：网络拥堵，丢帧中...");
|
||
}
|
||
}
|
||
}
|
||
|
||
public void Dispose()
|
||
{
|
||
_pubSocket?.Dispose();
|
||
}
|
||
}
|
||
} |