取消长时间运行的 Rx 流

Question

如果我有长时间运行操作的流，如下所示：

inputStream.Select(n => Task.Run(() =>
{
    // Long running operation
    Thread.Sleep(TimeSpan.FromSeconds(5));
    return n * n;
}).ToObservable())
.Switch()
.Subscribe(result => 
{
    // Use result in some way
    Console.WriteLine(result);
});

如何在Task.Run调用中获得取消令牌，以便在Switch处理正在进行的计算的订阅时，它将取消令牌设置为已取消，以便我知道中止计算。

Answer 1

您可以使用 Observable.StartAsync 方法，例如

inputStream.Select(n => Observable.StartAsync((token => Task.Run(() =>
{
    if (token.IsCancellationRequested)
    {
        // .. don't need to do anything
        return 0;
    }
    else
    {
        Thread.Sleep(TimeSpan.FromSeconds(1));
        return n * n;
    }                   
}))))
.Switch()
.Subscribe(Console.WriteLine);

或者，如果您要生成多个值，则可以使用与Task一起使用的Observable.Create重载来获取 CancelToken。 例如

inputStream.Select(n => Observable.Create<int>((observer, token) => Task.Run(() =>
{
    while (!token.IsCancellationRequested)
    {
        Thread.Sleep(TimeSpan.FromSeconds(1));
        observer.OnNext(n * n);
    }
    observer.OnCompleted();
})))
.Switch()
.Subscribe(Console.WriteLine);

在任务中，需要调用OnNext来生成值。任务的返回值（如果有）将被忽略。