116 lines
3.5 KiB
C#
116 lines
3.5 KiB
C#
using Core.WcfProtocol;
|
||
using System.Collections.Concurrent;
|
||
|
||
namespace SHH.MjpegPlayer;
|
||
|
||
/// <summary>
|
||
/// 服务器会话集合 (线程安全重构版)
|
||
/// </summary>
|
||
public class MjpegSessions
|
||
{
|
||
// 核心改变：使用字典建立索引，Key = DeviceId#TypeCode
|
||
// 这样可以将查找特定摄像头的复杂度从 O(N) 降低到 O(1)
|
||
private readonly ConcurrentDictionary<string, List<MjpegSession>> _sessionMap
|
||
= new ConcurrentDictionary<string, List<MjpegSession>>();
|
||
|
||
/// <summary>
|
||
/// 构造函数
|
||
/// </summary>
|
||
public MjpegSessions()
|
||
{
|
||
PrismMsg<UploadImageRequest>.Subscribe(ProcUploadImageRequest);
|
||
}
|
||
|
||
/// <summary>
|
||
/// 优化后的图片分发逻辑 (解决 O(N) 和 线程安全问题)
|
||
/// </summary>
|
||
/// <param name="req"></param>
|
||
public void ProcUploadImageRequest(UploadImageRequest req)
|
||
{
|
||
try
|
||
{
|
||
string key = $"{req.Id}#{req.Type}";
|
||
|
||
// 1. 更新通道信息 (ImageChannels 现已是线程安全的)
|
||
var chn = MjpegStatics.ImageChannels.Do(req, key);
|
||
|
||
// 2. O(1) 快速查找关注该流的会话列表
|
||
if (_sessionMap.TryGetValue(key, out var targetSessions))
|
||
{
|
||
// 必须加锁，防止遍历时 List 被其他线程(如 AddSession/RemoveSession) 修改
|
||
lock (targetSessions)
|
||
{
|
||
// 倒序遍历，方便在需要时移除失效会话
|
||
for (var i = targetSessions.Count - 1; i >= 0; i--)
|
||
{
|
||
var session = targetSessions[i];
|
||
session.DoImageProc(req);
|
||
}
|
||
}
|
||
|
||
if (chn != null) chn.IsPlaying = targetSessions.Count > 0;
|
||
}
|
||
else
|
||
{
|
||
if (chn != null) chn.IsPlaying = false;
|
||
}
|
||
}
|
||
catch (Exception ex)
|
||
{
|
||
//Logs.LogWarning<MjpegSessions>(ex.Message, ex.StackTrace);
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 添加会话
|
||
/// </summary>
|
||
/// <param name="session"></param>
|
||
public void AddSession(MjpegSession session)
|
||
{
|
||
if (session?.Info?.Key == null) return;
|
||
|
||
// 使用 GetOrAdd 确保线程安全地获取或创建 List
|
||
var list = _sessionMap.GetOrAdd(session.Info.Key, _ => new List<MjpegSession>());
|
||
|
||
lock (list)
|
||
{
|
||
list.Add(session);
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 移除会话
|
||
/// </summary>
|
||
/// <param name="session"></param>
|
||
public void RemoveSession(MjpegSession session)
|
||
{
|
||
if (session?.Info?.Key == null) return;
|
||
|
||
if (_sessionMap.TryGetValue(session.Info.Key, out var list))
|
||
{
|
||
lock (list)
|
||
{
|
||
list.Remove(session);
|
||
}
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 获取当前所有会话信息的快照 (用于 HTTP API 统计与展示)
|
||
/// [新增] 此方法替代旧版直接访问 Sessions 列表，防止 HTTP 线程与 MJPEG 线程发生冲突
|
||
/// </summary>
|
||
public List<SessionInfo> GetAllSessionInfos()
|
||
{
|
||
var result = new List<SessionInfo>();
|
||
// 遍历字典，线程安全地收集所有 Info
|
||
foreach (var kvp in _sessionMap)
|
||
{
|
||
// 对内部 List 加锁，确保复制过程不被打断
|
||
lock (kvp.Value)
|
||
{
|
||
result.AddRange(kvp.Value.Select(s => s.Info));
|
||
}
|
||
}
|
||
return result;
|
||
}
|
||
} |