我有一个IEnumerable<T>方法,我正在使用它来查找WebForms页面中的控件。
这个方法是递归的,当yield return返回递归调用的值时,我在返回我想要的类型时遇到了一些问题。
我的代码如下:
public static IEnumerable<Control> GetDeepControlsByType<T>(this Control control)
{
foreach(Control c in control.Controls)
{
if (c is T)
{
yield return c;
}
if(c.Controls.Count > 0)
{
yield return c.GetDeepControlsByType<T>();
}
}
}
当前抛出“无法转换表达式类型”错误。但是,如果此方法返回类型IEnumerable<Object>,则构建代码,但在输出中返回错误的类型。
有没有一种方法可以在使用递归的同时使用收益率?
正如Jon Skeet和Colonel Panic在他们的回答中指出的,如果树很深,在递归方法中使用yield return可能会导致性能问题。
下面是一个通用的非递归扩展方法,它执行深度优先遍历树序列:
public static IEnumerable<TSource> RecursiveSelect<TSource>(
this IEnumerable<TSource> source, Func<TSource, IEnumerable<TSource>> childSelector)
{
var stack = new Stack<IEnumerator<TSource>>();
var enumerator = source.GetEnumerator();
try
{
while (true)
{
if (enumerator.MoveNext())
{
TSource element = enumerator.Current;
yield return element;
stack.Push(enumerator);
enumerator = childSelector(element).GetEnumerator();
}
else if (stack.Count > 0)
{
enumerator.Dispose();
enumerator = stack.Pop();
}
else
{
yield break;
}
}
}
finally
{
enumerator.Dispose();
while (stack.Count > 0) // Clean up in case of an exception.
{
enumerator = stack.Pop();
enumerator.Dispose();
}
}
}
与Eric Lippert的解决方案不同,RecursiveSelect直接使用枚举器,因此不需要调用Reverse(这将在内存中缓冲整个序列)。
使用RecursiveSelect, OP的原始方法可以像这样简单地重写:
public static IEnumerable<Control> GetDeepControlsByType<T>(this Control control)
{
return control.Controls.RecursiveSelect(c => c.Controls).Where(c => c is T);
}
Seredynski的语法是正确的,但是您应该注意避免在递归函数中使用yield return,因为这对于内存使用来说是一场灾难。参见https://stackoverflow.com/a/3970171/284795,它会随着深度而爆炸性地扩展(在我的应用程序中,类似的函数使用了10%的内存)。
一个简单的解决方案是使用一个列表,并将递归https://codereview.stackexchange.com/a/5651/754传递给它
/// <summary>
/// Append the descendents of tree to the given list.
/// </summary>
private void AppendDescendents(Tree tree, List<Tree> descendents)
{
foreach (var child in tree.Children)
{
descendents.Add(child);
AppendDescendents(child, descendents);
}
}
或者,您可以使用堆栈和while循环来消除递归调用https://codereview.stackexchange.com/a/5661/754