Client/AntDesignClient/Gardener.Client.AntDesignUi.Base/Extensions/AntTreeExtension.cs

微信扫码关注公众号，代码整理好后第一时间通知你。

// -----------------------------------------------------------------------------
// 园丁,是个很简单的管理系统
//  gitee:https://gitee.com/hgflydream/Gardener 
//  issues:https://gitee.com/hgflydream/Gardener/issues 
// -----------------------------------------------------------------------------

namespace AntDesign
{
    public static class AntTreeExtension
    {
        
        /// <summary>
        /// 获取所有前辈，node
        /// </summary>
        /// <param name="node"></param>
        /// <returns></returns>
        public static List<TreeNode<T>> GetPredecessors<T>(this TreeNode<T> node)
        {
            List<TreeNode<T>> parents = new List<TreeNode<T>>();
            if (node.ParentNode != null)
            {
                parents.Add(node.ParentNode);
                parents.AddRange(node.ParentNode.GetPredecessors());
            }
            return parents;
        }

        /// <summary>
        /// 获取下一级所有node
        /// </summary>
        /// <param name="node"></param>
        /// <returns></returns>
        public static List<TreeNode<T>> ChildNodes<T>(this TreeNode<T> node)
        {
            TreeNode<T> childNode = node.FindFirstOrDefaultNode(x => x.ParentNode.Key.Equals(node.Key), true);
            if (childNode == null)
            {
                return new List<TreeNode<T>>();
            }
            return childNode.GetParentNodes();
        }
    }
}