我正在实施一个Windows Service
,它将执行一些昂贵的Database operations
,例如从一些表中检索和存储数据到其他表,目的是创建摘要,处理报告数据等。
我使用Timer object
来控制我希望操作发生的时间,而且这一切都很好,我的主要问题是当有人决定取消时如何进行取消停止service
或Windows重新启动,调用OnStop()
方法。
我的代码如下所示:
码
tmr = new Timer(new TimerCallback(tmr_Tick), cTs.Token, 0, 1000);
我有一个CancellationTokenSource
作为服务中名为cTs
的全局变量。
调用OnStop
方法后,我只需设置cTs.Cancel()
我的Callback
看起来像这样:
回调:
private void tmr_Tick(object sender)
{
if (((CancellationToken)sender).IsCancellationRequested)
{
tmr.Change(Timeout.Infinite, Timeout.Infinite);
return;
}
if (DateTime.Now.TimeOfDay.Equals(intervalRepetition))
Task.Run(async () => { await ProcessData((CancellationToken)sender); });
}
我的问题是关于TimerCallback
,因为我没有找到任何有助于处理令牌的内容,我将令牌作为State object
传递,但我不确定是否这是最好的方法,所以我在寻找建议。
设置IsCancellationRequested
时我该怎么办?我应该实施更好的退出策略,也许会抛出取消例外吗?
令牌传播到ProcessData
方法是否正确?如果我从它上面返回,ProcessData
方法将回答令牌?我认为它会,但我不完全确定,因为它的异步和所有这一切。
答案 0 :(得分:1)
您不仅需要取消当前的活动工作,还要等待所有活动任务对该取消做出反应并完成。在您当前的方法中,您可以这样做:
private readonly CancellationTokenSource _cts = new CancellationTokenSource();
public override void OnStop() {
_cts.Cancel();
}
这对任何事都没有帮助,因为一旦请求取消,您将从OnStop方法返回,并且所有后台线程(由于Task.Run
而在您的工作上执行)将被杀死而不等待它们完成。
所以你需要一些方法等待他们完成才能从OnStop
返回。一种方法是使用Monitor.Wait
和Monitor.Pulse
,如下所示:
private readonly object _activeTasksLock = new object();
private int _activeTasks;
private CancellationTokenSource _cts = new CancellationTokenSource();
private async Task ProcessData(CancellationToken ct) {
lock (_activeTasksLock) {
// increment number of active tasks
_activeTasks++;
}
try {
// do stuff
await Task.Delay(1000, ct);
// check cancellation token and exit if necessary
}
finally {
lock (_activeTasksLock) {
// decrement and notify
_activeTasks--;
Monitor.Pulse(_activeTasksLock);
}
}
}
public override void OnStop() {
// dispose\stop your timer here
// then cancel
_cts.Cancel();
// then wait until all is done
lock (_activeTasksLock) {
while (_activeTasks > 0)
Monitor.Wait(_activeTasksLock);
}
}