root / trunk / Pithos.Core / Agents / AsyncManualResetEvent.cs @ 174bbb6e
History | View | Annotate | Download (3.4 kB)
1 |
#region |
---|---|
2 |
/* ----------------------------------------------------------------------- |
3 |
* <copyright file="AsyncManualResetEvent.cs" company="GRNet"> |
4 |
* |
5 |
* Copyright 2011-2012 GRNET S.A. All rights reserved. |
6 |
* |
7 |
* Redistribution and use in source and binary forms, with or |
8 |
* without modification, are permitted provided that the following |
9 |
* conditions are met: |
10 |
* |
11 |
* 1. Redistributions of source code must retain the above |
12 |
* copyright notice, this list of conditions and the following |
13 |
* disclaimer. |
14 |
* |
15 |
* 2. Redistributions in binary form must reproduce the above |
16 |
* copyright notice, this list of conditions and the following |
17 |
* disclaimer in the documentation and/or other materials |
18 |
* provided with the distribution. |
19 |
* |
20 |
* |
21 |
* THIS SOFTWARE IS PROVIDED BY GRNET S.A. ``AS IS'' AND ANY EXPRESS |
22 |
* OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED |
23 |
* WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR |
24 |
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL GRNET S.A OR |
25 |
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, |
26 |
* SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT |
27 |
* LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF |
28 |
* USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED |
29 |
* AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT |
30 |
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN |
31 |
* ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE |
32 |
* POSSIBILITY OF SUCH DAMAGE. |
33 |
* |
34 |
* The views and conclusions contained in the software and |
35 |
* documentation are those of the authors and should not be |
36 |
* interpreted as representing official policies, either expressed |
37 |
* or implied, of GRNET S.A. |
38 |
* </copyright> |
39 |
* ----------------------------------------------------------------------- |
40 |
*/ |
41 |
#endregion |
42 |
using System.Threading; |
43 |
using System.Threading.Tasks; |
44 |
|
45 |
namespace Pithos.Core.Agents |
46 |
{ |
47 |
using System; |
48 |
using System.Collections.Generic; |
49 |
using System.Linq; |
50 |
using System.Text; |
51 |
|
52 |
/// <summary> |
53 |
/// TODO: Update summary. |
54 |
/// </summary> |
55 |
public class AsyncManualResetEvent |
56 |
{ |
57 |
private volatile TaskCompletionSource<bool> _tcs; |
58 |
|
59 |
public AsyncManualResetEvent() |
60 |
{ |
61 |
_tcs = new TaskCompletionSource<bool>(); |
62 |
} |
63 |
|
64 |
public AsyncManualResetEvent(bool signalled) |
65 |
{ |
66 |
_tcs = new TaskCompletionSource<bool>(); |
67 |
if (signalled) |
68 |
_tcs.SetResult(true); |
69 |
} |
70 |
public Task WaitAsync() { return _tcs.Task; } |
71 |
|
72 |
public void Wait() |
73 |
{ |
74 |
_tcs.Task.Wait(); |
75 |
} |
76 |
|
77 |
public void Set() |
78 |
{ |
79 |
var tcs = _tcs; |
80 |
Task.Factory.StartNew(s => ((TaskCompletionSource<bool>)s).TrySetResult(true), |
81 |
tcs, CancellationToken.None, TaskCreationOptions.PreferFairness, TaskScheduler.Default); |
82 |
tcs.Task.Wait(); |
83 |
} |
84 |
|
85 |
public void Reset() |
86 |
{ |
87 |
while (true) |
88 |
{ |
89 |
var tcs = _tcs; |
90 |
if (!tcs.Task.IsCompleted) |
91 |
return; |
92 |
#pragma warning disable 420 |
93 |
if (Interlocked.CompareExchange(ref _tcs, new TaskCompletionSource<bool>(), tcs) == tcs) |
94 |
#pragma warning restore 420 |
95 |
return; |
96 |
} |
97 |
} |
98 |
} |
99 |
} |