Masuit.Tools/Test/Masuit.Tools.Abstractions.Test/Tree/TreeTest.cs

using System.Collections.Generic;
using System.Linq;
using Masuit.Tools.Models;
using Xunit;

namespace Masuit.Tools.Abstractions.Test.Tree;

public class TreeTest
{
    [Fact]
    public void Can_BuildTree()
    {
        // arrange
        var list = new List<MyClass>()
        {
            new MyClass
            {
                Name = "Root",
                Id = 1
            },
            new MyClass
            {
                Name = "Root",
                Id = 20000
            }
        };
        for (int i = 2; i < 1500; i++)
        {
            list.Add(new MyClass
            {
                Name = $"这是第{i}个子节点",
                Id = i,
                ParentId = (i - 1)
            });
        }

        for (int i = 20001; i < 40000; i++)
        {
            list.Add(new MyClass
            {
                Name = $"这是第{i}个子节点",
                Id = i,
                ParentId = (i - 1)
            });
        }

        // act
        var tree = list.ToTree();

        // assert
        Assert.Equal(tree[0].Children.FirstOrDefault().Children.FirstOrDefault().Children.FirstOrDefault().Children.FirstOrDefault().Children.FirstOrDefault().Children.FirstOrDefault().Children.FirstOrDefault().Id, 8);
        Assert.Equal(tree.Count, 2);
        Assert.Equal(tree[0].AllChildren().Count, 1498);
    }

    [Fact]
    public void Can_BuildTree2()
    {
        // arrange
        var list = new List<MyClass2>()
        {
            new MyClass2
            {
                Name = "Root",
                Id = "1"
            },
            new MyClass2
            {
                Name = "Root",
                Id = "2000"
            }
        };
        for (int i = 2; i < 1500; i++)
        {
            list.Add(new MyClass2
            {
                Name = $"这是第{i}个子节点",
                Id = i.ToString(),
                ParentId = (i - 1).ToString()
            });
        }

        for (int i = 2001; i < 4000; i++)
        {
            list.Add(new MyClass2
            {
                Name = $"这是第{i}个子节点",
                Id = i.ToString(),
                ParentId = (i - 1).ToString()
            });
        }

        // act
        var tree = list.ToTree();

        // assert
        Assert.Equal(tree[0].Children.FirstOrDefault().Children.FirstOrDefault().Children.FirstOrDefault().Children.FirstOrDefault().Children.FirstOrDefault().Children.FirstOrDefault().Children.FirstOrDefault().Id, "8");
        Assert.Equal(tree.Count, 2);
        Assert.Equal(tree[0].AllChildren().Count, 1498);
    }
}

internal class MyClass : ITree<MyClass>, ITreeEntity<MyClass, int>
{
    /// <summary>
    /// 父节点
    /// </summary>
    public MyClass Parent { get; set; }

    /// <summary>
    /// 子级
    /// </summary>
    public ICollection<MyClass> Children { get; set; }

    /// <summary>
    /// 名字
    /// </summary>
    public string Name { get; set; }

    /// <summary>
    /// 主键id
    /// </summary>
    public int Id { get; set; }

    /// <summary>
    /// 父级id
    /// </summary>
    public int? ParentId { get; set; }
}

internal class MyClass2 : ITree<MyClass2>, ITreeEntity<MyClass2>
{
    /// <summary>
    /// 父节点
    /// </summary>
    public MyClass2 Parent { get; set; }

    /// <summary>
    /// 子级
    /// </summary>
    public ICollection<MyClass2> Children { get; set; }

    /// <summary>
    /// 名字
    /// </summary>
    public string Name { get; set; }

    /// <summary>
    /// 主键id
    /// </summary>
    public string Id { get; set; }

    /// <summary>
    /// 父级id
    /// </summary>
    public string ParentId { get; set; }
}