package entity import ( MC "app.yhyue.com/moapp/jybase/common" "app.yhyue.com/moapp/jybase/encrypt" "database/sql" "github.com/zeromicro/go-zero/core/logx" "sort" "strconv" "strings" "userCenter/rpc/pb" ) // WorkDesktopMenu 工作桌面菜单信息 type WorkDesktopMenu struct { MenuTree []*JYMenu UserId string AppId string TimeOut int BigMemberOff bool } // JYMenu 工作桌面菜单内容 type JYMenu struct { Id int Name string OrderId int ParentId int PowerIds string CheckCode int Url string Icon string AdditionalInfo Additional AppType string OpenType string Children []*JYMenu } type Additional struct { Title string `json:"title"` Content string `json:"content"` ConfirmUrl string `json:"confirmUrl"` ConfirmText string `json:"confirmText"` IsShowCancel bool `json:"isShowCancel"` } // WorkMenuTree 菜单-格式化-tree func (m *WorkDesktopMenu) WorkMenuTree(parentId int) (jyMenu []*JYMenu) { jyMenu = make([]*JYMenu, 0) for _, mv := range m.MenuTree { if mv.ParentId == parentId { mv.Children = m.WorkMenuTree(mv.Id) sort.SliceStable(mv.Children, func(i, j int) bool { return mv.Children[i].OrderId < mv.Children[j].OrderId }) jyMenu = append(jyMenu, mv) } } sort.SliceStable(jyMenu, func(i, j int) bool { return jyMenu[i].OrderId < jyMenu[j].OrderId }) //bytes, _ := json.MarshalIndent(jyMenu, "", " ") //fmt.Printf("%s\n", bytes) //log.Println("-------") return } var verification = map[string]string{ "商机管理": "100,101,102,110,111,112", "大会员": "1", "超级订阅": "200,201", } // WorkMenuFormat 菜单 格式化 func (m *WorkDesktopMenu) WorkMenuFormat() ([]*pb.MenuList, error) { logx.Info("树的长度:", len(m.MenuTree)) menuList := make([]*pb.MenuList, 0) for _, mv := range m.MenuTree { //0:默认展示;1:需验证权限,有权限展示;2:需验证权限,无权限也可展示(可用服务无权限不展示) usable := m.VerifyPermissions(mv.PowerIds) //checkCode==1 验证权限结果 无权限不显示 if !usable && mv.CheckCode == 1 { continue } oneLevel := &pb.MenuList{ Name: mv.Name, Icon: mv.Icon, Url: mv.Url, Id: encrypt.SE.EncodeString(strconv.Itoa(mv.Id)), //功能id加密 Usable: MC.If(mv.PowerIds != "", usable && mv.CheckCode == 1, true).(bool), AppType: mv.AppType, OpenType: mv.OpenType, Child: []*pb.SecondLevelMenu{}, TipInfo: &pb.TipInfo{ Title: mv.AdditionalInfo.Title, Content: mv.AdditionalInfo.Content, ConfirmUrl: mv.AdditionalInfo.ConfirmUrl, ConfirmText: mv.AdditionalInfo.ConfirmText, IsShowCancel: mv.AdditionalInfo.IsShowCancel, }, } if len(mv.Children) > 0 { for _, sv := range mv.Children { //0:默认展示;1:需验证权限,有权限展示;2:需验证权限,无权限也可展示(可用服务无权限不展示) usable := m.VerifyPermissions(mv.PowerIds) //checkCode==1 验证权限结果 无权限不显示 if !usable && mv.CheckCode == 1 { continue } secondLevel := &pb.SecondLevelMenu{ Name: sv.Name, Icon: sv.Icon, Url: sv.Url, Id: encrypt.SE.EncodeString(strconv.Itoa(sv.Id)), //功能id加密 Usable: MC.If(mv.PowerIds != "", usable && mv.CheckCode == 1, true).(bool), Child: []*pb.ThreeLevelMenu{}, AppType: sv.AppType, OpenType: sv.OpenType, TipInfo: &pb.TipInfo{ Title: sv.AdditionalInfo.Title, Content: sv.AdditionalInfo.Content, ConfirmUrl: sv.AdditionalInfo.ConfirmUrl, ConfirmText: sv.AdditionalInfo.ConfirmText, IsShowCancel: sv.AdditionalInfo.IsShowCancel, }, } if len(sv.Children) > 0 { for _, tv := range sv.Children { //0:默认展示;1:需验证权限,有权限展示;2:需验证权限,无权限也可展示(可用服务无权限不展示) usable := m.VerifyPermissions(mv.PowerIds) //checkCode==1 验证权限结果 无权限不显示 if !usable && mv.CheckCode == 1 { continue } threeLevel := &pb.ThreeLevelMenu{ Name: tv.Name, Icon: tv.Icon, Url: tv.Url, Id: encrypt.SE.EncodeString(strconv.Itoa(tv.Id)), //功能id加密 Usable: MC.If(mv.PowerIds != "", usable && mv.CheckCode == 1, true).(bool), AppType: tv.AppType, OpenType: tv.OpenType, TipInfo: &pb.TipInfo{ Title: tv.AdditionalInfo.Title, Content: tv.AdditionalInfo.Content, ConfirmUrl: tv.AdditionalInfo.ConfirmUrl, ConfirmText: tv.AdditionalInfo.ConfirmText, IsShowCancel: tv.AdditionalInfo.IsShowCancel, }, } if len(tv.Children) > 0 { for _, fv := range tv.Children { if fv.Name == "" { continue } threeLevel.Url = MC.If(fv.Url != "", fv.Url, tv.Url).(string) threeLevel.Icon = MC.If(fv.Icon != "", fv.Icon, tv.Icon).(string) if fv.AdditionalInfo.Title != "" { threeLevel.TipInfo = &pb.TipInfo{ Title: fv.AdditionalInfo.Title, Content: fv.AdditionalInfo.Content, ConfirmUrl: fv.AdditionalInfo.ConfirmUrl, ConfirmText: fv.AdditionalInfo.ConfirmText, IsShowCancel: fv.AdditionalInfo.IsShowCancel, } } //订阅是商机管理优先 其他功能是大会员优先(这样是有问题的 有些功能超级订阅用户可用 大会员用户权不可用) //解决上面的问题:菜单只保留三级菜单;四级菜单信息不入库;判断三级菜单权限 switch { case strings.Contains(tv.Name, "订阅"): switch { case strings.HasPrefix(fv.Name, "商机管理") && m.VerifyPermissions(verification["商机管理"]): break case strings.HasPrefix(fv.Name, "大会员") && m.VerifyPermissions(verification["大会员"]): break case strings.HasPrefix(fv.Name, "超级订阅") && m.VerifyPermissions(verification["超级订阅"]): break } default: switch { case strings.HasPrefix(fv.Name, "大会员") && m.VerifyPermissions(verification["大会员"]): break case strings.HasPrefix(fv.Name, "商机管理") && m.VerifyPermissions(verification["商机管理"]): break case strings.HasPrefix(fv.Name, "超级订阅") && m.VerifyPermissions(verification["超级订阅"]): break } } } } if threeLevel != nil { secondLevel.Child = append(secondLevel.Child, threeLevel) } } } oneLevel.Child = append(oneLevel.Child, secondLevel) } } menuList = append(menuList, oneLevel) } //bytes, _ := json.MarshalIndent(menuList, "", " ") //fmt.Printf("%s\n", bytes) return menuList, nil } // VerifyPermissions 是否有权限可用此服务 func (m *WorkDesktopMenu) VerifyPermissions(powerIds string) (b bool) { if powerIds == "" { return true } if len(strings.Split(powerIds, ",")) > 0 { userPower := AutoUserPowerInfo(m.UserId, m.AppId, m.TimeOut, m.BigMemberOff) func(powerIds string) { for _, pv := range strings.Split(powerIds, ",") { if userPower[pv] > 0 { b = true break } } }(powerIds) } return } //常用功能 const ( WorkCommonly = "work_commonly" WorkMenu = "work_menu" ) //更新常用功能 func CommonlyUpdate(in *pb.WorkDesktopComprehensiveReq) (B bool, M string) { //事务 1:查; 2: 存;3: 删; var ids []string for mk, mid := range strings.Split(in.MenuIds, ",") { //常用功能数量限制 if in.CommonlySize > 0 && mk >= int(in.CommonlySize) { break } //id 解密 ids = append(ids, encrypt.SE.DecodeString(mid)) } //更新此用户设置的常用功能 if B = BaseMysql.ExecTx("常用功能批量更新", func(tx *sql.Tx) bool { //查询此用户常用功能是否已存在记录 var id = 0 existingData := BaseMysql.SelectBySqlByTx(tx, `SELECT id FROM `+WorkCommonly+` WHERE userid = ? AND appid = ? AND field = ? AND platform = ? ORDER BY id DESC `, in.UserId, in.AppId, in.ActionMode, in.Platform) if existingData != nil && len(*existingData) > 0 { id = MC.IntAll((*existingData)[0]["id"]) } switch { case id > 0: //更新 if BaseMysql.UpdateOrDeleteBySqlByTx(tx, `UPDATE `+WorkCommonly+` SET value = ? WHERE id = ?`, strings.Join(ids, ","), id) < 0 { logx.Info("常用功能-更新数据失败") return false } default: //插入 if BaseMysql.InsertBySqlByTx(tx, `INSERT INTO `+WorkCommonly+` (appid,userid,platform,field,value) VALUES (?,?,?,?,?)`, in.AppId, in.UserId, in.Platform, in.ActionMode, strings.Join(ids, ",")) < 0 { logx.Info("常用功能-插入数据失败") return false } } return true }); !B { M = "常用功能更新数据失败" } return }