From 9b4433fddf5b401edb0aace8a404ac733b122702 Mon Sep 17 00:00:00 2001 From: 田源 <tianyuan@vci-tech.com> Date: 星期四, 03 四月 2025 14:35:02 +0800 Subject: [PATCH] 添加非密字段显示 --- Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/RoleMenuServiceImpl.java | 47 +++++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 47 insertions(+), 0 deletions(-) diff --git a/Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/RoleMenuServiceImpl.java b/Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/RoleMenuServiceImpl.java index a29ea7c..dc02fbd 100644 --- a/Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/RoleMenuServiceImpl.java +++ b/Source/UBCS/ubcs-service/ubcs-system/src/main/java/com/vci/ubcs/system/service/impl/RoleMenuServiceImpl.java @@ -16,11 +16,27 @@ */ package com.vci.ubcs.system.service.impl; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import com.baomidou.mybatisplus.core.toolkit.Wrappers; import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl; +import com.vci.ubcs.starter.util.MybatisParameterUtil; import com.vci.ubcs.system.entity.RoleMenu; import com.vci.ubcs.system.mapper.RoleMenuMapper; import com.vci.ubcs.system.service.IRoleMenuService; +import com.vci.ubcs.system.service.IRoleService; +import com.vci.ubcs.system.user.entity.User; +import com.vci.ubcs.system.user.feign.IUserClient; +import org.springblade.core.log.exception.ServiceException; +import org.springblade.core.secure.utils.AuthUtil; +import org.springblade.core.tool.api.R; +import org.springblade.core.tool.utils.Func; +import org.springframework.beans.factory.annotation.Autowired; import org.springframework.stereotype.Service; + +import javax.annotation.Resource; +import java.util.ArrayList; +import java.util.List; +import java.util.stream.Collectors; /** * 鏈嶅姟瀹炵幇绫� @@ -30,4 +46,35 @@ @Service public class RoleMenuServiceImpl extends ServiceImpl<RoleMenuMapper, RoleMenu> implements IRoleMenuService { + @Resource + private RoleMenuMapper roleMenuMapper; + + @Resource + private IUserClient userClient; + + /** + * 鏍规嵁鐢ㄦ埛id鏌ヨ瑙掕壊id锛屽啀鏌ヨ琚巿鏉冪殑鑿滃崟id + * @param userId + * @return + */ + @Override + public List<Long> getMenuIdByUserId(Long userId) { + // 褰撴煡璇㈡潯浠秈n澶т簬涓�鍗冩潯鏃� + List<String> roleIdList = Func.toStrList(AuthUtil.getUser().getRoleId()); + //鏋勫缓鏌ヨ鑿滃崟id鐨勬煡璇㈡潯浠� + LambdaQueryWrapper<RoleMenu> roleMenuWrapper = Wrappers.<RoleMenu>query() + .lambda().select(RoleMenu::getMenuId); + if(roleIdList.size()>=1000){ + MybatisParameterUtil.cutInParameter(roleMenuWrapper,RoleMenu::getRoleId,roleIdList); + }else { + roleMenuWrapper.in(RoleMenu::getRoleId,roleIdList); + } + List<RoleMenu> roleMenus = roleMenuMapper.selectList(roleMenuWrapper); + if(roleMenus.isEmpty()){ + return new ArrayList<>(); + } + return roleMenus.stream().map(RoleMenu::getMenuId).collect(Collectors.toList()); + } + + } -- Gitblit v1.9.3