Реализация IEnumerable на древовидной структуре - PullRequest
Новая
       31

Реализация IEnumerable на древовидной структуре

3 голосов
/ 18 декабря 2009

На основании работы этих ребят:

Я пытаюсь реализовать помощник TreeView, который будет использоваться следующим образом: 

<%= Html.TreeView("records", 
                  Library.Instance.Records, 
                  r => r.Children, 
                  r => r.ID) %>

И древовидная структура определяется так: 

public class Tree<T> : TreeNode<T> where T : TreeNode<T>
{ }


public class TreeNode<T> : IDisposable where T : TreeNode<T>
{
    public T Parent { get; set; }
    public TreeNodeList<T> Children { get; set; }
}


public class TreeNodeList<T> : List<TreeNode<T>> where T : TreeNode<T>
{
    public T Parent;

    public T Add(T node)
    {
        base.Add(node);
        node.Parent = (T)Parent;
        return node;
    }

    public void Remove(T node)
    {
        if (node != null)
            node.Parent = null;
        base.Remove(node);
    }
}

И помощник TreeView имеет эту подпись: 

public static string TreeView<T>(this HtmlHelper htmlHelper, string treeId,
   IEnumerable<T> rootItems, Func<T, IEnumerable<T>> childrenProperty, 
   Func<T, string> itemContent, bool includeJavascript, string emptyContent)
{
    ...
}

Так что в результате мне нужно, чтобы моя структура дерева реализовала IEnumerable, чтобы я мог использовать его с помощником TreeView, и это приводит к вопросу: где и как мне реализовать IEnumerable в этой ситуации?

Ответы [ 2 ]

5 голосов
/ 18 декабря 2009

Эта статья Уэса Дайера хороша для тех, кто хочет реализовать итератор рекурсивного дерева:

http://blogs.msdn.com/wesdyer/archive/2007/03/23/all-about-iterators.aspx

1 голос
/ 18 декабря 2009

Я не совсем понимаю точные детали вашей древовидной структуры, но вот простая реализация, которая берет общее дерево узлов и рекурсивно отображает его в html-списках. 

public static string TreeView<T>(IEnumerable<T> rootItems,
                                 Func<T, IEnumerable<T>> childrenProperty,
                                 Func<T, string> itemContent)
{
    if (rootItems == null || !rootItems.Any()) return null;

    var builder = new StringBuilder();
    builder.AppendLine("<ul>");

    foreach (var item in rootItems)
    {
        builder.Append("  <li>").Append(itemContent(item)).AppendLine("</li>");
        var childContent = htmlHelper.TreeView(treeId,
                                               childrenProperty(item),
                                               childrenProperty,
                                               itemContent);

        if (childContent != null)
        {
            var indented = childContent.Replace(Environment.NewLine,
                                                Environment.NewLine + "  ");
            builder.Append("  ").AppendLine(indented);
        }
    }

    builder.Append("</ul>");
    return builder.ToString();
}

Класс узла, который я использую, относительно прост с двумя свойствами. 

public class Node<T>
{
    public Node(T data)
    {
        Data = data;
        Children = new List<Node<T>>();
    }

    public T Data { get; private set; }
    public ICollection<Node<T>> Children { get; private set; }
}

Вот некоторый тестовый код, который выводит дерево на консоль. 

var Records = new[] {
    new Node<string>("one") {
        Children = {
            new Node<string>("one-one") {
                Children = {
                    new Node<string>("one-one-one"),
                    new Node<string>("one-one-two"),
                    new Node<string>("one-one-three")
                }
            },
            new Node<string>("one-two"),
            new Node<string>("one-three")
        }
    },
    new Node<string>("two"),
    new Node<string>("three")
};
Console.WriteLine(TreeView(Records,
                           r => r.Children,
                           r => r.Data));

А вот результаты из приведенного выше кода. 

<ul>
  <li>one</li>
  <ul>
    <li>one-one</li>
    <ul>
      <li>one-one-one</li>
      <li>one-one-two</li>
      <li>one-one-three</li>
    </ul>
    <li>one-two</li>
    <li>one-three</li>
  </ul>
  <li>two</li>
  <li>three</li>
</ul>
...