Vô số và đệ quy sử dụng lợi nhuận

Tôi có một IEnumerable<T>phương pháp mà tôi đang sử dụng để tìm các điều khiển trong trang WebForms.

Phương thức này là đệ quy và tôi gặp một số vấn đề khi trả về kiểu tôi muốn khi yield returntrả về giá trị của cuộc gọi đệ quy.

Mã của tôi trông như sau:

    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>();
            }
        }
    }

Điều này hiện gây ra lỗi "Không thể chuyển đổi loại biểu thức". Tuy nhiên, nếu phương thức này trả về kiểu IEnumerable<Object>, mã sẽ xây dựng, nhưng loại sai được trả về trong đầu ra.

Có cách nào để sử dụng yield returntrong khi cũng sử dụng đệ quy không?

Bên trong một phương thức trả về IEnumerable<T>, yield returnphải trả về T, không phải là một IEnumerable<T>.

Thay thế

yield return c.GetDeepControlsByType<T>();

với:

foreach (var x in c.GetDeepControlsByType<T>())
{
  yield return x;
}

Bạn cần nhường từng mục được mang lại bởi lệnh gọi đệ quy:

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)
        {
            foreach (Control control in c.GetDeepControlsByType<T>())
            {
                yield return control;
            }
        }
    }
}

Lưu ý rằng có một chi phí để đệ quy theo cách này - cuối cùng bạn sẽ tạo ra rất nhiều trình lặp, điều này có thể tạo ra vấn đề về hiệu suất nếu bạn có một cây điều khiển thực sự sâu. Nếu bạn muốn tránh điều đó, về cơ bản bạn cần tự thực hiện đệ quy trong phương thức, để đảm bảo chỉ có một trình lặp (máy trạng thái) được tạo. Xem câu hỏi này để biết thêm chi tiết và triển khai mẫu - nhưng điều này rõ ràng cũng thêm một số độ phức tạp nhất định.

Như Jon Skeet và Đại tá Panic lưu ý trong câu trả lời của họ, sử dụng yield returncác phương pháp đệ quy có thể gây ra vấn đề về hiệu suất nếu cây rất sâu.

Đây là một phương thức mở rộng không đệ quy chung , thực hiện một giao dịch theo chiều sâu của chuỗi cây:

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();
        }
    }
}

Không giống như giải pháp của Eric Lippert , RecursiveSelect hoạt động trực tiếp với các điều tra viên để không cần gọi Reverse (bộ đệm toàn bộ chuỗi trong bộ nhớ).

Sử dụng RecursiveSelect, phương thức ban đầu của OP có thể được viết lại đơn giản như thế này:

public static IEnumerable<Control> GetDeepControlsByType<T>(this Control control)
{
    return control.Controls.RecursiveSelect(c => c.Controls).Where(c => c is T);
}

Những người khác cung cấp cho bạn câu trả lời chính xác, nhưng tôi không nghĩ trường hợp của bạn được hưởng lợi từ năng suất.

Đây là một đoạn mà đạt được điều tương tự mà không mang lại.

public static IEnumerable<Control> GetDeepControlsByType<T>(this Control control)
{
   return control.Controls
                 .Where(c => c is T)
                 .Concat(control.Controls
                                .SelectMany(c =>c.GetDeepControlsByType<T>()));
}

Bạn cần trả lại các mục từ điều tra viên, chứ không phải chính điều tra viên, trong lần thứ hai của bạnyield 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)
        {
            foreach (Control ctrl in c.GetDeepControlsByType<T>())
            {
                yield return ctrl;
            }
        }
    }
}

Tôi nghĩ rằng bạn phải mang lại lợi nhuận cho mỗi điều khiển trong bảng liệt kê.

    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)
            {
                foreach (Control childControl in c.GetDeepControlsByType<T>())
                {
                    yield return childControl;
                }
            }
        }
    }

Cú pháp của Seredynski là chính xác, nhưng bạn nên cẩn thận để tránh yield returntrong các hàm đệ quy vì đó là một thảm họa cho việc sử dụng bộ nhớ. Xem https://stackoverflow.com/a/3970171/284795 nó có tỷ lệ bùng nổ với độ sâu (một chức năng tương tự đã sử dụng 10% bộ nhớ trong ứng dụng của tôi).

Một giải pháp đơn giản là sử dụng một danh sách và chuyển nó với đệ quy 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);
    }
}

Ngoài ra, bạn có thể sử dụng ngăn xếp và vòng lặp while để loại bỏ các cuộc gọi đệ quy https://codereview.stackexchange.com/a/5661/754

Mặc dù có nhiều câu trả lời hay ngoài kia, tôi vẫn sẽ nói thêm rằng có thể sử dụng các phương pháp LINQ để thực hiện điều tương tự ,.

Chẳng hạn, mã gốc của OP có thể được viết lại thành:

public static IEnumerable<Control> 
                           GetDeepControlsByType<T>(this Control control)
{
   return control.Controls.OfType<T>()
          .Union(control.Controls.SelectMany(c => c.GetDeepControlsByType<T>()));        
}