js根据树结构查找某个节点的下面的所有父节点或者子节点

我们先看一下树的结构：

  [
            {
                "id":6,
                "name":"测试",
                "parent_id":1,
                "children":[
                    {
                        "id":9,
                        "name":"测试1",
                        "parent_id":6,
                        "children":[
                            {
                                "id":20,
                                "name":"测试2",
                                "parent_id":9,
                                "children":[
                                    {
                                        "id":21,
                                        "name":"测试3",
                                        "parent_id":20,
                                    },
                                    {
                                        "id":22,
                                        "name":"测试4",
                                        "parent_id":20,
                                    },
                                    {
                                         "id":23,
                                        "name":"测试5",
                                        "parent_id":20,
                                    }
                                ], 
                            }
                        ], 
                    }, 
                ]

接着通过递归来实现即可，下面废话不多说，直接上代码：

       //查找父节点
      parentTreeNode (treeList, id) {
        var temp = []
        var searchFn = function (treeList, id) {
          for (var i = 0; i < treeList.length; i++) {
            var item = treeList[i]
            if (item.id === id) {
              temp.push(item)
              searchFn(treeList, item.parent_id)
              break
            } else {
              if (item.children) {
                searchFn(item.children, id)
              }
            }
          }
        }
        searchFn(treeList, id)
        return temp
   }

       //查找子节点
      childrenTreeNode (treeList, id) {
        var temp = []
        var searchFn = function (treeList, id) {
          for (var i = 0; i < treeList.length; i++) {
            var item = treeList[i]
            if (item.parent_id === id) {
              temp.push(item)
              searchFn(treeList, item.id)
              break
            } else {
              if (item.children) {
                searchFn(item.children, id)
              }
            }
          }
        }
        searchFn(treeList, id)
        return temp
   }