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"
	"time"
	"userCenter/entity"
	"userCenter/rpc/pb"
	. "userCenter/rpc/usercenter"
)

//工作桌面--菜单当前选择模式--全部:all/可用:usable
//工作桌面--常用功能更新
func RenewWorkDesktopMenuModeOrCommonly(in *WorkDesktopComprehensiveReq) (r *WorkDesktopComprehensiveResp) {
	//
	r = &WorkDesktopComprehensiveResp{}
	switch in.ActionMode {
	case "commonly": //常用功能更新
		//in.MenuIds  为空；删除
		if in.MenuIds == "" {
			if !func() bool {
				if entity.BaseMysql.CountBySql(`SELECT COUNT(id)  FROM `+entity.WorkCommonly+` WHERE userid = '?' AND appid = ? AND platform = '?' AND field = ?`, in.UserId, in.AppId, in.Platform, in.ActionMode) > 0 {
					return entity.BaseMysql.UpdateOrDeleteBySql(`DELETE  FROM `+entity.WorkCommonly+` WHERE userid = '?' AND appid = ? AND platform = '?' AND field = ?`, in.UserId, in.AppId, in.Platform, in.ActionMode) > 0
				}
				return true
			}() {
				r.ErrorCode = -1
				r.ErrorMsg = "常用功能-清空常用功能异常"
			}
		} else {
			//in.MenuIds 不为空：更新
			if b, m := entity.CommonlyUpdate(in); !b {
				r.ErrorCode = -1
				r.ErrorMsg = m
			}
		}

	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 userid=? AND field = ? AND platform = ? LIMIT 1`, in.AppId, in.UserId, 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 = "菜单模式-更新异常"
					}
				}
			}
		}
	}
	return
}

//获取工作桌面菜单树
func GetWordDesktopMenuTree(in *WorkDesktopMenuInfoReq, redisOutTime int, bigMemberOff bool) ([]*pb.MenuList, error) {
	t1 := time.Now()
	//redis缓存
	var menuList []*pb.MenuList
	RedisMenuKey := fmt.Sprintf(entity.RedisMenuKey, in.AppId, 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.WorkCommonly+` 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,
		TimeOut:      redisOutTime,
		BigMemberOff: bigMemberOff,
	}
	for _, v := range *menuData {
		menu := &entity.JYMenu{
			Id:        MC.IntAll(v["id"]),
			Name:      MC.ObjToString(v["name"]),
			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 {
			if redis.PutBytes(entity.RedisCode, RedisMenuKey, &menuBytes, redisOutTime) != nil {
				logx.Info("工作桌面菜单 redis缓存异常")
			}
		} else {
			logx.Info("菜单数据序列化异常")
		}
	}
	logx.Info("整体耗时：", time.Since(t1))
	return menuList, err
}