86 lines
3.4 KiB
C#
86 lines
3.4 KiB
C#
#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
|
||
// 感谢您的下载和使用
|
||
//------------------------------------------------------------------------------
|
||
//------------------------------------------------------------------------------
|
||
|
||
namespace ThingsGateway.Foundation.Dmtp
|
||
{
|
||
/// <summary>
|
||
/// TcpDmtpService
|
||
/// </summary>
|
||
public class TcpDmtpService : TcpDmtpService<TcpDmtpSocketClient>
|
||
{
|
||
}
|
||
|
||
/// <summary>
|
||
/// TcpDmtpService泛型类型
|
||
/// </summary>
|
||
/// <typeparam name="TClient"></typeparam>
|
||
public partial class TcpDmtpService<TClient> : TcpService<TClient>, ITcpDmtpService where TClient : TcpDmtpSocketClient, new()
|
||
{
|
||
#region 字段
|
||
|
||
private bool m_allowRoute;
|
||
private Func<string, IDmtpActor> m_findDmtpActor;
|
||
|
||
#endregion 字段
|
||
|
||
/// <summary>
|
||
/// 连接令箭
|
||
/// </summary>
|
||
public string VerifyToken => this.Config?.GetValue(DmtpConfigExtension.VerifyTokenProperty);
|
||
|
||
/// <inheritdoc/>
|
||
protected override void LoadConfig(TouchSocketConfig config)
|
||
{
|
||
config.SetTcpDataHandlingAdapter(() => new TcpDmtpAdapter());
|
||
base.LoadConfig(config);
|
||
|
||
if (this.Container.IsRegistered(typeof(IDmtpRouteService)))
|
||
{
|
||
this.m_allowRoute = true;
|
||
this.m_findDmtpActor = this.Container.Resolve<IDmtpRouteService>().FindDmtpActor;
|
||
}
|
||
}
|
||
|
||
/// <summary>
|
||
/// 客户端请求连接
|
||
/// </summary>
|
||
/// <param name="socketClient"></param>
|
||
/// <param name="e"></param>
|
||
protected override async Task OnConnecting(TClient socketClient, ConnectingEventArgs e)
|
||
{
|
||
socketClient.SetDmtpActor(new SealedDmtpActor(this.m_allowRoute)
|
||
{
|
||
Id = e.Id,
|
||
OnFindDmtpActor = this.m_allowRoute ? (this.m_findDmtpActor ?? this.OnServiceFindDmtpActor) : null
|
||
});
|
||
await base.OnConnecting(socketClient, e);
|
||
}
|
||
|
||
private IDmtpActor OnServiceFindDmtpActor(string id)
|
||
{
|
||
return this.TryGetSocketClient(id, out var client) ? client.DmtpActor : null;
|
||
}
|
||
}
|
||
} |