userCenter/service/workDesktop.go

package service

import (
	MC "app.yhyue.com/moapp/jybase/common"
	"app.yhyue.com/moapp/jybase/redis"
	"encoding/json"
	"errors"
	"fmt"
	"github.com/zeromicro/go-zero/core/logx"
	"math/rand"
	"strings"
	"time"
	"userCenter/entity"
	"userCenter/rpc/pb"
	. "userCenter/rpc/usercenter"
)

//工作桌面--菜单当前选择模式--全部:all/可用:usable
//工作桌面--常用功能更新
//工作桌面--常用功能列表
func RenewWorkDesktopMenuModeOrCommonly(in *WorkDesktopComprehensiveReq) (r *WorkDesktopComprehensiveResp) {
	r = &WorkDesktopComprehensiveResp{}
	switch in.ActionMode {
	case "commonlyRenew": //常用功能更新
		//in.MenuIds  为空；删除
		if in.MenuIds == "" {
			count := entity.BaseMysql.CountBySql(`SELECT COUNT(id)  FROM `+entity.WorkCommonly+` WHERE base_userid = ? AND appid = ? AND platform = ? AND field = ?`, in.NewUserId, in.AppId, in.Platform, in.ActionMode)
			if count > 0 {
				if count := entity.BaseMysql.UpdateOrDeleteBySql(`DELETE  FROM `+entity.WorkCommonly+` WHERE base_userid = ? AND appid = ? AND platform = ? AND field = ?`, in.NewUserId, in.AppId, in.Platform, in.ActionMode); count <= 0 {
					r.ErrorCode = -1
					r.ErrorMsg = "常用功能-清空常用功能异常"
				}
			}
		} else {
			//in.MenuIds 不为空：更新
			if b, m := entity.CommonlyUpdate(in); !b {
				r.ErrorCode = -1
				r.ErrorMsg = m
			}
		}
	case "commonlyList":
		//查询常用功能列表
		in.ActionMode = "commonlyRenew"
		existingData := entity.BaseMysql.SelectBySql(`SELECT value  FROM `+entity.WorkCommonly+` WHERE base_userid = ? AND appid = ? AND field = ? AND platform = ?  ORDER BY id DESC `, in.NewUserId, in.AppId, in.ActionMode, in.Platform)
		if existingData != nil && len(*existingData) > 0 {
			eData := (*existingData)[0]
			if MC.ObjToString(eData["value"]) != "" {
				var (
					//params []string
					pIds []string
				)
				for _, pv := range strings.Split(MC.ObjToString(eData["value"]), ",") {
					//params = append(params, "?")
					pIds = append(pIds, pv)
				}
				//常用功能存储是三级菜单id，需要四级菜单信息也查出来
				//会出现 用户设置常用功能，此功能下线，用户设置依然保存的有此功能  查询不能加 AND status = 0
				menuData := entity.BaseMysql.SelectBySql(fmt.Sprintf(`SELECT * from %s WHERE  ( id IN (%s)  OR parentid IN (%s))`, entity.WorkMenu, strings.Join(pIds, ","), strings.Join(pIds, ",")))
				if menuData != nil && len(*menuData) > 0 {
					var (
						m = &entity.WorkDesktopMenu{
							MenuTree:     []*entity.JYMenu{},
							UserId:       in.UserId,
							NewUserId:    in.NewUserId,
							AppId:        in.AppId,
							TimeOut:      int(in.RedisOutTime),
							BigMemberOff: in.BigMemberOff,
						}
						childMenus = map[int][]*entity.JYMenu{}
					)
					//是否存在父级
					parentIsExists := map[int]bool{}
					for _, mv := range *menuData {
						//id
						id := MC.IntAll(mv["id"])
						//parentid
						parentid := MC.IntAll(mv["parentid"])
						parentIsExists[id] = true
						menu := &entity.JYMenu{
							Id:        MC.IntAll(mv["id"]),
							Name:      MC.ObjToString(mv["name"]),
							Match:     MC.ObjToString(mv["match"]),
							OrderId:   MC.IntAll(mv["orderid"]),
							ParentId:  MC.IntAll(mv["parentid"]),
							PowerIds:  MC.ObjToString(mv["powerids"]),
							CheckCode: MC.IntAll(mv["checkcode"]),
							Icon:      MC.ObjToString(mv["icon"]),
							AppType:   MC.ObjToString(mv["apptype"]),
							OpenType:  MC.ObjToString(mv["opentype"]),
						}
						switch in.Platform {
						case "WX":
							menu.Url = MC.ObjToString(mv["wxurl"])
						case "APP":
							menu.Url = MC.ObjToString(mv["appurl"])
						default:
							menu.Url = MC.ObjToString(mv["pcurl"])
						}
						if additionalInfo := MC.ObjToString(mv["additionalinfo"]); additionalInfo != "" {
							additional := entity.Additional{}
							if json.Unmarshal([]byte(additionalInfo), &additional) == nil {
								menu.AdditionalInfo = additional
							}
						}
						if parentIsExists[parentid] {
							childMenus[parentid] = append(childMenus[parentid], menu)
						} else {
							m.MenuTree = append(m.MenuTree, menu)
						}
					}
					//常用功能格式化
					r.Data, _ = m.CommonlyFormat(childMenus)
				}
			}

		}
	case "menuMode": //菜单模式更新
		if in.MenuMode == "" || (in.MenuMode != "usable" && in.MenuMode != "all") {
			r.ErrorCode = -1
			r.ErrorMsg = "菜单模式-参数异常"
		} else {
			menuModes := entity.BaseMysql.SelectBySql(`SELECT id,value FROM `+entity.WorkCommonly+` WHERE appid=? AND base_userid =? AND field = ? AND platform = ? LIMIT 1`, in.AppId, in.NewUserId, in.ActionMode, in.Platform)
			if menuModes != nil && len(*menuModes) > 0 {
				menuMode := (*menuModes)[0]
				if MC.ObjToString(menuMode["value"]) != in.MenuMode {
					if entity.BaseMysql.UpdateOrDeleteBySql(`UPDATE `+entity.WorkCommonly+` SET value = ? WHERE id = ?`, in.MenuMode, MC.IntAll(menuMode["id"])) < 0 {
						r.ErrorCode = -1
						r.ErrorMsg = "菜单模式-更新异常"
					}
				}
			} else {
				if entity.BaseMysql.Insert(entity.WorkCommonly, map[string]interface{}{
					"appid":       in.AppId,
					"base_userid": in.NewUserId,
					"field":       in.ActionMode,
					"platform":    in.Platform,
					"value":       in.MenuMode,
				}) < 0 {
					r.ErrorCode = -1
					r.ErrorMsg = "菜单模式-插入异常"
				}
			}
		}
	}
	return
}

