package BaseRoleMenuService

import (
	"context"
	"dsBaseWeb/Business/BaseRoleMenu/BaseRoleMenuProto"
	"dsBaseWeb/Utils/GRpcUtil"
	"dsBaseWeb/Utils/LogUtil"
	"errors"
	"time"
)

/**
功能：获取角色菜单对应关系
作者：吴缤
时间：2021-01-04
*/
func GetBaseRoleMenuCorr(arg BaseRoleMenuProto.GetCorrArg) (*BaseRoleMenuProto.Reply, error) {
	//1、准备动作：连接服务器
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()
	conn := GRpcUtil.GetConnection()
	if conn == nil {
		return nil, errors.New("RPC服务未启动！")
	}
	//2、业务传参
	c := BaseRoleMenuProto.NewBaseRoleMenuManageClient(conn)
	Reply, err := c.GetBaseRoleMenuCorr(ctx, &arg)
	if err != nil {
		LogUtil.Error("获取角色菜单对应关系错误: ", err.Error())
	}
	return Reply, err
}

/**
功能：设置角色菜单对应关系
作者：吴缤
时间：2021-01-04
*/
func SetBaseRoleMenuCorr(arg BaseRoleMenuProto.SetCorrArg) (*BaseRoleMenuProto.Reply, error) {
	//1、准备动作：连接服务器
	ctx, cancel := context.WithTimeout(context.Background(), 10*time.Second)
	defer cancel()
	conn := GRpcUtil.GetConnection()
	if conn == nil {
		return nil, errors.New("RPC服务未启动！")
	}
	//2、业务传参
	c := BaseRoleMenuProto.NewBaseRoleMenuManageClient(conn)
	Reply, err := c.SetBaseRoleMenuCorr(ctx, &arg)
	if err != nil {
		LogUtil.Error("设置角色菜单对应关系错误: ", err.Error())
	}
	return Reply, err
}