C# 将扁平状数据链接成树状结构的通用方法

在项目中经常会遇到从数据库查询数据绑定到TreeVIew上，这时我们需要将查询出来的数据转换成树形结构数据，每次写觉得工作会很重复，所以写了一个通用的转换类。

 

第一步，我们需要建一个基类，这个基类的意义主要是扩展数据库实体类做连接用，用于确定树形结构中节点与子项的关系，

其中Parent为当前节点的父节点，Children为当前节点的子节点，IsLinked是判断当前节点是否已连接，用于防止数据中有循环依赖导致创建树的时候形成死循环。

TreeModel基类

public class TreeBase<T>
    {
        private bool isLinked = false;
        /// <summary>
        /// 是否已创建连接
        /// </summary>
        public bool IsLinked
        {
            get { return isLinked; }
            set { isLinked = value; }
        }

        /// <summary>
        /// 父节点
        /// </summary>
        public T Parent { get; set; }

        /// <summary>
        /// 子节点
        /// </summary>
        public ObservableCollection<T> Children { get; set; }
    }

第二步，我们需要创建一个连接方法，将输入的扁平状数据转换为树状结构，我希望在使用的时候可以自己指定实体的哪个属性为ID,哪个为父ID

,这里我们使用了Expression，这样就可以使用linq表达式去指定属性了，剩下的就是利用反射获取实体的值与递归连接了，这样一个简单的通用创建树的方法就有了。

    public class TreeHelper
    {
        /// <summary>
        /// 创建树
        /// </summary>
        /// <typeparam name="T">实体类型</typeparam>
        /// <param name="root">根节点</param>
        /// <param name="list">所有数据</param>
        /// <param name="idProperty">节点唯一标识属性表达式</param>
        /// <param name="parentIdProperty">父节点属性表达式</param>
        public static void CreateTree<T>(T root, IList<T> list, string idPropertyName, string parentIdPropertyName) where T : TreeBase<T>
        {
            root.Children = new ObservableCollection<T>();
            list.Where(e => (string)GetPropertyValue(e, parentIdPropertyName) == (string)GetPropertyValue(root, idPropertyName) && !e.IsLinked).ToList().ForEach(e => { root.Children.Add(e); e.IsLinked = true; });
            foreach (var leaf in root.Children)
            {
                leaf.Parent = root;
                CreateTree<T>(leaf, list, idPropertyName, parentIdPropertyName);
            }
        }

        /// <summary>
        /// 创建多个根节点的树
        /// </summary>
        /// <typeparam name="T">实体类型</typeparam>
        /// <param name="root">根节点</param>
        /// <param name="list">所有数据</param>
        /// <param name="idProperty">节点唯一标识属性表达式</param>
        /// <param name="parentIdProperty">父节点属性表达式</param>
        public static ObservableCollection<T> CreateTree<T>(IList<T> list, Expression<Func<T, object>> idProperty, Expression<Func<T, object>> parentIdProperty) where T : TreeBase<T>
        {
            //查找父节点不存在的leaf,作伪根节点
            var roots = new ObservableCollection<T>();
            var idPropertyName = GetMemberName(idProperty);
            var parentIdPropertyName = GetMemberName(parentIdProperty);
            list.Where(e => list.Count(item =>
                    (string)GetPropertyValue(item, idPropertyName) == (string)GetPropertyValue(e, parentIdPropertyName)) == 0).ToList().ForEach(e => roots.Add(e));
            foreach (var root in roots)
            {
                CreateTree<T>(root, list, idPropertyName, parentIdPropertyName);
            }
            return roots;
        }

        private static object GetPropertyValue<T>(T t, string propertyName)
        {
            return t.GetType().GetProperty(propertyName).GetValue(t, null);
        }

        private static string GetMemberName<T, TMember>(Expression<Func<T, TMember>> propertySelector)
        {
            var propertyExp = propertySelector.Body as MemberExpression;
            if (propertyExp == null)
            {
                throw new ArgumentException("不合理的表达式!");
            }
            return propertyExp.Member.Name;
        }



    }

 

 

 使用示例：

先声明一个绑定实体,继承自TreeBase

public class Item : TreeBase<Item>
{
　　public string Id{get;set;}
　　public string ParentID{get;set;}
　　public string DisplayText{get;set;}
}

创建测试数据

            IList<Item> data = new List<Item>();
            var group1= new Item(){ Id = Guid.NewGuid().ToString(), DisplayText="Root"};
            var item1 = new Item { Id = Guid.NewGuid().ToString(), ParentId = group1.Id, DisplayText = "Leaf1" };
            var item2 = new Item { Id = Guid.NewGuid().ToString(), ParentId = group1.Id, DisplayText = "Leaf2" };
            var item3 = new Item { Id = Guid.NewGuid().ToString(), ParentId = group1.Id, DisplayText = "Leaf3" };
            var item4 = new Item() { Id = Guid.NewGuid().ToString(), ParentId = item3.Id, DisplayText = "Leaf3:Leaf1" };
            var item5 = new Item() { Id = Guid.NewGuid().ToString(), ParentId = item3.Id, DisplayText = "Leaf3:Leaf2" };
            var item6 = new Item() { Id = Guid.NewGuid().ToString(), ParentId = item3.Id, DisplayText = "Leaf3:Leaf3" };
            data.Add(group1);
            data.Add(item1);
            data.Add(item2);
            data.Add(item3);
            data.Add(item4);
            data.Add(item5);
            data.Add(item6);

界面模板绑定

        <TreeView Height="245" HorizontalAlignment="Left" Margin="46,35,0,0" Name="treeView1" VerticalAlignment="Top" Width="156" >
            <TreeView.ItemTemplate>
                <HierarchicalDataTemplate ItemsSource="{Binding Children}">
                    <TextBlock Text="{Binding DisplayText}"></TextBlock>
                </HierarchicalDataTemplate>
            </TreeView.ItemTemplate>
        </TreeView>

后台数据绑定

            treeView1.ItemsSource = TreeHelper.CreateTree<Item>(data, item => item.Id, item => item.ParentId);

以后绑定树形数据是不是很简单呢？

这种方法不适应主键为Guid的数据类型

因为以下代码无法获取Guid类型的名称，欢迎各位大叔，大婶，大神们指点

        private static string GetMemberName<T, TMember>(Expression<Func<T, TMember>> propertySelector)
        {
            var propertyExp = propertySelector.Body as MemberExpression;
            if (propertyExp == null)
            {
                throw new ArgumentException("不合理的表达式!");
            }
            return propertyExp.Member.Name;
        }