//
func GetWorkDesktopMenuMode(in *WorkDesktopMenuInfoReq) (str string, err error) {
	if in.UserId == "" {
		return "", fmt.Errorf("参数异常")
	}
	str = "all"
	menuModes := entity.BaseMysql.SelectBySql(`SELECT value FROM `+entity.WorkCommonly+` WHERE appid=? AND base_userid=? AND field = 'menuMode' AND platform = ? LIMIT 1`, in.AppId, in.NewUserId, in.Platform)
	if menuModes != nil && len(*menuModes) > 0 {
		menuMode := (*menuModes)[0]
		if MC.ObjToString(menuMode["value"]) != "" {
			str = MC.ObjToString(menuMode["value"])
		}
	}
	return
}

//获取工作桌面菜单树
func GetWordDesktopMenuTree(in *WorkDesktopMenuInfoReq) ([]*pb.MenuList, error) {
	t1 := time.Now()
	//redis缓存
	var menuList []*pb.MenuList
	RedisMenuKey := fmt.Sprintf(entity.RedisMenuKey, in.AppId, in.Platform, in.UserId)
	menuBytes, err := redis.GetBytes(entity.RedisCode, RedisMenuKey)
	if err == nil && len(*menuBytes) > 0 {
		if json.Unmarshal(*menuBytes, &menuList) == nil {
			return menuList, nil
		}
	}
	//tidb数据
	menuData := entity.BaseMysql.SelectBySql(`SELECT * FROM `+entity.WorkMenu+` WHERE status=0 AND appid=? ORDER BY id ASC`, in.AppId)
	if menuData == nil || len(*menuData) == 0 {
		return nil, errors.New("查询数据异常")
	}
	wdm := &entity.WorkDesktopMenu{
		MenuTree:     []*entity.JYMenu{},
		UserId:       in.UserId,
		NewUserId:    in.NewUserId,
		AppId:        in.AppId,
		TimeOut:      int(in.RedisOutTime),
		BigMemberOff: in.BigMemberOff,
	}
	for _, v := range *menuData {
		menu := &entity.JYMenu{
			Id:        MC.IntAll(v["id"]),
			Name:      MC.ObjToString(v["name"]),
			Match:     MC.ObjToString(v["match"]),
			OrderId:   MC.IntAll(v["orderid"]),
			ParentId:  MC.IntAll(v["parentid"]),
			PowerIds:  MC.ObjToString(v["powerids"]),
			CheckCode: MC.IntAll(v["checkcode"]),
			Icon:      MC.ObjToString(v["icon"]),
			AppType:   MC.ObjToString(v["apptype"]),
			OpenType:  MC.ObjToString(v["opentype"]),
		}
		switch in.Platform {
		case "WX":
			menu.Url = MC.ObjToString(v["wxurl"])
		case "APP":
			menu.Url = MC.ObjToString(v["appurl"])
		default:
			menu.Url = MC.ObjToString(v["pcurl"])
		}
		if additionalInfo := MC.ObjToString(v["additionalinfo"]); additionalInfo != "" {
			additional := entity.Additional{}
			if json.Unmarshal([]byte(additionalInfo), &additional) == nil {
				menu.AdditionalInfo = additional
			}
		}
		if menu.Id > 0 {
			wdm.MenuTree = append(wdm.MenuTree, menu)
		}
	}
	logx.Info("数据长度:", len(wdm.MenuTree), "-- 查询菜单数据耗时:", time.Since(t1))
	//jyMenu := wdm.WorkMenuTree(0)
	wdm.MenuTree = wdm.WorkMenuTree(0)
	logx.Info("菜单树生成耗时：", time.Since(t1))
	menuList, err = wdm.WorkMenuFormat()
	logx.Info("菜单格式化耗时：", time.Since(t1))
	if err == nil && len(menuList) > 0 {
		if menuBytes, err := json.Marshal(menuList); err == nil {
			redisOutTime := int(in.RedisOutTime) + rand.Intn(60*60)
			if redis.PutBytes(entity.RedisCode, RedisMenuKey, &menuBytes, redisOutTime) != nil {
				logx.Info("工作桌面菜单 redis缓存异常")
			}
		} else {
			logx.Info("菜单数据序列化异常")
		}
	}
	logx.Info("整体耗时：", time.Since(t1))
	return menuList, err
}