动态减少信号量的容量

Question

我一直在尝试使用信号量来控制我的服务可以处理的请求数量。即。

class Service : IDisposable {
    SemaphoreSlim s = new SemaphoreSlim(InitialCapacity);

    ....
    async void ProcessRequest() {
        await s.WaitAsync();
        try {
             ......
        } finally {
            s.Release();
        }
    }
}

我遇到了 2 个问题，我不知道如何解决。我一直在使用类似的 hack 来解决这些问题，但我想知道是否有更好的方法

1. 我希望能够动态改变我的服务类的容量，所以我有这样的东西。

   void ChangeCapacity(int newCapacity) {
       int extraRequestsCount = newCapacity - oldCapacity;
       if (extraRequestsCount > 0) {
           s.Release(extraRequestsCount);
       }
       else if (extraRequestsCount < 0) {
           for (int i = 0; i < -extraRequestsCount; i++) {
               s.WaitAsync(); // try to steal some resources, over time...
           }
       }
   }
   

2. 在 dispose 方法中，我想确保所有请求处理在我处理信号量之前完成，否则我的 ProcessRequest() 中的 s.Release() 调用会抛出 ObjectDisposedException，所以我做了以下操作

   public void Dispose() {
       if (s!= null) {
           for (int i = 0; i < oldCapacity; i++) {
               s.Wait();
           }
           s.Dispose();
       }
   }
   

请注意，我一直在使用循环手动等待很多次。如果容量很大，这真的很慢。有一个更好的方法吗？信号量有一个 Release(int count) 为什么没有 Wait(int count)？

Answer 1

我可能会做的是在进行调整时，将您的信号量实例替换为具有所需容量的新信号量，并将所有未来的工作分配给该信号量。在所有线程完成对它的引用之前，不会对现有的和现在取消引用的信号量进行垃圾收集，因此只要您将信号量变量本地分配给每个线程，这样做应该是安全的。