90 lines
2.7 KiB
C#
90 lines
2.7 KiB
C#
|
//------------------------------------------------------------------------------
|
|||
|
|
// 此代码版权（除特别声明或在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首页：https://touchsocket.net/
|
|||
|
|
// 交流QQ群：234762506
|
|||
|
|
// 感谢您的下载和使用
|
|||
|
|
//------------------------------------------------------------------------------
|
|||
|
|
|
|||
|
|
using System.Threading;
|
|||
|
|
using TouchSocket.Core;
|
|||
|
|
|
|||
|
|
namespace TouchSocket.Rpc
|
|||
|
|
{
|
|||
|
|
/// <summary>
|
|||
|
|
/// Rpc调用上下文的基本实现
|
|||
|
|
/// </summary>
|
|||
|
|
public abstract class CallContext : ICallContext
|
|||
|
|
{
|
|||
|
|
private bool m_canceled;
|
|||
|
|
private readonly object m_locker = new object();
|
|||
|
|
private CancellationTokenSource m_tokenSource;
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// CallContext
|
|||
|
|
/// </summary>
|
|||
|
|
/// <param name="caller"></param>
|
|||
|
|
/// <param name="rpcMethod"></param>
|
|||
|
|
/// <param name="resolver"></param>
|
|||
|
|
public CallContext(object caller, RpcMethod rpcMethod, IResolver resolver)
|
|||
|
|
{
|
|||
|
|
this.Caller = caller;
|
|||
|
|
this.RpcMethod = rpcMethod;
|
|||
|
|
this.Resolver = resolver;
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <summary>
|
|||
|
|
/// CallContext
|
|||
|
|
/// </summary>
|
|||
|
|
public CallContext()
|
|||
|
|
{
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <inheritdoc/>
|
|||
|
|
public object Caller { get; protected set; }
|
|||
|
|
|
|||
|
|
/// <inheritdoc/>
|
|||
|
|
public RpcMethod RpcMethod { get; protected set; }
|
|||
|
|
|
|||
|
|
/// <inheritdoc/>
|
|||
|
|
public IResolver Resolver { get; protected set; }
|
|||
|
|
|
|||
|
|
/// <inheritdoc/>
|
|||
|
|
public CancellationToken Token
|
|||
|
|
{
|
|||
|
|
get
|
|||
|
|
{
|
|||
|
|
lock (this.m_locker)
|
|||
|
|
{
|
|||
|
|
if (this.m_canceled)
|
|||
|
|
{
|
|||
|
|
return new CancellationToken(true);
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
this.m_tokenSource ??= new CancellationTokenSource();
|
|||
|
|
return this.m_tokenSource.Token;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
/// <inheritdoc/>
|
|||
|
|
public void Cancel()
|
|||
|
|
{
|
|||
|
|
lock (this.m_locker)
|
|||
|
|
{
|
|||
|
|
if (this.m_tokenSource != null)
|
|||
|
|
{
|
|||
|
|
this.m_tokenSource.Cancel();
|
|||
|
|
}
|
|||
|
|
else
|
|||
|
|
{
|
|||
|
|
this.m_canceled = true;
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
}
|