golang 组成树形格式

该封装受到前端 js filter函数的启发看着特别简洁

一、封装一个函数

比较简单就是循环根据传入的 回调函数 进行过滤组成新的数组返回

func Filter[T any](arr []*T, f func(item *T) bool) (list []*T) {
	for _, el := range arr {
		if f(el) {
			list = append(list, el)
		}
	}
	return list
}

二、封装通用的树形方法

第一层先过滤出父级 第二层过滤出子级 如果子及存在那就就加到父级的Children中

type Tree[T any] interface {
	GeteIsQual(father *T, childId *T) bool
	SetChild(father *T, branchArr []*T)
	RetFather(father *T) bool
}
func ToTree[T any](list []*T, fun Tree[T]) []*T {
	return Filter(list, func(index int, father *T) bool {
		branchArr := Filter(list, func(i int, childId *T) bool {
			return fun.GeteIsQual(father, childId)
		})

		if len(branchArr) > 0 {
			fun.SetChild(father, branchArr)
		}
		return fun.RetFather(father)
	})
}

使用

package main

func main (){
	type Dept struct {
		id
		DeptName string  `json:"deptName"`
		Remark   string  ` json:"remark"`
		ParentId string  ` son:"parentId"`
		Children []*Dept `json:"children"`
	}
	func (d *Dept) GeteIsQual(father *Dept, childId *Dept) bool {
		return father.Id == childId.ParentId
	}
	func (d *Dept) SetChild(father *Dept, branchArr []*Dept) {

		father.Children = branchArr
	}
	func (d *Dept) RetFather(father *Dept) bool {
	// 顶级的ParentId这块可以看一下保存的时候ParentId 默认值是多少
		return father.ParentId == "0"
	}
	dept := []models.Dept{}
	// dept 是数据库查出来的数据这块就不展示了
	list = ToTree[Dept ](dept , &models.dept{})
}