-
Notifications
You must be signed in to change notification settings - Fork 0
/
TimedThread.cs
122 lines (106 loc) · 2.92 KB
/
TimedThread.cs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
using System;
using System.Threading;
using System.Threading.Tasks;
public class TimedThread
{
private readonly System.Action action;
private readonly int timeoutMiliseconds;
public TimedThread(Action action, int timeoutMiliseconds)
{
this.action = action;
this.timeoutMiliseconds = timeoutMiliseconds;
}
public ThreadStatus Invoke()
{
Exception exception = null;
var cancellationTokenSource = new CancellationTokenSource();
Task task = Task.Run(() =>
{
try
{
using (cancellationTokenSource.Token.Register(Thread.CurrentThread.Abort))
{
action();
}
}
catch (Exception e)
{
if (!(e is ThreadAbortException))
{
exception = e;
}
}
}, cancellationTokenSource.Token);
bool done = task.Wait(timeoutMiliseconds);
if (exception != null)
{
throw exception;
}
if (done)
{
return ThreadStatus.DONE;
}
cancellationTokenSource.Cancel();
return ThreadStatus.TIMEOUT;
}
public async Task<ThreadStatus> InvokeAsync()
{
return await Task.Factory.StartNew(Invoke);
}
}
public enum ThreadStatus
{
DONE,
TIMEOUT
}
public class TimedThread<TResult>
{
private readonly Func<TResult> action;
private readonly int timeoutMiliseconds;
public TimedThread(Func<TResult> action, int timeoutMiliseconds)
{
this.action = action;
this.timeoutMiliseconds = timeoutMiliseconds;
}
public (ThreadStatus, TResult) Invoke()
{
Exception exception = null;
var cancellationTokenSource = new CancellationTokenSource();
var myTask = new Task<TResult>(() => action(), cancellationTokenSource.Token);
myTask.Start();
Task<TResult> task = Task.Run(() =>
{
TResult result = default;
try
{
using (cancellationTokenSource.Token.Register(Thread.CurrentThread.Abort))
{
result = action();
}
}
catch (Exception e)
{
if (!(e is ThreadAbortException))
{
exception = e;
}
}
return result;
}, cancellationTokenSource.Token);
bool done = task.Wait(timeoutMiliseconds);
if (exception != null)
{
throw exception;
}
if (done)
{
return (ThreadStatus.DONE, task.Result);
}
cancellationTokenSource.Cancel();
return (ThreadStatus.TIMEOUT, task.Result);
}
public async Task<(ThreadStatus, TResult)> InvokeAsync()
{
return await Task.Factory.StartNew(Invoke);
}
}