|
|
package BaseMenuService
|
|
|
|
|
|
import (
|
|
|
"dsBaseWeb/Business/BaseMenu/BaseMenuProto"
|
|
|
"dsBaseWeb/Utils/CommonUtil"
|
|
|
"encoding/json"
|
|
|
"strings"
|
|
|
)
|
|
|
|
|
|
func PageBaseMenuInfo(appId string, identityId int32) (*BaseMenuProto.Reply, error) {
|
|
|
Reply, err := PageBaseMenu(BaseMenuProto.QueryArg{AppId: appId, IdentityId: identityId})
|
|
|
Reply.List = convertPositionTree(Reply.List)
|
|
|
return Reply, err
|
|
|
}
|
|
|
|
|
|
func GetBaseMenuInfo(memuId string) (*BaseMenuProto.Reply, error) {
|
|
|
Reply, err := GetBaseMenu(BaseMenuProto.ModelArg{MenuId: memuId})
|
|
|
return Reply, err
|
|
|
}
|
|
|
|
|
|
func AddBaseMenuInfo(appId string, parentId string, identityId int32, menuName string, menuCode string, menuUrl string, menuIcon string) (*BaseMenuProto.Reply, error) {
|
|
|
Reply, err := AddBaseMenu(BaseMenuProto.ModelArg{AppId: appId, ParentId: parentId, IdentityId: identityId, MenuName: menuName, MenuCode: menuCode, MenuUrl: menuUrl, MenuIcon: menuIcon})
|
|
|
return Reply, err
|
|
|
}
|
|
|
|
|
|
func DeleteBaseMenuInfo(ids string) (*BaseMenuProto.Reply, error) {
|
|
|
idsArr := strings.Split(ids, ",")
|
|
|
Reply, err := DeleteBaseMenu(BaseMenuProto.DeleteIdsArg{Ids: idsArr})
|
|
|
return Reply, err
|
|
|
}
|
|
|
|
|
|
func UpdateBaseMenuInfo(menuId string, menuName string, menuCode string, menuUrl string, menuIcon string) (*BaseMenuProto.Reply, error) {
|
|
|
Reply, err := UpdateBaseMenu(BaseMenuProto.ModelArg{MenuId: menuId, MenuName: menuName, MenuCode: menuCode, MenuUrl: menuUrl, MenuIcon: menuIcon})
|
|
|
return Reply, err
|
|
|
}
|
|
|
|
|
|
func SetMenuSortInfo(menuId string, direction int32) (*BaseMenuProto.Reply, error) {
|
|
|
Reply, err := SetMenuSort(BaseMenuProto.SetMenuSortArg{MenuId: menuId, Direction: direction})
|
|
|
return Reply, err
|
|
|
}
|
|
|
|
|
|
func convertPositionTree(listStr string) string {
|
|
|
list := CommonUtil.ConvertJsonStringToMapArray(listStr)
|
|
|
ListOrgNode := make([]*OrgNode, 0)
|
|
|
for i := range list {
|
|
|
record := list[i]
|
|
|
//添加数据
|
|
|
a := OrgNode{
|
|
|
MenuId: record["menu_id"].(string),
|
|
|
MenuName: record["menu_name"].(string),
|
|
|
ParentId: record["parent_id"].(string),
|
|
|
MenuCode: record["menu_code"].(string),
|
|
|
MenuIcon: record["menu_icon"].(string),
|
|
|
AppId: record["app_id"].(string),
|
|
|
SortId: record["sort_id"].(float64),
|
|
|
}
|
|
|
ListOrgNode = append(ListOrgNode, &a)
|
|
|
}
|
|
|
//结构体的SortId根据数据库获取时，使用order by方法获取有序的子元素集合，这里就不在代码中进行排序了。
|
|
|
|
|
|
//从根节点开始进行构建,递归调用生成tree
|
|
|
startOrgNode := ListOrgNode[0]
|
|
|
makeTree(ListOrgNode, startOrgNode)
|
|
|
//根据结构体生成json数据
|
|
|
jsonByte, _ := json.Marshal(startOrgNode)
|
|
|
|
|
|
return string(jsonByte)
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：部门实体Struct
|
|
|
作者：黄海
|
|
|
时间：2020-06-03
|
|
|
*/
|
|
|
type OrgNode struct {
|
|
|
MenuId string `json:"menu_id"`
|
|
|
MenuName string `json:"menu_name"`
|
|
|
ParentId string `json:"parent_id"`
|
|
|
MenuCode string `json:"menu_code"`
|
|
|
MenuIcon string `json:"menu_icon"`
|
|
|
AppId string `json:"app_id"`
|
|
|
SortId float64 `json:"sort_id"`
|
|
|
Children []*OrgNode `json:"children,omitempty"`
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能:从指定节点开始构建树状结构
|
|
|
作者：黄海
|
|
|
时间：2020-06-03
|
|
|
*/
|
|
|
func makeTree(allNode []*OrgNode, node *OrgNode) {
|
|
|
children, _ := haveChild(allNode, node) //是不是有子节点
|
|
|
if children != nil {
|
|
|
node.Children = append(node.Children, children[0:]...) //添加子节点
|
|
|
for _, v := range children { //查询子节点的子节点，并添加到子节点
|
|
|
_, has := haveChild(allNode, v)
|
|
|
if has {
|
|
|
makeTree(allNode, v) //递归添加节点
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
}
|
|
|
|
|
|
/**
|
|
|
功能：判断是不是有子节点
|
|
|
作者：黄海
|
|
|
时间：2020-06-03
|
|
|
*/
|
|
|
func haveChild(AllNode []*OrgNode, node *OrgNode) (children []*OrgNode, yes bool) {
|
|
|
for _, v := range AllNode {
|
|
|
if v.ParentId == node.MenuId {
|
|
|
children = append(children, v)
|
|
|
}
|
|
|
}
|
|
|
if children != nil {
|
|
|
yes = true
|
|
|
}
|
|
|
return
|
|
|
}
|
