gpt4 book ai didi

c# - 如何使用超时取消 TaskCompletionSource

转载 作者:可可西里 更新时间:2023-11-01 09:07:20 32 4
gpt4 key购买 nike

我有使用 await 关键字异步调用的函数:

public Task<StatePropertyEx> RequestStateForEntity(EntityKey entity, string propName)
{
var tcs = new TaskCompletionSource<StateInfo>();
try
{
var propInstance = BuildCacheKey(entity, propName);
StateCacheItem cacheItem;
if (_stateCache.TryGetValue(propInstance, out cacheItem))
{
tcs.SetResult( new StateInfo (cacheItem.State.Name, cacheItem.State.Value) );
return tcs.Task;
}

//state not found in local cache so save the tcs for later and request the state
var cacheKey = BuildCacheKey(entity, propName);
_stateRequestItemList.TryAdd(cacheKey, new StateRequestItem(entity, propName, tcs));

_evtClient.SubmitStateRequest(entity, propName);

return tcs.Task;
}
catch (Exception ex)
{
tcs.SetException(ex);
return tcs.Task;
}
}

该函数查看它是否具有所需的信息,如果有则返回它。如果它没有详细信息,它会发出一个请求,该请求最终应该作为一个事件出现。那时我的代码(未显示)找到存储的 TaskCompletionSource 项,设置结果并返回它。这一切都很好,但现在我被要求考虑一种情况,当我通过“_evtClient.SubmitStateRequest(entity, propName);”请求状态时,可能永远不会返回回复。线。我需要实现某种超时机制,以便我可以取消 TCS 任务,以便函数调用者可以正常失败。我一直在寻找 SO 和互联网,但找不到任何看起来正确的东西。我现在不确定是否需要以不同的方式重构上面的代码。任何人都可以建议或指出类似的情况吗?

调用上述函数的代码可以像这样一次性调用:

var stateProperty = await RequestStateForEntity(key, stateName);

或者批量处理,像这样:

await
Task.WhenAll(
stateDefinitions.Select(stateDefinition => stateDefinition.Name)
.Select(
stateName =>
Task.Factory.StartNew(
async () => results.Add(await RequestStateForEntity(key, stateName)))
.Unwrap())
.ToArray());

最佳答案

首先,您真正想要启用的是取消。取消来自超时的事实只是一个脚注。

.NET 有一些 great built-in support for cancellation , 和 Task-based Asynchronous Pattern prescribes how to use it .

本质上,您要做的是获取一个CancellationToken:

Task<StatePropertyEx> RequestStateForEntity(EntityKey entity, string propName,
CancellationToken cancellationToken);

接下来,您想在收到该 token 时做出响应。理想情况下,您只想将 CancellationToken 传递给 _evtClient,以便真正取消请求:

_evtClient.SubmitStateRequest(entity, propName, cancellationToken);

这是实现取消的正常方式,如果 SubmitStateRequest 已经理解取消,它会很好用。事件参数通常有一个指示取消的标志(例如 AsyncCompletedEventArgs.Cancelled )。如果可能,请使用此方法(即根据需要更改 _evtClient 以支持取消)。

但有时这是不可能的。在这种情况下,您可以选择假装支持取消。如果请求在取消后完成,您实际上 只是忽略请求。这不是最理想的情况,但有时您别无选择。

就我个人而言,我不太喜欢这种方法,因为它让 API 变得“谎言”:方法签名声称支持取消,但它实际上只是伪造的。所以首先,我建议记录下来。在代码注释中道歉说明 _evtClient 不支持取消,“取消”实际上只是假装取消。

然后,您需要自己挂接到 CancellationToken,在状态请求项出现在列表中之后但在发送实际请求之前:

var item = new StateRequestItem(entity, propName, tcs);
_stateRequestItemList.TryAdd(cacheKey, item);
item.CancellationRegistration = cancellationToken.Register(() =>
{
StateRequestItem cancelledItem;
if (!_stateRequestItemList.TryRemove(cacheKey, out cancelledItem))
return;
cancelledItem.TaskCompletionSource.TrySetCanceled();
});
_evtClient.SubmitStateRequest(entity, propName);

最后,您需要更新事件处理程序完成代码(未显示)以忽略状态请求项已被删除的情况,并在状态请求时处理 CancellationRegistration找到项目。


一旦你的方法支持取消,那么很容易通过定时器取消:

var cts = new CancellationTokenSource(TimeSpan.FromSeconds(10));
CancellationToken token = cts.Token;

或来自任何其他类型的情况。比如说,如果用户取消了他正在做的任何事情。或者,如果系统的另一部分决定不再需要该数据。一旦您的代码支持取消,它就可以处理任何原因的取消。

关于c# - 如何使用超时取消 TaskCompletionSource,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/25263082/

32 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com