如果需要超过 x 毫秒，如何超时调用？

Question

我正在使用 C# 语言并正在从 cassandra 中获取数据，因此我正在使用以下代码从 Cassandra 中获取数据并且它工作正常。

我唯一的问题是我的 ProcessCassQuery 方法 - 我将 CancellationToken.None 传递给我的 requestExecuter 函数，这可能不是正确的做法。

如果查询时间超过800 milliseconds，我需要使查询超时，那么我怎样才能在此处正确使用CancellationToken 来使查询超时？ Cassandra 驱动程序尚未处理CancellationToken，因此我需要通过其他方式在外部执行此操作。此代码将以非常高的吞吐量调用，因此在超时调用时需要高效..

/**
 *
 * Below method does multiple async calls on each table for their corresponding id's by limiting it down using Semaphore.
 *
 */
private async Task<List<T>> ProcessCassQueries<T>(IList<int> ids, Func<CancellationToken, int, Task<T>> mapperFunc, string msg) where T : class
{
    var tasks = ids.Select(async id => 
    {
        await semaphore.WaitAsync();
        try
        {
            return await ProcessCassQuery(ct => mapperFunc(ct, id), msg);
        }
        finally
        {
            semaphore.Release();
        }
    });

  return (await Task.WhenAll(tasks)).Where(e => e != null).ToList();
}

// this might not be good idea to do it. how can I improve below method?
private Task<T> ProcessCassQuery<T>(Func<CancellationToken, Task<T>> requestExecuter, string msg) where T : class
{
    return requestExecuter(CancellationToken.None);
}

Answer 1

您可以使用以下代码：

// this might not be good idea to do it. how can I improve below method?
private async Task<T> ProcessCassQuery<T>(Func<CancellationToken, Task<T>> requestExecuter, string msg) where T : class
{
    using var cts = new CancellationTokenSource();
    cts.CancelAfter(TimeSpan.FromMilliseconds(800));
    return await requestExecuter(cts.Token);
}