91 lines
2.9 KiB
C#
91 lines
2.9 KiB
C#
|
using SHH.CameraSdk;
|
|||
|
|
using System.Collections.Concurrent;
|
|||
|
|
|
|||
|
|
namespace SHH.CameraService;
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 网络推流管理器
|
|||
|
|
/// 职责：管理 ZeroMQ 推流任务的生命周期
|
|||
|
|
/// 类似于 DisplayWindowManager，它负责订阅数据并将其桥接到传输层
|
|||
|
|
/// </summary>
|
|||
|
|
public class NetworkStreamManager
|
|||
|
|
{
|
|||
|
|
private readonly VideoDataChannel _channel;
|
|||
|
|
// 记录当前活跃的推流任务，防止重复订阅
|
|||
|
|
private readonly ConcurrentDictionary<string, bool> _activeStreams = new();
|
|||
|
|
|
|||
|
|
public NetworkStreamManager(VideoDataChannel channel)
|
|||
|
|
{
|
|||
|
|
_channel = channel;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 启动推流任务
|
|||
|
|
/// </summary>
|
|||
|
|
public void StartStream(string appId, long deviceId)
|
|||
|
|
{
|
|||
|
|
// 1. 防止重复启动
|
|||
|
|
if (_activeStreams.ContainsKey(appId)) return;
|
|||
|
|
|
|||
|
|
// 2. 向全局分发器订阅精准数据
|
|||
|
|
// 这里实现了业务逻辑的闭环：只有被 Manager 管理的任务才会消耗 CPU 去转码
|
|||
|
|
GlobalStreamDispatcher.Subscribe(appId, deviceId, (frame) =>
|
|||
|
|
{
|
|||
|
|
// --- 这里的代码运行在分发线程中 ---
|
|||
|
|
|
|||
|
|
// A. 转码 (耗时操作封装在这里，不污染 Controller)
|
|||
|
|
byte[] jpgBytes = EncodeFrameToJpg(frame);
|
|||
|
|
|
|||
|
|
if (jpgBytes != null && jpgBytes.Length > 0)
|
|||
|
|
{
|
|||
|
|
var payload = new VideoPayload
|
|||
|
|
{
|
|||
|
|
CameraId = appId, // 使用 AppId 作为 Topic (给 Dashboard 订阅用)
|
|||
|
|
OriginalImageBytes = jpgBytes,
|
|||
|
|
CaptureTime = DateTime.Now,
|
|||
|
|
OriginalWidth = frame.TargetWidth,
|
|||
|
|
OriginalHeight = frame.TargetHeight
|
|||
|
|
};
|
|||
|
|
|
|||
|
|
// B. 写入传输通道
|
|||
|
|
_ = _channel.WriteAsync(payload);
|
|||
|
|
}
|
|||
|
|
});
|
|||
|
|
|
|||
|
|
_activeStreams.TryAdd(appId, true);
|
|||
|
|
Console.WriteLine($"[Network] 推流任务已启动: {appId} -> Device {deviceId}");
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// 停止推流任务
|
|||
|
|
/// </summary>
|
|||
|
|
public void StopStream(string appId)
|
|||
|
|
{
|
|||
|
|
if (_activeStreams.TryRemove(appId, out _))
|
|||
|
|
{
|
|||
|
|
// 1. 从全局分发器注销
|
|||
|
|
GlobalStreamDispatcher.Unsubscribe(appId);
|
|||
|
|
Console.WriteLine($"[Network] 推流任务已停止: {appId}");
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// --- 辅助方法 ---
|
|||
|
|
private byte[] EncodeFrameToJpg(SmartFrame frame)
|
|||
|
|
{
|
|||
|
|
try
|
|||
|
|
{
|
|||
|
|
// 优先使用处理后的 TargetMat，如果没有则用原始的 InternalMat
|
|||
|
|
var mat = frame.TargetMat ?? frame.InternalMat;
|
|||
|
|
if (mat != null && !mat.Empty())
|
|||
|
|
{
|
|||
|
|
// 80 质量平衡体积与画质
|
|||
|
|
return mat.ImEncode(".jpg", new int[] { 1, 80 });
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
catch (Exception ex)
|
|||
|
|
{
|
|||
|
|
Console.WriteLine($"[Network] 转码失败: {ex.Message}");
|
|||
|
|
}
|
|||
|
|
return Array.Empty<byte>();
|
|||
|
|
}
|
|||
|
|
}
|