public void QueueWorkItem([NotNull][Pooled] Action workItem)
{
var lockTaken = false;
try
{
spinLock.Enter(ref lockTaken);
PooledDelegateHelper.AddReference(workItem);
workItems.Enqueue(workItem);
if (activeThreadCount + 1 >= workers.Count && workers.Count < MaxThreadCount)
{
var worker = Task.Factory.StartNew(ProcessWorkItems, workers.Count, TaskCreationOptions.LongRunning);
workers.Add(worker);
//Console.WriteLine($"Thread {workers.Count} added");
}
workAvailable.Set();
}
finally
{
if (lockTaken)
{
spinLock.Exit(true);
}
}
}
private void ProcessWorkItems(object state)
{
//var spinWait = new SpinWait();
while (true)
{
Action workItem = null;
//while (!spinWait.NextSpinWillYield)
{
var lockTaken = false;
try
{
spinLock.Enter(ref lockTaken);
if (workItems.Count > 0)
{
try
{
workItem = workItems.Dequeue();
//Interlocked.Increment(ref activeThreadCount);
if (workItems.Count == 0)
{
workAvailable.Reset();
}
}
catch
{
}
}
//if (workItems.Count > 0)
//{
// // If we didn't consume the last work item, kick off another worker
// workAvailable.Set();
//}
}
finally
{
if (lockTaken)
{
spinLock.Exit(true);
}
}
if (workItem != null)
{
try
{
Interlocked.Increment(ref activeThreadCount);
workItem.Invoke();
//spinWait.Reset();
}
catch (Exception)
{
// Ignoring Exception
}
finally
{
PooledDelegateHelper.Release(workItem);
Interlocked.Decrement(ref activeThreadCount);
}
}
else
{
//spinWait.SpinOnce();
}
}
// Wait for another work item to be (potentially) available
workAvailable.WaitOne();
}
}