任务取消不停止任务执行

时间:2016-07-06 15:17:56

标签: c# task

我有Task查询活动目录并使用结果填充列表。我已经设置了我的任务以便可以取消它,但是,当调用取消时,任务会继续执行其工作。我知道任务已被取消,因为它返回并且运行了要在任务返回上执行的操作,但查询在后台继续运行,使用内存和处理能力。任务可以重复启动和“取消”,每次迭代任务都在运行并使用资源。如何取消实际取消?

视图模型

private async Task RunQuery(QueryType queryType,
    string selectedItemDistinguishedName = null)
{
    StartTask();
    try
    {
        _activeDirectoryQuery = new ActiveDirectoryQuery(queryType,
            CurrentScope, selectedItemDistinguishedName);
        await _activeDirectoryQuery.Execute();
        Data = _activeDirectoryQuery.Data.ToDataTable().AsDataView();
        CurrentQueryType = queryType;
    }
    catch (ArgumentNullException)
    {
        ShowMessage(
            "No results of desired type found in selected context.");
    }
    catch (OutOfMemoryException)
    {
        ShowMessage("The selected query is too large to run.");
    }
    FinishTask();
}

private void CancelCommandExecute()
{
    _activeDirectoryQuery?.Cancel();
}

ActiveDirectoryQuery

public async Task Execute()
{
    _cancellationTokenSource = new CancellationTokenSource();
    var taskCompletionSource = new TaskCompletionSource<object>();
    _cancellationTokenSource.Token.Register(
        () => taskCompletionSource.TrySetCanceled());
    DataPreparer dataPreparer = null;
    var task = Task.Run(() =>
    {
        if (QueryTypeIsOu())
        {
            dataPreparer = SetUpOuDataPreparer();
        }
        else if (QueryTypeIsContextComputer())
        {
            dataPreparer = SetUpComputerDataPreparer();
        }
        else if (QueryTypeIsContextDirectReportOrUser())
        {
            dataPreparer = SetUpDirectReportOrUserDataPreparer();
        }
        else if (QueryTypeIsContextGroup())
        {
            dataPreparer = SetUpGroupDataPreparer();
        }
        Data = GetData(dataPreparer);
    },
    _cancellationTokenSource.Token);
    await Task.WhenAny(task, taskCompletionSource.Task);
}

public void Cancel()
{
    _cancellationTokenSource?.Cancel();
}

Cancel()由绑定到Command的{​​{1}}调用。该任务可能需要几分钟才能执行,并且可能消耗数百兆字节的RAM。如果它有帮助,我可以提供任何引用的方法或任何其他信息。

1 个答案:

答案 0 :(得分:4)

Cancellation is cooperative,如果您想取消操作,则需要编辑要取消的功能。所以执行将成为

public async Task Execute()
{
    _cancellationTokenSource = new CancellationTokenSource();
    var taskCompletionSource = new TaskCompletionSource<object>();

    //Token registrations need to be disposed when done.
    using(_cancellationTokenSource.Token.Register(
        () => taskCompletionSource.TrySetCanceled()))
    {
        DataPreparer dataPreparer = null;
        var task = Task.Run(() =>
        {
            if (QueryTypeIsOu())
            {
                dataPreparer = SetUpOuDataPreparer(_cancellationTokenSource.Token);
            }
            else if (QueryTypeIsContextComputer())
            {
                dataPreparer = SetUpComputerDataPreparer(_cancellationTokenSource.Token);
            }
            else if (QueryTypeIsContextDirectReportOrUser())
            {
                dataPreparer = SetUpDirectReportOrUserDataPreparer(_cancellationTokenSource.Token);
            }
            else if (QueryTypeIsContextGroup())
            {
                dataPreparer = SetUpGroupDataPreparer(_cancellationTokenSource.Token);
            }
            Data = GetData(dataPreparer, _cancellationTokenSource.Token);
        },
        _cancellationTokenSource.Token);
        await Task.WhenAny(task, taskCompletionSource.Task);
   }
}

然后从那些方法里面。如果在这些函数中有循环,则需要从循环内部调用token.ThrowIfCancellationRequested()。如果您没有循环并调用某个外部API,则需要使用该API的取消方法,希望API接受CancellationToken,如果不是,则需要调用.Cancel()方法像Register中一样使用Execute方法。

如果API未公开取消查询的方法,则只需 安全 方法提前停止查询,您需要将查询移至单独的exe。执行查询时,执行var proc = Process.Start(...)以启动exe。要与它通信使用某种形式的IPC,如WCF over Named Pipes,您可以在进程启动之前生成Guid并将其作为参数传递,然后使用该guid作为命名管道的名称。如果您需要提前结束查询,请执行proc.Kill()以结束外部流程。