WorkerPool.cs
69 lines
| 1.9 KiB
| text/x-csharp
|
CSharpLexer
cin
|
r12 | using System; | ||
| using System.Collections.Generic; | ||||
| using System.Linq; | ||||
| using System.Text; | ||||
| using System.Threading; | ||||
| using System.Diagnostics; | ||||
| namespace Implab.Parallels { | ||||
|
cin
|
r15 | public class WorkerPool : DispatchPool<Action> { | ||
|
cin
|
r12 | |||
|
cin
|
r15 | MTQueue<Action> m_queue = new MTQueue<Action>(); | ||
| int m_queueLength = 0; | ||||
|
cin
|
r13 | |||
|
cin
|
r15 | public WorkerPool(int minThreads, int maxThreads) | ||
| : base(minThreads, maxThreads) { | ||||
| InitPool(); | ||||
|
cin
|
r13 | } | ||
|
cin
|
r15 | public WorkerPool(int threads) | ||
| : base(threads) { | ||||
| InitPool(); | ||||
|
cin
|
r13 | } | ||
|
cin
|
r15 | public WorkerPool() | ||
| : base() { | ||||
| InitPool(); | ||||
|
cin
|
r13 | } | ||
|
cin
|
r12 | public Promise<T> Invoke<T>(Func<T> task) { | ||
| if (task == null) | ||||
| throw new ArgumentNullException("task"); | ||||
|
cin
|
r15 | if (IsDisposed) | ||
| throw new ObjectDisposedException(ToString()); | ||||
|
cin
|
r12 | |||
| var promise = new Promise<T>(); | ||||
|
cin
|
r15 | EnqueueTask(delegate() { | ||
|
cin
|
r13 | try { | ||
| promise.Resolve(task()); | ||||
| } catch (Exception e) { | ||||
| promise.Reject(e); | ||||
| } | ||||
| }); | ||||
|
cin
|
r12 | |||
| return promise; | ||||
| } | ||||
|
cin
|
r15 | protected void EnqueueTask(Action unit) { | ||
| Debug.Assert(unit != null); | ||||
| Interlocked.Increment(ref m_queueLength); | ||||
| m_queue.Enqueue(unit); | ||||
| // if there are sleeping threads in the pool wake one | ||||
| // probably this will lead a dry run | ||||
| WakeNewWorker(); | ||||
|
cin
|
r12 | } | ||
|
cin
|
r15 | protected override bool TryDequeue(out Action unit) { | ||
| if (m_queue.TryDequeue(out unit)) { | ||||
| Interlocked.Decrement(ref m_queueLength); | ||||
| return true; | ||||
|
cin
|
r12 | } | ||
|
cin
|
r15 | return false; | ||
|
cin
|
r12 | } | ||
|
cin
|
r15 | protected override void InvokeUnit(Action unit) { | ||
| unit(); | ||||
|
cin
|
r12 | } | ||
| } | ||||
| } | ||||