Files
KinginfoGateway/framework/Foundation/ThingsGateway.Foundation/TouchSocket/Dmtp/Components/Udp/UdpDmtp.cs
2023-10-13 19:16:12 +08:00

127 lines
4.7 KiB
C#
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

#region copyright
//------------------------------------------------------------------------------
// 此代码版权声明为全文件覆盖,如有原作者特别声明,会在下方手动补充
// 此代码版权除特别声明外的代码归作者本人Diego所有
// 源代码使用协议遵循本仓库的开源协议及附加协议
// Gitee源代码仓库https://gitee.com/diego2098/ThingsGateway
// Github源代码仓库https://github.com/kimdiego2098/ThingsGateway
// 使用文档https://diego2098.gitee.io/thingsgateway-docs/
// QQ群605534569
//------------------------------------------------------------------------------
#endregion
//------------------------------------------------------------------------------
// 此代码版权除特别声明或在XREF结尾的命名空间的代码归作者本人若汝棋茗所有
// 源代码使用协议遵循本仓库的开源协议及附加协议若本仓库没有设置则按MIT开源协议授权
// CSDN博客https://blog.csdn.net/qq_40374647
// 哔哩哔哩视频https://space.bilibili.com/94253567
// Gitee源代码仓库https://gitee.com/RRQM_Home
// Github源代码仓库https://github.com/RRQM
// API首页http://rrqm_home.gitee.io/touchsocket/
// 交流QQ群234762506
// 感谢您的下载和使用
//------------------------------------------------------------------------------
//------------------------------------------------------------------------------
using System.Collections.Concurrent;
using System.Net;
namespace ThingsGateway.Foundation.Dmtp.Rpc
{
/// <summary>
/// UdpDmtpService
/// </summary>
public partial class UdpDmtp : UdpSessionBase, IUdpDmtp
{
private readonly Timer m_timer;
private readonly ConcurrentDictionary<EndPoint, UdpDmtpClient> m_udpDmtpClients = new ConcurrentDictionary<EndPoint, UdpDmtpClient>();
/// <summary>
/// 构造函数
/// </summary>
public UdpDmtp()
{
this.m_timer = new Timer((obj) =>
{
//this.m_udpDmtpClients.RemoveWhen((kv) =>
//{
// if (DateTime.Now - kv.Value.LastActiveTime > TimeSpan.FromMinutes(1))
// {
// return true;
// }
// return false;
//});
}, null, 1000 * 10, 1000 * 10);
}
/// <inheritdoc/>
public IDmtpActor DmtpActor => this.PrivateGetUdpDmtpClient().DmtpActor;
/// <summary>
/// 通过终结点获取<see cref="IUdpDmtpClient"/>
/// </summary>
/// <param name="endPoint"></param>
/// <returns></returns>
public IUdpDmtpClient GetUdpDmtpClient(EndPoint endPoint)
{
return this.PrivateGetUdpDmtpClient(endPoint);
}
/// <inheritdoc/>
protected override void Dispose(bool disposing)
{
this.m_timer.SafeDispose();
this.m_udpDmtpClients.Clear();
base.Dispose(disposing);
}
/// <inheritdoc/>
protected override async Task ReceivedData(UdpReceivedDataEventArgs e)
{
var client = this.PrivateGetUdpDmtpClient(e.EndPoint);
if (client == null)
{
return;
}
var message = DmtpMessage.CreateFrom(e.ByteBlock);
if (!client.InputReceivedData(message))
{
if (this.PluginsManager.Enable)
{
await this.PluginsManager.RaiseAsync(nameof(IDmtpReceivedPlugin.OnDmtpReceived), client, new DmtpMessageEventArgs(message));
}
}
}
/// <inheritdoc/>
protected override void LoadConfig(TouchSocketConfig config)
{
base.LoadConfig(config);
}
private UdpDmtpClient PrivateGetUdpDmtpClient(EndPoint endPoint)
{
if (!this.m_udpDmtpClients.TryGetValue(endPoint, out var udpRpcActor))
{
udpRpcActor = new UdpDmtpClient(this, endPoint, this.Container.Resolve<ILog>())
{
Client = this,
};
if (udpRpcActor.Created(this.PluginsManager))
{
this.m_udpDmtpClients.TryAdd(endPoint, udpRpcActor);
}
}
return udpRpcActor;
}
private IUdpDmtpClient PrivateGetUdpDmtpClient()
{
if (this.RemoteIPHost == null)
{
throw new ArgumentNullException(nameof(this.RemoteIPHost));
}
return this.PrivateGetUdpDmtpClient(this.RemoteIPHost.EndPoint);
}
}
}