mirror of
https://gitee.com/ThingsGateway/ThingsGateway.git
synced 2025-10-28 14:13:59 +08:00
313 lines
10 KiB
C#
313 lines
10 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>
|
||
/// TcpDmtpSocketClient
|
||
/// </summary>
|
||
public partial class TcpDmtpSocketClient : SocketClient, ITcpDmtpSocketClient
|
||
{
|
||
private DmtpActor m_dmtpActor;
|
||
|
||
/// <summary>
|
||
/// TcpDmtpSocketClient
|
||
/// </summary>
|
||
public TcpDmtpSocketClient()
|
||
{
|
||
this.Protocol = DmtpUtility.DmtpProtocol;
|
||
}
|
||
|
||
/// <inheritdoc/>
|
||
public IDmtpActor DmtpActor { get => this.m_dmtpActor; }
|
||
|
||
/// <inheritdoc cref="IDmtpActor.IsHandshaked"/>
|
||
public bool IsHandshaked => this.DmtpActor != null && this.DmtpActor.IsHandshaked;
|
||
|
||
/// <summary>
|
||
/// 验证超时时间,默认为3000ms
|
||
/// </summary>
|
||
public int VerifyTimeout => this.Config.GetValue(DmtpConfigExtension.VerifyTimeoutProperty);
|
||
|
||
/// <summary>
|
||
/// 连接令箭
|
||
/// </summary>
|
||
public string VerifyToken => this.Config.GetValue(DmtpConfigExtension.VerifyTokenProperty);
|
||
|
||
#region 内部委托绑定
|
||
|
||
private Task OnDmtpActorClose(DmtpActor actor, string msg)
|
||
{
|
||
base.BreakOut(false, msg);
|
||
return EasyTask.CompletedTask;
|
||
}
|
||
|
||
private Task OnDmtpActorCreateChannel(DmtpActor actor, CreateChannelEventArgs e)
|
||
{
|
||
return this.OnCreateChannel(e);
|
||
}
|
||
|
||
private Task OnDmtpActorHandshaked(DmtpActor actor, DmtpVerifyEventArgs e)
|
||
{
|
||
return this.OnHandshaked(e);
|
||
}
|
||
|
||
private async Task OnDmtpActorHandshaking(DmtpActor actor, DmtpVerifyEventArgs e)
|
||
{
|
||
if (e.Token == this.VerifyToken)
|
||
{
|
||
e.IsPermitOperation = true;
|
||
}
|
||
else
|
||
{
|
||
e.Message = "Token不受理";
|
||
}
|
||
|
||
await this.OnHandshaking(e).ConfigureFalseAwait();
|
||
}
|
||
|
||
private Task OnDmtpActorRouting(DmtpActor actor, PackageRouterEventArgs e)
|
||
{
|
||
return this.OnRouting(e);
|
||
}
|
||
|
||
#endregion 内部委托绑定
|
||
|
||
#region 事件
|
||
|
||
/// <summary>
|
||
/// 当创建通道
|
||
/// </summary>
|
||
/// <param name="e"></param>
|
||
protected virtual async Task OnCreateChannel(CreateChannelEventArgs e)
|
||
{
|
||
if (e.Handled)
|
||
{
|
||
return;
|
||
}
|
||
|
||
await this.PluginsManager.RaiseAsync(nameof(IDmtpCreateChannelPlugin.OnCreateChannel), this, e).ConfigureFalseAwait();
|
||
}
|
||
|
||
/// <summary>
|
||
/// 在完成握手连接时
|
||
/// </summary>
|
||
/// <param name="e"></param>
|
||
protected virtual async Task OnHandshaked(DmtpVerifyEventArgs e)
|
||
{
|
||
if (e.Handled)
|
||
{
|
||
return;
|
||
}
|
||
await this.PluginsManager.RaiseAsync(nameof(IDmtpHandshakedPlugin.OnDmtpHandshaked), this, e).ConfigureFalseAwait();
|
||
}
|
||
|
||
/// <summary>
|
||
/// 在验证Token时
|
||
/// </summary>
|
||
/// <param name="e">参数</param>
|
||
protected virtual async Task OnHandshaking(DmtpVerifyEventArgs e)
|
||
{
|
||
if (e.Handled)
|
||
{
|
||
return;
|
||
}
|
||
await this.PluginsManager.RaiseAsync(nameof(IDmtpHandshakingPlugin.OnDmtpHandshaking), this, e).ConfigureFalseAwait();
|
||
}
|
||
|
||
/// <summary>
|
||
/// 在需要转发路由包时。
|
||
/// </summary>
|
||
/// <param name="e"></param>
|
||
protected virtual async Task OnRouting(PackageRouterEventArgs e)
|
||
{
|
||
if (e.Handled)
|
||
{
|
||
return;
|
||
}
|
||
await this.PluginsManager.RaiseAsync(nameof(IDmtpRoutingPlugin.OnDmtpRouting), this, e);
|
||
}
|
||
|
||
#endregion 事件
|
||
|
||
#region 断开
|
||
|
||
/// <summary>
|
||
/// 发送<see cref="IDmtpActor"/>关闭消息。
|
||
/// </summary>
|
||
/// <param name="msg"></param>
|
||
/// <returns></returns>
|
||
public override void Close(string msg = "")
|
||
{
|
||
if (this.IsHandshaked)
|
||
{
|
||
this.m_dmtpActor?.SendClose(msg);
|
||
this.m_dmtpActor?.Close(msg);
|
||
}
|
||
base.Close(msg);
|
||
}
|
||
|
||
/// <inheritdoc/>
|
||
protected override void Dispose(bool disposing)
|
||
{
|
||
if (this.IsHandshaked)
|
||
{
|
||
this.m_dmtpActor?.SafeDispose();
|
||
}
|
||
base.Dispose(disposing);
|
||
}
|
||
|
||
/// <inheritdoc/>
|
||
protected override async Task OnDisconnected(DisconnectEventArgs e)
|
||
{
|
||
if (this.IsHandshaked)
|
||
{
|
||
this.m_dmtpActor?.Close(e.Message);
|
||
}
|
||
await base.OnDisconnected(e).ConfigureFalseAwait();
|
||
}
|
||
|
||
#endregion 断开
|
||
|
||
#region ResetId
|
||
|
||
///<inheritdoc cref="IDmtpActor.ResetId(string)"/>
|
||
public override void ResetId(string id)
|
||
{
|
||
this.m_dmtpActor.ResetId(id);
|
||
}
|
||
|
||
///<inheritdoc cref="IDmtpActor.ResetIdAsync(string)"/>
|
||
public Task ResetIdAsync(string newId)
|
||
{
|
||
return this.m_dmtpActor.ResetIdAsync(newId);
|
||
}
|
||
|
||
#endregion ResetId
|
||
|
||
internal void SetDmtpActor(DmtpActor actor)
|
||
{
|
||
actor.IdChanged = this.ThisOnResetId;
|
||
actor.OutputSend = this.ThisDmtpActorOutputSend;
|
||
actor.OutputSendAsync = this.ThisDmtpActorOutputSendAsync;
|
||
actor.Client = this;
|
||
actor.Closed = this.OnDmtpActorClose;
|
||
actor.Routing = this.OnDmtpActorRouting;
|
||
actor.Handshaked = this.OnDmtpActorHandshaked;
|
||
actor.Handshaking = this.OnDmtpActorHandshaking;
|
||
actor.CreatedChannel = this.OnDmtpActorCreateChannel;
|
||
actor.Logger = this.Logger;
|
||
this.m_dmtpActor = actor;
|
||
}
|
||
|
||
/// <inheritdoc/>
|
||
protected override async Task ReceivedData(ReceivedDataEventArgs e)
|
||
{
|
||
var message = (DmtpMessage)e.RequestInfo;
|
||
if (!await this.m_dmtpActor.InputReceivedData(message).ConfigureFalseAwait())
|
||
{
|
||
await this.PluginsManager.RaiseAsync(nameof(IDmtpReceivedPlugin.OnDmtpReceived), this, new DmtpMessageEventArgs(message)).ConfigureFalseAwait();
|
||
}
|
||
await base.ReceivedData(e).ConfigureFalseAwait();
|
||
}
|
||
|
||
/// <inheritdoc/>
|
||
protected override async Task OnConnected(ConnectedEventArgs e)
|
||
{
|
||
this.m_dmtpActor.Id = this.Id;
|
||
await base.OnConnected(e).ConfigureFalseAwait();
|
||
|
||
_ = Task.Run(async () =>
|
||
{
|
||
await Task.Delay(this.VerifyTimeout).ConfigureFalseAwait();
|
||
if (!this.IsHandshaked)
|
||
{
|
||
this.TryShutdown();
|
||
base.Close("Handshak验证超时");
|
||
}
|
||
});
|
||
}
|
||
|
||
private void ThisDmtpActorOutputSend(DmtpActor actor, ArraySegment<byte>[] transferBytes)
|
||
{
|
||
base.Send(transferBytes);
|
||
}
|
||
|
||
private Task ThisDmtpActorOutputSendAsync(DmtpActor actor, ArraySegment<byte>[] transferBytes)
|
||
{
|
||
return base.SendAsync(transferBytes);
|
||
}
|
||
|
||
private Task ThisOnResetId(DmtpActor rpcActor, IdChangedEventArgs e)
|
||
{
|
||
this.DirectResetId(e.NewId);
|
||
return EasyTask.CompletedTask;
|
||
}
|
||
|
||
#region 发送
|
||
|
||
/// <summary>
|
||
/// 不允许直接发送
|
||
/// </summary>
|
||
/// <param name="buffer"></param>
|
||
/// <param name="offset"></param>
|
||
/// <param name="length"></param>
|
||
public override void Send(byte[] buffer, int offset, int length)
|
||
{
|
||
throw new Exception("不允许直接发送，请指定任意大于0的协议，然后发送。");
|
||
}
|
||
|
||
/// <summary>
|
||
/// 不允许直接发送
|
||
/// </summary>
|
||
/// <param name="transferBytes"></param>
|
||
public override void Send(IList<ArraySegment<byte>> transferBytes)
|
||
{
|
||
throw new Exception("不允许直接发送，请指定任意大于0的协议，然后发送。");
|
||
}
|
||
|
||
/// <summary>
|
||
/// 不允许直接发送
|
||
/// </summary>
|
||
/// <param name="buffer"></param>
|
||
/// <param name="offset"></param>
|
||
/// <param name="length"></param>
|
||
public override Task SendAsync(byte[] buffer, int offset, int length)
|
||
{
|
||
throw new Exception("不允许直接发送，请指定任意大于0的协议，然后发送。");
|
||
}
|
||
|
||
/// <summary>
|
||
/// 不允许直接发送
|
||
/// </summary>
|
||
/// <param name="transferBytes"></param>
|
||
public override Task SendAsync(IList<ArraySegment<byte>> transferBytes)
|
||
{
|
||
throw new Exception("不允许直接发送，请指定任意大于0的协议，然后发送。");
|
||
}
|
||
|
||
#endregion 发送
|
||
}
|
||
} |