rakor
/
wms
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

优化了获取角色权限的方法

This commit is contained in:
zhangke 2025-09-03 12:33:36 +08:00
parent 9eb7ffbe02
commit ecfe26d708
8 changed files with 107 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
nflg-wms-admin/src/main/java/com/nflg/wms/admin/controller/CeneterOutboundControlelr.java
View File

@ -58,6 +58,32 @@ public class CeneterOutboundControlelr {
@Resource
private IWmsModelService wmsModelService;
@Resource
private IRoleButtonMapService roleButtonMapService ;
@Resource
private IMenuButtonService menuButtonService;
@GetMapping("idGenery")
public ApiResult<Long> idGenery() {
return ApiResult.success(IdUtil.getSnowflakeNextId());
}
@GetMapping("initrole")
public ApiResult<Void> initRole() {
List<RoleButtonMap> roleButtonMaps = new ArrayList<>();
List<MenuButton> menuButtons = menuButtonService.list();
for (MenuButton menuButton : menuButtons) {
RoleButtonMap roleButtonMap = new RoleButtonMap();
roleButtonMap.setRoleId(1L);
roleButtonMap.setId(IdUtil.getSnowflakeNextId());
roleButtonMap.setButtonId(menuButton.getId());
roleButtonMaps.add(roleButtonMap);
}
roleButtonMapService.saveBatch(roleButtonMaps);
return ApiResult.success();
}
@GetMapping("syncData")
public ApiResult<Void> syncData(@RequestParam String orderId) {

3
nflg-wms-admin/src/main/java/com/nflg/wms/admin/service/MenuControllerService.java
View File

@ -37,7 +37,8 @@ public class MenuControllerService {
private IMenuButtonApiMapService menuButtonApiMapService;
public List<MenuAuthorizeVO> getMenuForAuthorize(Long roleId) {
return menuService.getMenuForAuthorize(roleId);
return menuService.getNodeForAuthorize(roleId);
// return menuService.getMenuForAuthorize(roleId);
}
public IPage<MenuVO> searchMenu(@Valid MenuSearchQO request) {

4
nflg-wms-common/src/main/java/com/nflg/wms/common/pojo/qo/UserAddQO.java
View File

@ -39,7 +39,7 @@ public class UserAddQO{
/**
* 部门id
*/
@NotBlank
@NotNull
private Long deptId;
/**
@ -62,12 +62,10 @@ public class UserAddQO{
/**
* 初始化密码后是否强制要求登录后重设密码
*/
@NotNull
private Boolean enableMustResetPwd;
/**
* 默认语言
*/
@NotBlank
private String languageCode;
}

4
nflg-wms-common/src/main/java/com/nflg/wms/common/pojo/vo/MenuAuthorizeVO.java
View File

@ -11,6 +11,8 @@ public class MenuAuthorizeVO {
private Long id;
private Long parentId;
private String key;
//名称
@ -22,6 +24,8 @@ public class MenuAuthorizeVO {
//是否已授权
private Boolean selected;
//排序
private Short sort;
//下级
private List<MenuAuthorizeVO> children;
}

3
nflg-wms-repository/src/main/java/com/nflg/wms/repository/mapper/MenuMapper.java
View File

@ -2,6 +2,7 @@ package com.nflg.wms.repository.mapper;
import com.baomidou.mybatisplus.core.mapper.BaseMapper;
import com.nflg.wms.common.pojo.vo.ButtonVO;
import com.nflg.wms.common.pojo.vo.MenuAuthorizeVO;
import com.nflg.wms.repository.entity.Menu;
import java.util.List;
@ -19,4 +20,6 @@ public interface MenuMapper extends BaseMapper<Menu> {
List<ButtonVO> getButtonsByMenuId(Long userId, Long menuId);
Boolean menuIsSelected(Long roleId, Long menuId);
List<MenuAuthorizeVO> getAllDataForAuthorize(Long roleId);
}

2
nflg-wms-repository/src/main/java/com/nflg/wms/repository/service/IMenuService.java
View File

@ -38,4 +38,6 @@ public interface IMenuService extends IService<Menu> {
void enableMenu(@Valid EnableMenuQO request);
List<ButtonVO> getButtonsByMenuId(Long userId, @Valid @NotNull Long menuId);
List<MenuAuthorizeVO> getNodeForAuthorize(Long roleId);
}

34
nflg-wms-repository/src/main/java/com/nflg/wms/repository/service/impl/MenuServiceImpl.java
View File

@ -60,6 +60,40 @@ public class MenuServiceImpl extends ServiceImpl<MenuMapper, Menu> implements IM
return vos;
}
@Override
public List<MenuAuthorizeVO> getNodeForAuthorize(Long roleId) {
List<MenuAuthorizeVO> nodes = baseMapper.getAllDataForAuthorize(roleId);
List<MenuAuthorizeVO> rootNodes = nodes.stream().filter(v ->
v.getType() == 1 && v.getParentId() == 0
).toList();
for (MenuAuthorizeVO node : rootNodes) {
List<MenuAuthorizeVO> childNodes = getChildNode(node.getId(), nodes);
if (CollectionUtil.isNotEmpty(childNodes)) {
node.setChildren(new ArrayList<>());
node.getChildren().addAll(childNodes);
}
}
// rootNodes.sort(Comparator.comparing(MenuAuthorizeVO::getSort));
return rootNodes;
}
private List<MenuAuthorizeVO> getChildNode(Long parentId, List<MenuAuthorizeVO> nodes) {
List<MenuAuthorizeVO> rootNodes = nodes.stream().filter(v -> v.getParentId().equals(parentId)
).toList();
if (CollectionUtil.isNotEmpty(rootNodes)) {
for (MenuAuthorizeVO node : rootNodes) {
List<MenuAuthorizeVO> childNodes = getChildNode(node.getId(), nodes);
if (CollectionUtil.isNotEmpty(childNodes)) {
node.setChildren(new ArrayList<>());
node.getChildren().addAll(childNodes);
}
}
}
// rootNodes.sort(Comparator.comparing(MenuAuthorizeVO::getSort));
return rootNodes;
}
@Override
public IPage<MenuVO> search(MenuSearchQO request) {
if (StrUtil.isBlank(request.getName())) {

25
nflg-wms-repository/src/main/resources/mapper/MenuMapper.xml
View File

@ -19,4 +19,29 @@
WHERE menu_id = #{menuId}
AND role_id = #{roleId})
</select>
<select id="getAllDataForAuthorize" resultType="com.nflg.wms.common.pojo.vo.MenuAuthorizeVO">
select id,
parent_id,
"name",
CONCAT('menu-', id) as key,
sort,
1 as type,
case when b.menu_id is null then false else true end as selected
from menu a
left join (select menu_id from role_menu_map where role_id = #{roleId}) b on a.id = b.menu_id
where "enable" = true
UNION
select id,
menu_id,
"name",
CONCAT('button-', id) as key,
sort,
2 as type,
case when b.button_id is null then false else true end as selected
from menu_button a
left join (select button_id from role_button_map where role_id = #{roleId}) b on a.id = b.button_id
where "enable" = true
order by sort;
</select>
</mapper>