信号量包装方法

时间:2017-10-06 17:51:07

标签: c# async-await task semaphore

根据有关SO的一些问题,主要是这一个: Throttling asynchronous tasks

我已经实现了SemaphoreSlim对象,以便在我的应用程序中通过一系列方法同时处理请求。这些方法中的大多数都采用ID列表,并从Web以并发方式获取每个ID的单字节数组。实现如下:

 using (var semaphore = new SemaphoreSlim(MaxConcurrency))
                {
                    var tasks = fileMetadata.GroupBy(x => x.StorageType).Select(async storageTypeFileMetadata=>
                    {
                        await semaphore.WaitAsync();
                        try
                        {
                            var fileManager = FileManagerFactory.CreateFileManager((StorageType)storageTypeFileMetadata.Key);
                            await fileManager.UpdateFilesAsync(storageTypeFileMetadata);
                        }
                        finally
                        {
                            semaphore.Release();
                        }
                    });

                    await Task.WhenAll(tasks);
                }

有没有办法为信号量代码抽象出一个方法或一些可重用的代码片段,并传递我需要完成的工作,因此可以重复使用它而无需每次重写信号量代码?使用相同信号量模式的多个方法之间的唯一区别是我正在迭代的列表以及它在try {}中所做的工作。

我正在考虑将list list.select(x =>我的工作中的任务方法)传递给信号量方法,这是所有包装器信号量代码。

1 个答案:

答案 0 :(得分:1)

所以我猜是这样的:

public static class Extension
{
    public static async Task ExecuteAsync<T>(this IEnumerable<T> items, Func<T, Task> task, int concurrency)
    {
        var tasks = new List<Task>();

        using (var semaphore = new SemaphoreSlim(concurrency))
        {
            foreach (var item in items)
            {
                tasks.Add(ExecuteInSemaphore(semaphore, task, item));
            }

            await Task.WhenAll(tasks);
        }
    }

    private static async Task ExecuteInSemaphore<T>(SemaphoreSlim semaphore, Func<T, Task> task, T item)
    {
        await semaphore.WaitAsync();

        try
        {
            await task(item);
        }
        finally
        {
            semaphore.Release();
        }
    }
}

然后你会像:

一样使用它
await fileMetadata.GroupBy(x => x.StorageType).ExecuteAsync(storageTypeFileMetadata =>
{
    var fileManager = FileManagerFactory.CreateFileManager((StorageType)storageTypeFileMetadata.Key);
    return fileManager.UpdateFilesAsync(storageTypeFileMetadata);
}, 4);