在Timer回调方法中使用CancellationTokens

时间:2017-12-22 12:16:16

标签: c# timer windows-services cancellation-token

我正在实施一个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方法将回答令牌?我认为它会,但我不完全确定,因为它的异步和所有这一切。

1 个答案:

答案 0 :(得分:1)

您不仅需要取消当前的活动工作,还要等待所有活动任务对该取消做出反应并完成。在您当前的方法中,您可以这样做:

private readonly CancellationTokenSource _cts = new CancellationTokenSource();
public override void OnStop() {
    _cts.Cancel();
}

这对任何事都没有帮助,因为一旦请求取消,您将从OnStop方法返回,并且所有后台线程(由于Task.Run而在您的工作上执行)将被杀死而不等待它们完成。

所以你需要一些方法等待他们完成才能从OnStop返回。一种方法是使用Monitor.WaitMonitor.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);
    }
}