feat(tree): node map to node tree

tree/tree.go

@@ -50,3 +50,12 @@ func intoTree[T cmp.Ordered, E NodeTree[T, E]](nodeMaps map[T]E, rows []E, rootP
 	}
 	return root
 }
+
+// Map implement Node map to NodeTree
+func Map[I cmp.Ordered, E Node[I, U], U NodeTree[I, U]](rows []E) []U {
+	nodes := make([]U, 0, len(rows))
+	for _, v := range rows {
+		nodes = append(nodes, v.MapTree())
+	}
+	return nodes
+}

tree/tree_test.go

@@ -62,12 +62,7 @@ func TestTree(t *testing.T) {
 	tree1, err := json.MarshalIndent(gotTree1, " ", "  ")
 	require.NoError(t, err)
 
-	arrTree := make([]*DeptTree, 0, len(arr))
-	for _, v := range arr {
-		vv := *v
-		arrTree = append(arrTree, vv.MapTree())
-	}
-
+	arrTree := Map(arr)
 	gotTree2 := IntoTree2(arrTree, 0)
 	tree2, err := json.MarshalIndent(gotTree2, " ", "  ")
 	require.NoError(t, err)