userCenter/rpc/internal/logic/workdesktopmenuinfologic.go

package logic

import (
	"bp.jydev.jianyu360.cn/BaseService/userCenter/service"
	"context"

	"bp.jydev.jianyu360.cn/BaseService/userCenter/rpc/internal/svc"
	"bp.jydev.jianyu360.cn/BaseService/userCenter/rpc/pb"

	"github.com/zeromicro/go-zero/core/logx"
)

type WorkDesktopMenuInfoLogic struct {
	ctx    context.Context
	svcCtx *svc.ServiceContext
	logx.Logger
}

func NewWorkDesktopMenuInfoLogic(ctx context.Context, svcCtx *svc.ServiceContext) *WorkDesktopMenuInfoLogic {
	return &WorkDesktopMenuInfoLogic{
		ctx:    ctx,
		svcCtx: svcCtx,
		Logger: logx.WithContext(ctx),
	}
}

// 获取菜单树
func (l *WorkDesktopMenuInfoLogic) WorkDesktopMenuInfo(in *pb.WorkDesktopMenuInfoReq) (*pb.WorkDesktopMenuInfoResp, error) {
	if in.UserId == "" {
		return &pb.WorkDesktopMenuInfoResp{
			ErrorCode: -1,
			ErrorMsg:  "用户信息有误",
			Data:      nil,
		}, nil
	}
	var (
		errorCode int64 = 0
		errorMsg        = ""
		menuMode        = "all"
	)
	//菜单树
	menuList, err := service.GetWordDesktopMenuTree(in)
	//菜单模式
	menuMode, err = service.GetWorkDesktopMenuMode(in)
	if err != nil {
		errorCode = -1
		errorMsg = err.Error()
	}
	return &pb.WorkDesktopMenuInfoResp{
		ErrorCode: errorCode,
		ErrorMsg:  errorMsg,
		Data: &pb.MenuData{
			MenuType: menuMode,
			MenuList: menuList,
		},
	}, nil
}