信号量包装方法

Question

基于一些关于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.select(x=> 我的任务方法及其中的工作) 传递给一个信号量方法，它是所有包装信号量代码。

Answer 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);

信号量包装方法

Semaphore wrapper method

c#

semaphore

task

async-await