首页 > 其他 > 详细

【转】无限极列表转换树结构的通用方法

时间:2020-08-29 23:07:29      阅读:69      评论:0      收藏:0      [点我收藏+]

原文链接:http://www.loogn.net/list2tree

参考链接:https://blog.csdn.net/qq_19244927/article/details/106481777

后台做菜单管理、商品分类管理的时候,我们建表一般会有个ParentId对应父ID,这样能形成无限极的分类。存在数据库中是列表数据,当我们返回前端的时候一般会处理成树形数据,方便前端展示,这个场景比较多,所以可以想到使用泛型来写一个算法模板。

首先定义数据库模型的接口:

    /// <summary>
    /// 水平对象接口,一般是数据库实体对象
    /// </summary>
    /// <typeparam name="TId"></typeparam>
    public interface ILevelModel<TId> where TId : struct
    {
        TId Id { get; set; }
        TId ParentId { get; set; }
    }

ID的类型一般是int类型,但是也有习惯使用long类型的,所以这里使用TId泛型来表示。

然后定义返回前端的接口:

    /// <summary>
    /// 有children属性的可内嵌本身的层级对象
    /// </summary>
    /// <typeparam name="T"></typeparam>
    public interface INestedModel<T>
    {
        ICollection<T> Children { get; set; }
    }

这个接口超级简单,就是一个子集合。

接下来就可以实现我们的通用方法了:

        /// <summary>
        /// 从水平list转变成嵌套tree列表
        /// </summary>
        /// <param name="targetCollection">目标tree列表</param>
        /// <param name="sourceData">所有源数据</param>
        /// <param name="parentId">第一级的parentId,默认0</param>
        /// <typeparam name="T">目标对象类型</typeparam>
        /// <typeparam name="S">原对象类型</typeparam>
        /// <typeparam name="IId">原对象id类型</typeparam>
        public static void ListToTree<T, S, IId>(ICollection<T> targetCollection, IEnumerable<S> sourceData,
            IId parentId = default(IId)) where T : INestedModel<T> where S : ILevelModel<IId> where IId : struct

        {
            foreach (var sysRes in sourceData.Where(x => x.ParentId.Equals(parentId)))
            {
                var info = SimpleMapper.Map<T>(sysRes);
                if (sourceData.Any(x => x.ParentId.Equals(sysRes.Id)))
                {
                    info.Children = new List<T>();
                    ListToTree(info.Children, sourceData, sysRes.Id);
                }

                targetCollection.Add(info);
            }
        }

这样基本就完成了,但是有可能有这样的情况,现有的类中属性名字不是Id、不是ParentId或者不是Children,这个时候上面的接口定义就不方便了,为了更加通用,我们可以另外增加两个接口,里面只包含方法,不使用属性:

    /// <summary>
    /// 水平对象接口,一般是数据库实体对象,版本2
    /// </summary>
    /// <typeparam name="IId"></typeparam>
    public interface ILevelModel2<out IId> where IId : struct
    {
        IId GetId();
        IId GetParentId();
    }

    /// <summary>
    /// 有children属性的可内嵌本身的层级对象,版本2
    /// </summary>
    /// <typeparam name="T"></typeparam>
    public interface INestedModel2<T>
    {
        ICollection<T> GetChildren();
        void SetChildren(ICollection<T> children);
    }

这样就更灵活了,然后在实现一个针对这两个接口的方法:

        /// <summary>
        /// 从水平list转变成嵌套tree列表
        /// </summary>
        /// <param name="targetCollection">目标tree列表</param>
        /// <param name="sourceData">所有源数据</param>
        /// <param name="parentId">第一级的parentId,默认0</param>
        /// <typeparam name="T">目标对象类型</typeparam>
        /// <typeparam name="S">原对象类型</typeparam>
        /// <typeparam name="IId">原对象id类型</typeparam>
        public static void ListToTree2<T, S, IId>(ICollection<T> targetCollection, IEnumerable<S> sourceData,
            IId parentId = default(IId)) where T : INestedModel2<T> where S : ILevelModel2<IId> where IId : struct

        {
            foreach (var sysRes in sourceData.Where(x => x.GetParentId().Equals(parentId)))
            {
                var info = SimpleMapper.Map<T>(sysRes);
                if (sourceData.Any(x => x.GetParentId().Equals(sysRes.GetId())))
                {
                    info.SetChildren(new List<T>());
                    ListToTree2(info.GetChildren(), sourceData, sysRes.GetId());
                }

                targetCollection.Add(info);
            }
        }

调用的例子:

    public class Res : ILevelModel<int>
    {
        public int Id { get; set; }
        public int ParentId { get; set; }
        public string Name { get; set; }
    }

    public class ResInfo : INestedModel<ResInfo>
    {
        public string Name { get; set; }
        public int Id { get; set; }
        public ICollection<ResInfo> Children { get; set; }
    }
        static void Main(string[] args)
        {
            var sourceData = new List<Res>();
            sourceData.Add(new Res() {Id = 1, Name = "第1个", ParentId = 0});
            sourceData.Add(new Res() {Id = 2, Name = "第2个", ParentId = 0});
            sourceData.Add(new Res() {Id = 3, Name = "第1-1个", ParentId = 1});
            sourceData.Add(new Res() {Id = 4, Name = "第1-2个", ParentId = 1});
            sourceData.Add(new Res() {Id = 5, Name = "第2-1个", ParentId = 2});
            sourceData.Add(new Res() {Id = 6, Name = "第2-1-1个", ParentId = 5});
            sourceData.Add(new Res() {Id = 7, Name = "第2-1-2个", ParentId = 5});


            var list = new List<ResInfo>();
            CollectionHelper.ListToTree(list, sourceData, 0);

            var json = JsonSerializer.Serialize(list, new JsonSerializerOptions()
            {
                WriteIndented = true,
                Encoder = JavaScriptEncoder.UnsafeRelaxedJsonEscaping
            });
            Console.WriteLine(json);
        }

标题:无限极列表转换树结构的通用方法
作者:loogn
地址:http://www.loogn.net/list2tree

【转】无限极列表转换树结构的通用方法

原文:https://www.cnblogs.com/binzhou75/p/13583722.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!