hotgo/server/utility/tree/tree.go

// Package tree
// @Link  https://github.com/bufanyun/hotgo
// @Copyright  Copyright (c) 2023 HotGo CLI
// @Author  Ms <133814250@qq.com>
// @License  https://github.com/bufanyun/hotgo/blob/master/LICENSE
//
package tree

import (
	"fmt"
	"github.com/gogf/gf/v2/util/gconv"
	"strings"
)

var (
	pidName      = "pid"
	treeBeginCut = "tr_" // 树标识开头
	treeEndCut   = " "   // 树标识结尾
)

// GenLabel 生成关系树标识
func GenLabel(basic string, appendId int64) string {
	return fmt.Sprintf("%v%v%v%v", basic, treeBeginCut, appendId, treeEndCut)
}

// GetIds 获取上级ID集合
func GetIds(tree string) (ids []int64) {
	idsStr := strings.Split(tree, treeEndCut)
	if len(idsStr) == 0 {
		return
	}

	for _, v := range idsStr {
		newId := gconv.Int64(strings.ReplaceAll(v, treeBeginCut, ""))
		if newId > 0 {
			ids = append(ids, newId)
		}
	}

	return
}

/////////////////////////// 转换类

// GenTree 生成关系树
func GenTree(menus []map[string]interface{}) (realMenu []map[string]interface{}) {
	if len(menus) < 1 {
		return nil
	}

	minPid := GetMinPid(menus)
	formatMenu := make(map[int]map[string]interface{})
	for _, m := range menus {
		formatMenu[gconv.Int(m["id"])] = m
		if gconv.Int(m[pidName]) == minPid {
			realMenu = append(realMenu, m) // 需要返回的顶级菜单
		}
	}
	// 得益于都是地址操作,可以直接往父级塞
	for _, m := range formatMenu {
		if formatMenu[gconv.Int(m[pidName])] == nil {
			continue
		}
		if formatMenu[gconv.Int(m[pidName])]["children"] == nil {
			formatMenu[gconv.Int(m[pidName])]["children"] = []map[string]interface{}{}
		}

		formatMenu[gconv.Int(m[pidName])]["children"] = append(formatMenu[gconv.Int(m[pidName])]["children"].([]map[string]interface{}), m)
	}
	return
}

func GetMinPid(menus []map[string]interface{}) int {
	index := -1
	for _, m := range menus {
		if index == -1 {
			index = gconv.Int(m[pidName])
			continue
		}
		if gconv.Int(m[pidName]) < index {
			index = gconv.Int(m[pidName])
		}
	}

	if index == -1 {
		return 0
	}
	return index
}