From e68555e9762a244b2a6440a30dcc9b62233fe522 Mon Sep 17 00:00:00 2001
From: yubaolee <yubaolee@163.com>
Date: Fri, 14 Oct 2016 17:39:21 +0800
Subject: [PATCH] routine update

---
 Infrastructure/GenericHelpers.cs | 42 ++++++++++++++++++++++++++++++++
 Infrastructure/TreeItem.cs       | 10 ++++++++
 2 files changed, 52 insertions(+)
 create mode 100644 Infrastructure/GenericHelpers.cs
 create mode 100644 Infrastructure/TreeItem.cs

diff --git a/Infrastructure/GenericHelpers.cs b/Infrastructure/GenericHelpers.cs
new file mode 100644
index 00000000..43182db4
--- /dev/null
+++ b/Infrastructure/GenericHelpers.cs
@@ -0,0 +1,42 @@
+using System;
+using System.Collections.Generic;
+using System.Linq;
+
+namespace Infrastructure
+{
+    /// <summary>
+    /// List转成Tree
+    /// <para>李玉宝新增于2016-10-09 19:54:07</para>
+    /// </summary>
+    public static class GenericHelpers
+    {
+        /// <summary>
+        /// Generates tree of items from item list
+        /// </summary>
+        /// 
+        /// <typeparam name="T">Type of item in collection</typeparam>
+        /// <typeparam name="K">Type of parent_id</typeparam>
+        /// 
+        /// <param name="collection">Collection of items</param>
+        /// <param name="idSelector">Function extracting item's id</param>
+        /// <param name="parentIdSelector">Function extracting item's parent_id</param>
+        /// <param name="rootId">Root element id</param>
+        /// 
+        /// <returns>Tree of items</returns>
+        public static IEnumerable<TreeItem<T>> GenerateTree<T, K>(
+            this IEnumerable<T> collection,
+            Func<T, K> idSelector,
+            Func<T, K> parentIdSelector,
+            K rootId = default(K))
+        {
+            foreach (var c in collection.Where(c => parentIdSelector(c).Equals(rootId)))
+            {
+                yield return new TreeItem<T>
+                {
+                    Item = c,
+                    Children = collection.GenerateTree(idSelector, parentIdSelector, idSelector(c))
+                };
+            }
+        }
+    }
+}
\ No newline at end of file
diff --git a/Infrastructure/TreeItem.cs b/Infrastructure/TreeItem.cs
new file mode 100644
index 00000000..c4db87bd
--- /dev/null
+++ b/Infrastructure/TreeItem.cs
@@ -0,0 +1,10 @@
+using System.Collections.Generic;
+
+namespace Infrastructure
+{
+    public class TreeItem<T>
+    {
+        public T Item { get; set; }
+        public IEnumerable<TreeItem<T>> Children { get; set; }
+    }
+}
\ No newline at end of file