添加项目文件。

2023-03-13 15:00:34 +08:00
parent 42bf06ca3e
commit 1d73df3235
1205 changed files with 185078 additions and 0 deletions
--- a/common/Tnb.Common/Security/QueryTreeHelper.cs
+++ b/common/Tnb.Common/Security/QueryTreeHelper.cs
@@ -0,0 +1,132 @@
+using JNPF.DependencyInjection;
+using System.Data;
+using System.Reflection;
+
+namespace JNPF.Common.Security;
+
+/// <summary>
+/// 树形结构查询
+/// 版 本：V3.2.0
+/// 版 权：拓通智联科技有限公司（http://www.tuotong-tech.com）
+/// 作 者：JNPF开发平台组.
+/// </summary>
+[SuppressSniffer]
+public static class QueryTreeHelper
+{
+    /// <summary>
+    /// 递归查询.
+    /// </summary>
+    /// <typeparam name="T">输出模型.</typeparam>
+    /// <param name="data">数据源.</param>
+    /// <param name="condition">查询条件.</param>
+    /// <param name="idSelector">主键.</param>
+    /// <param name="parentIdSelector">上级.</param>
+    /// <returns></returns>
+    public static List<T> TreeWhere<T>(this List<T> data, Predicate<T> condition, Func<T, string> idSelector, Func<T, string> parentIdSelector)
+    {
+        List<T> locateList = data.FindAll(condition);
+        List<T> treeList = new List<T>();
+        foreach (T entity in locateList)
+        {
+            treeList.Add(entity);
+            T currentNode = entity;
+            while (true)
+            {
+                string parentId = parentIdSelector(currentNode);
+                if (parentId == null)
+                    break;
+                T upRecord = data.Find(a => idSelector(a) == parentId);
+                if (upRecord != null)
+                {
+                    treeList.Add(upRecord);
+                    currentNode = upRecord;
+                }
+                else
+                {
+                    break;
+                }
+            }
+        }
+
+        return treeList.Distinct().ToList();
+    }
+
+    /// <summary>
+    /// 递归查询.
+    /// </summary>
+    /// <param name="data">数据源.</param>
+    /// <param name="condition">查询条件.</param>
+    /// <param name="idSelector">主键.</param>
+    /// <param name="parentIdSelector">上级.</param>
+    /// <returns></returns>
+    public static DataTable TreeWhere(this DataTable data, string condition, string idSelector = "F_Id", string parentIdSelector = "F_ParentId")
+    {
+        DataRow[] drs = data.Select(condition);
+        DataTable treeTable = data.Clone();
+        foreach (DataRow dr in drs)
+        {
+            treeTable.ImportRow(dr);
+            string pId = dr[parentIdSelector].ToString();
+            while (true)
+            {
+                if (string.IsNullOrEmpty(pId) && pId == "0")
+                    break;
+                DataRow[] pdrs = data.Select(idSelector + "='" + pId + "'");
+                if (pdrs.Length > 0)
+                {
+                    treeTable.ImportRow(pdrs[0]);
+                    pId = pdrs[0][parentIdSelector].ToString();
+                }
+                else
+                {
+                    break;
+                }
+            }
+        }
+
+        return treeTable.DefaultView.ToTable(true);
+    }
+
+    /// <summary>
+    /// 获取全部子节点.
+    /// </summary>
+    /// <typeparam name="T">输出模型.</typeparam>
+    /// <param name="data">数据源.</param>
+    /// <param name="idValue">主键值.</param>
+    /// <param name="idSelector">主键.</param>
+    /// <param name="parentIdSelector">上级.</param>
+    /// <returns></returns>
+    public static List<T> TreeChildNode<T>(this List<T> data, string idValue, Func<T, string> idSelector, Func<T, string> parentIdSelector)
+    {
+        T thisEntity = data.Find(a => idSelector(a) == idValue);
+        foreach (var prop in from PropertyInfo prop in thisEntity.GetType().GetProperties()
+                             where prop.Name == "ParentId" || prop.Name == "parentId"
+                             select prop)
+        {
+            prop.SetValue(thisEntity, "0", null);
+            break;
+        }
+
+        List<T> treeList = new List<T>();
+        treeList.Add(thisEntity);
+        ChildNode(data, idValue, idSelector, parentIdSelector, ref treeList);
+        return treeList;
+    }
+
+    #region Method
+
+    private static void ChildNode<T>(this List<T> data, string idValue, Func<T, string> idSelector, Func<T, string> parentIdSelector, ref List<T> treeNodes)
+    {
+        List<T> locateList = data.FindAll(a => parentIdSelector(a) == idValue);
+        if (locateList.Count > 0)
+        {
+            foreach (var item in locateList)
+            {
+                treeNodes.Add(item);
+                ChildNode(data, idSelector(item), idSelector, parentIdSelector, ref treeNodes);
+            }
+        }
+    }
+
+    #endregion
+}