MiFaZhan
59eb34976a
iotlcm-admin-svr：管理后台服务。 - Go module：iotlcm → git.5iot.com/iotlcm-public/iotlcm-admin-svr（729 文件） - 凭据与地址脱敏：JWT signingKey（含代码里的 MustGet 默认值）、 MySQL DSN、Redis 口令、告警邮件发件人邮箱全部改为占位； 私网 IP → 文档地址段；库名 xiujie_admin → iotlcm - 真实手机号/邮箱 → 合法 11 位编造号 / example.com - 移除内部资产：CI 触发器、过时 DDL（sql/db_iotlcm.sql）、 私有依赖 go get 行、明文 registry 登录口令 - 启动 banner 统一 IOTLCM 方块字 + 社区惯例署名 - Dockerfile 改用公共基础镜像 - 拼错的 RADEME.md → README.md，README 重写 - 需要真实数据库的用例改为可用 IOTLCM_TEST_WITH_DB=1 显式启用 - 补 Apache-2.0 LICENSE，LF 行尾 + .gitattributes 数据库 schema 不随本仓发布（生产 DDL 已过时），README 已说明。
110 lines
2.7 KiB
Go
110 lines
2.7 KiB
Go
// package tree
|
|
// @Link https://github.com/xiujiecn/xiu-admin
|
|
// @Copyright Copyright (c) 2025 LiXiujie
|
|
// @Author Lxj <li@xiujie.cn>
|
|
// @License https://github.com/xiujiecn/xiu-admin/blob/master/LICENSE
|
|
package tree
|
|
|
|
import (
|
|
"fmt"
|
|
"strings"
|
|
|
|
"github.com/gogf/gf/v2/util/gconv"
|
|
)
|
|
|
|
const (
|
|
treeDefaultId = "id" // 树ID
|
|
treeDefaultPid = "pid" // 上级树ID
|
|
treeDefaultChildren = "children" // 下级属性
|
|
treeBeginCut = "tr_" // 树标识开头
|
|
treeEndCut = " " // 树标识结尾
|
|
)
|
|
|
|
// GenOption 生成选项
|
|
type GenOption struct {
|
|
IdField string
|
|
PidField string
|
|
ChildrenField string
|
|
}
|
|
|
|
// GenLabel 生成关系树标识
|
|
func GenLabel(basic string, appendId int64) string {
|
|
return fmt.Sprintf("%v%v%v%v", basic, treeBeginCut, appendId, treeEndCut)
|
|
}
|
|
|
|
// GetIdLabel 获取指定Id的树标签
|
|
func GetIdLabel(id int64) string {
|
|
return fmt.Sprintf("%v%v%v", treeBeginCut, id, 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{}) {
|
|
return GenTreeWithField(menus, GenOption{
|
|
IdField: treeDefaultId,
|
|
PidField: treeDefaultPid,
|
|
ChildrenField: treeDefaultChildren,
|
|
})
|
|
}
|
|
|
|
// GenTreeWithField 生成关系树 自定义生成属性
|
|
func GenTreeWithField(menus []map[string]interface{}, op GenOption) (realMenu []map[string]interface{}) {
|
|
if len(menus) < 1 {
|
|
return
|
|
}
|
|
|
|
minPid := GetMinPid(menus, op.PidField)
|
|
formatMenu := make(map[int]map[string]interface{})
|
|
for _, m := range menus {
|
|
formatMenu[gconv.Int(m[op.IdField])] = m
|
|
if gconv.Int(m[op.PidField]) == minPid {
|
|
realMenu = append(realMenu, m) // 需要返回的顶级菜单
|
|
}
|
|
}
|
|
// 得益于都是地址操作,可以直接往父级塞
|
|
for _, m := range formatMenu {
|
|
if formatMenu[gconv.Int(m[op.PidField])] == nil {
|
|
continue
|
|
}
|
|
if formatMenu[gconv.Int(m[op.PidField])][op.ChildrenField] == nil {
|
|
formatMenu[gconv.Int(m[op.PidField])][op.ChildrenField] = []map[string]interface{}{}
|
|
}
|
|
|
|
formatMenu[gconv.Int(m[op.PidField])][op.ChildrenField] = append(formatMenu[gconv.Int(m[op.PidField])][op.ChildrenField].([]map[string]interface{}), m)
|
|
}
|
|
return
|
|
}
|
|
|
|
func GetMinPid(menus []map[string]interface{}, pidField string) int {
|
|
index := -1
|
|
for _, m := range menus {
|
|
if index == -1 {
|
|
index = gconv.Int(m[pidField])
|
|
continue
|
|
}
|
|
if gconv.Int(m[pidField]) < index {
|
|
index = gconv.Int(m[pidField])
|
|
}
|
|
}
|
|
|
|
if index == -1 {
|
|
return 0
|
|
}
|
|
return index
|
|
}
|