mirror of
https://github.com/Radarr/Radarr.git
synced 2024-09-11 20:12:41 +02:00
57 lines
1.2 KiB
C#
57 lines
1.2 KiB
C#
using System;
|
|
using System.Collections.Generic;
|
|
using System.Threading;
|
|
|
|
namespace NzbDrone.Test.Common
|
|
{
|
|
public class ConcurrencyCounter
|
|
{
|
|
private int _items;
|
|
readonly object _mutex = new object();
|
|
readonly Dictionary<int, int> _threads = new Dictionary<int, int>();
|
|
|
|
public int MaxThreads { get { return _threads.Count; } }
|
|
|
|
public ConcurrencyCounter(int items)
|
|
{
|
|
_items = items;
|
|
}
|
|
|
|
public void WaitForAllItems()
|
|
{
|
|
while (_items != 0)
|
|
{
|
|
Thread.Sleep(500);
|
|
}
|
|
}
|
|
|
|
public int Start()
|
|
{
|
|
int threadId = Thread.CurrentThread.ManagedThreadId;
|
|
lock (_mutex)
|
|
{
|
|
|
|
_threads[threadId] = 1;
|
|
}
|
|
|
|
Console.WriteLine("Starting " + threadId);
|
|
return threadId;
|
|
}
|
|
|
|
public void SimulateWork(int sleepInMs)
|
|
{
|
|
var id = Start();
|
|
Thread.Sleep(sleepInMs);
|
|
Stop(id);
|
|
}
|
|
|
|
public void Stop(int id)
|
|
{
|
|
Console.WriteLine("Finished " + id);
|
|
lock (_mutex)
|
|
{
|
|
_items--;
|
|
}
|
|
}
|
|
}
|
|
} |