PollingComponent.cs
88 lines
| 2.7 KiB
| text/x-csharp
|
CSharpLexer
cin
|
r258 | using System; | |
using System.Threading; | |||
using System.Threading.Tasks; | |||
namespace Implab.Components { | |||
public abstract class PollingComponent : RunnableComponent { | |||
readonly Timer m_timer; | |||
readonly CancellationTokenSource m_cancellation = new CancellationTokenSource(); | |||
cin
|
r259 | Task m_pending; | |
Task m_poll; | |||
cin
|
r258 | /// <summary> | |
/// Poll interval in milliseconds. | |||
/// </summary> | |||
/// <returns></returns> | |||
public int Interval { get; set; } | |||
/// <summary> | |||
/// Delay to the first poll after start in milliseconds | |||
/// </summary> | |||
/// <returns></returns> | |||
public int Delay { get; set; } | |||
/// <summary> | |||
/// Indicates how to handle unhandled exceptions in <see cref="Poll()"/> method. | |||
/// </summary> | |||
/// <returns></returns> | |||
public bool FailOnError { get; set; } | |||
/// <summary> | |||
/// Event for the unhandled exceptions in <see cref="Poll()"/> method. | |||
/// </summary> | |||
public event EventHandler<UnhandledExceptionEventArgs> UnhandledException; | |||
protected PollingComponent(bool initialized) : base(initialized) { | |||
m_timer = new Timer(OnTimer); | |||
} | |||
protected override void RunInternal() { | |||
ScheduleNextPoll(Delay); | |||
} | |||
cin
|
r259 | protected override async Task StopInternalAsync(CancellationToken ct) { | |
// component in Stopping state, no new polls will be scheduled | |||
m_cancellation.Cancel(); | |||
try { | |||
// await for pending poll | |||
await m_poll; | |||
} catch (OperationCanceledException e) { | |||
// OK | |||
} | |||
} | |||
cin
|
r258 | ||
protected abstract Task Poll(CancellationToken ct); | |||
void ScheduleNextPoll(int timeout) { | |||
lock (SynchronizationObject) { | |||
cin
|
r259 | if (State == ExecutionState.Running) { | |
m_pending = Safe.CreateTask(m_cancellation.Token); | |||
m_poll = m_pending.Then(() => Poll(m_cancellation.Token)); | |||
cin
|
r258 | m_timer.Change(timeout, Timeout.Infinite); | |
cin
|
r259 | } | |
cin
|
r258 | } | |
} | |||
cin
|
r259 | async void OnTimer(object state) { | |
cin
|
r258 | try { | |
cin
|
r259 | m_pending.Start(); | |
await m_poll; | |||
cin
|
r258 | } catch (Exception e) { | |
UnhandledException.DispatchEvent(this, new UnhandledExceptionEventArgs(e, false)); | |||
if (FailOnError) | |||
Fail(e); | |||
} | |||
ScheduleNextPoll(Interval); | |||
} | |||
protected override void Dispose(bool disposing) { | |||
if (disposing) | |||
Safe.Dispose(m_timer, m_cancellation); | |||
base.Dispose(disposing); | |||
} | |||
} | |||
} |