feat: 😀增加数据集合拓展类

Admin.NET/Admin.NET.Core/Extension/EnumerableExtension.cs

@@ -0,0 +1,109 @@
+// Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
+// 
+// 本项目主要遵循 MIT 许可证和 Apache 许可证（版本 2.0）进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
+// 
+// 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动！任何基于本项目二次开发而产生的一切法律纠纷和责任，我们不承担任何责任！
+
+namespace Admin.NET.Core;
+
+/// <summary>
+/// 数据集合拓展类
+/// </summary>
+public static class EnumerableExtension 
+{
+    private static readonly ConcurrentDictionary<string, PropertyInfo> PropertyCache = new();
+    
+    /// <summary>
+    /// 查询有父子关系的数据集
+    /// </summary>
+    /// <param name="list">数据集</param>
+    /// <param name="idExpression">主键ID字段</param>
+    /// <param name="parentIdExpression">父级字段</param>
+    /// <param name="topParentIdValue">顶级节点父级字段值</param>
+    /// <param name="isContainOneself">是否包含顶级节点本身</param>
+    /// <returns></returns>
+    public static IEnumerable<T> ToChildList<T, P>(this IEnumerable<T> list,
+        Expression<Func<T, P>> idExpression,
+        Expression<Func<T, P>> parentIdExpression,
+        object topParentIdValue,
+        bool isContainOneself = true)
+    {
+        if (list == null || !list.Any()) return Enumerable.Empty<T>();
+
+        var propId = GetPropertyInfo(idExpression);
+        var propParentId = GetPropertyInfo(parentIdExpression);
+
+        // 查找所有顶级节点
+        var topNodes = list.Where(item => Equals(propId.GetValue(item), topParentIdValue)).ToList();
+
+        return TraverseHierarchy(list, propId, propParentId, topNodes, isContainOneself);
+    }
+
+    /// <summary>
+    /// 查询有父子关系的数据集
+    /// </summary>
+    /// <param name="list">数据集</param>
+    /// <param name="idExpression">主键ID字段</param>
+    /// <param name="parentIdExpression">父级字段</param>
+    /// <param name="topLevelPredicate">顶级节点的选择条件</param>
+    /// <param name="isContainOneself">是否包含顶级节点本身</param>
+    /// <returns></returns>
+    public static IEnumerable<T> ToChildList<T, P>(this IEnumerable<T> list,
+        Expression<Func<T, P>> idExpression,
+        Expression<Func<T, P>> parentIdExpression,
+        Expression<Func<T, bool>> topLevelPredicate,
+        bool isContainOneself = true)
+    {
+        if (list == null || !list.Any()) return Enumerable.Empty<T>();
+
+        // 获取顶级节点
+        var topNodes = list.Where(topLevelPredicate.Compile()).ToList();
+
+        if (!topNodes.Any()) return Enumerable.Empty<T>();
+
+        var idPropertyInfo = GetPropertyInfo(idExpression);
+        var parentPropertyInfo = GetPropertyInfo(parentIdExpression);
+
+        return TraverseHierarchy(list, idPropertyInfo, parentPropertyInfo, topNodes, isContainOneself);
+    }
+
+    /// <summary>
+    /// 辅助方法，从表达式中提取属性信息并使用临时缓存
+    /// </summary>
+    private static PropertyInfo GetPropertyInfo<T, P>(Expression<Func<T, P>> expression)
+    {
+        // 使用 ConcurrentDictionary 确保线程安全
+        return PropertyCache.GetOrAdd(typeof(T).FullName + "." + ((MemberExpression)expression.Body).Member.Name, k =>
+        {
+            if (expression.Body is UnaryExpression { Operand: MemberExpression member }) return (PropertyInfo)member.Member;
+            if (expression.Body is MemberExpression memberExpression) return (PropertyInfo)memberExpression.Member;
+            throw Oops.Oh("表达式必须是一个属性访问: " + expression);
+        });
+    }
+
+    /// <summary>
+    /// 使用队列遍历层级结构
+    /// </summary>
+    private static IEnumerable<T> TraverseHierarchy<T>(IEnumerable<T> list,
+        PropertyInfo idPropertyInfo,
+        PropertyInfo parentPropertyInfo,
+        List<T> topNodes,
+        bool isContainOneself)
+    {
+        var queue = new Queue<T>(topNodes);
+        var result = new HashSet<T>(topNodes);
+
+        while (queue.Count > 0)
+        {
+            var currentNode = queue.Dequeue();
+            var children = list.Where(item => Equals(parentPropertyInfo.GetValue(item), idPropertyInfo.GetValue(currentNode))).ToList();
+            children.Where(child => result.Add(child)).ForEach(child => queue.Enqueue(child));
+        }
+        if (isContainOneself) return result;
+
+        // 如果不需要包含顶级节点本身，则移除它们
+        topNodes.ForEach(e => result.Remove(e));
+        
+        return result;
+    }
+}