使IEnumerable方法异步

时间:2014-10-30 20:56:14

标签: c# multithreading callback cancellation

我有以下节点:

class Node
{
    public string Name;        
    public IEnumerable<Node> Children;
}

我有以下扩展方法:

public static class ExtensionMethods
{
    public static IEnumerable<Node> TraverseTree(this Node root)
    {
        if (root.Children != null)
        {
            foreach (var child in root.Children)
            {
                var nodes = TraverseTree(child);
                foreach (var node in nodes)
                {
                    yield return node;
                }
            }
        }

        yield return root;
    }       
}

我想在树中搜索名为&#34; Foo&#34;的节点。为了做到这一点,我做了:

Node myNode = /* some large tree! */
var search = myNode.TraverseTree().Where(x=>x.Name == "Foo").FirstOrDefault();

我有3个目标

  1. 让方法TraverseTree以yield(IEnumerable)遍历树,这样如果第3个节点发生了名称==&#34; Foo&#34;然后我不必遍历整棵树。 现在这个案例是真的
  2. 使方法TraverseTree在单独的线程上运行,因为它可能需要很长时间才能找到。因此我想TraverseTree方法应该采用callBack参数?
  3. 最后能够取消该操作会很高兴。我是否还需要将取消令牌传递给该方法?
  4. 这样做的正确方法是什么?

    抱歉,我忘了提及我正在使用.Net Framework 4.0

2 个答案:

答案 0 :(得分:1)

  1. 这已经使用您拥有的代码完成了。是推迟执行。
  2. 到目前为止,更简单的方法是保持方法同步并将整个查询移动到另一个线程。
  3. 是的,通过遍历算法,子选择器或两者检查的CancellationToken肯定是您可以添加的。另一个选择就是让它等待遍历停止等待的结果,而不是试图实际停止计算的发生。

答案 1 :(得分:0)

首先,我会像这样实现你的功能(注意使用CancellationToken

public static IEnumerable<Node> TraverseTree(this Node root, CancellationToken token)
{
    if (root.Children != null)
    {
        foreach (var child in root.Children)
        {
            if (token.IsCancellationRequested) return; //cancel if requested.

            var nodes = TraverseTree(child);
            foreach (var node in nodes)
            {
                yield return node;
                if (token.IsCancellationRequested) return; //cancel if requested.
            }
        }
    }

    yield return root;
}     

然后,这就是电话的样子:

var cts = new CancellationTokenSource();

var task = Task.Run(
    () => myNode.TraverseTree(cts.Token).Where(x=>x.Name == "Foo").FirstOrDefault(),
    cts.Token);

然后,如果你想取消,你只需致电:

cts.Cancel();