userCenter/service/workDesktop.go

package service

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

// RenewWorkDesktopMenuModeOrCommonly 工作桌面--菜单当前选择模式--全部:all/可用:usable
//工作桌面--常用功能更新
//工作桌面--常用功能列表
func RenewWorkDesktopMenuModeOrCommonly(in *WorkDesktopComprehensiveReq) (r *WorkDesktopComprehensiveResp) {
	r = &WorkDesktopComprehensiveResp{}
	switch in.ActionMode {
	case "commonlyRenew": //常用功能更新
		//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)
		var (
			pIds []string
		)
		if existingData != nil && len(*existingData) > 0 {
			eData := (*existingData)[0]
			if MC.ObjToString(eData["value"]) != "" {
				for _, pv := range strings.Split(MC.ObjToString(eData["value"]), ",") {
					//params = append(params, "?")
					if pv == "" {
						continue
					}
					pIds = append(pIds, pv)
				}
			}
		} else {
			if in.CommonlyIds != "" && len(strings.Split(in.CommonlyIds, ",")) > 0 {
				var ids []string
				for _, v := range strings.Split(in.CommonlyIds, ",") {
					ids = append(ids, encrypt.SE.EncodeString(v))
				}
				in.MenuIds = strings.Join(ids, ",")
				//初始化 常用功能
				if b, _ := entity.CommonlyUpdate(in); b {
					pIds = strings.Split(in.CommonlyIds, ",")
				} else {
					logx.Info("初始化常用功能异常")
				}
			}
		}
		if len(pIds) > 0 {
			//常用功能存储是三级菜单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:  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"]),
						Status:    MC.IntAll(mv["status"]),
					}
					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)
					}
				}
				//常用功能格式化
				menu, saveIds, delBool := m.CommonlyFormat(childMenus)
				if delBool && len(saveIds) > 0 {
					in.MenuIds = strings.Join(saveIds, ",")
					go func(in *WorkDesktopComprehensiveReq) {
						if b, _ := entity.CommonlyUpdate(in); !b {
							logx.Info("常用功能-更新数据失败")
						}
					}(in)
				}
				r.Data = menu
			}
		}
	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
}

// GetWorkDesktopMenuMode 获取工作菜单模式
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
}

// GetWordDesktopMenuTree 获取工作桌面菜单树
func GetWordDesktopMenuTree(in *WorkDesktopMenuInfoReq) ([]*pb.MenuList, error) {
	if in.EntId > 0 {
		//判断商机管理用户是否切换企业
		userEntIdKey := fmt.Sprintf(entity.UserEntIdKey, in.AppId, time.Now().Day(), in.UserId)
		redisEntid := redis.GetInt(entity.RedisCode, userEntIdKey)
		logx.Info(redisEntid, "----EntId:", in.EntId)
		if int64(redisEntid) > 0 && int64(redisEntid) != in.EntId {
			//商机管理用户切换企业---清除用户权限缓存&&清除用户菜单缓存
			entity.ClearUserPowerFunc(in.UserId, in.AppId)
			logx.Info("+++++++++++++++++++++++++++++++++++++")
		}
	}
	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数据
	query := `SELECT * FROM ` + entity.WorkMenu + ` WHERE  appid=? AND status=0 ORDER BY id ASC`
	//是否开启内网访问数据权限
	if in.IntranetBool {
		query = `SELECT * FROM ` + entity.WorkMenu + ` WHERE  appid=? AND status>=0 ORDER BY id ASC`
	}
	menuData := entity.BaseMysql.SelectBySql(query, 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.InternalTime) + rand.Intn(60), //用户权限 和 企业id
		BigMemberOff: in.BigMemberOff,
		EntId:        in.EntId,
		EntUserId:    in.EntUserId,
	}
	logx.Info("-----TimeOut:", wdm.TimeOut)
	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"]),
			Status:         MC.IntAll(v["status"]),
			PermissionCode: MC.ObjToString(v["permissioncode"]),
			CapitalCode:    MC.ObjToString(v["capitalcode"]),
		}
		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 capitalinfo := MC.ObjToString(v["capitalinfo"]); capitalinfo != "" {
			additional := entity.Additional{}
			if json.Unmarshal([]byte(capitalinfo), &additional) == nil {
				menu.CapitalInfo = 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)
			logx.Info("-----redisOutTime:", redisOutTime)
			if redis.PutBytes(entity.RedisCode, RedisMenuKey, &menuBytes, redisOutTime) != nil {
				logx.Info("工作桌面菜单 redis缓存异常")
			}
		} else {
			logx.Info("菜单数据序列化异常")
		}
	}
	logx.Info("整体耗时：", time.Since(t1))
	return menuList, err
}