xiaoyong931011
2022-07-07 bc43681f185af1edf833cf6c94833cb1cdd44a8e
src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/MenuServiceImpl.java
@@ -13,12 +13,15 @@
import com.xcong.farmer.cms.modules.system.dto.AdminUpdateMenuDto;
import com.xcong.farmer.cms.modules.system.entity.MenuEntity;
import com.xcong.farmer.cms.modules.system.entity.RoleMenuEntity;
import com.xcong.farmer.cms.modules.system.entity.UserEntity;
import com.xcong.farmer.cms.modules.system.entity.UserRoleEntity;
import com.xcong.farmer.cms.modules.system.mapper.MenuMapper;
import com.xcong.farmer.cms.modules.system.mapper.RoleMenuMapper;
import com.xcong.farmer.cms.modules.system.service.IMenuService;
import com.xcong.farmer.cms.modules.system.util.LoginUserUtil;
import com.xcong.farmer.cms.modules.system.vo.AdminMenuVo;
import com.xcong.farmer.cms.modules.system.vo.AdminSeeMenuInfoVo;
import com.xcong.farmer.cms.modules.system.vo.AdminUserMenuVo;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
@@ -41,8 +44,10 @@
    @Override
    public Result getMenuInPage(AdminMenuDto adminMenuDto) {
        Page<AdminMenuVo> page = new Page<>(adminMenuDto.getPageNum(), adminMenuDto.getPageSize());
        UserEntity userlogin = LoginUserUtil.getLoginUser();
        MenuEntity menuEntity = new MenuEntity();
        menuEntity.setParentId(MenuEntity.PARENTID_DEFAULT);
//        menuEntity.setBelongId(userlogin.getBelongId()==null?UserEntity.USER_BELONG_TOP:userlogin.getBelongId());
        IPage<AdminMenuVo> list = this.baseMapper.selectAdminMenuVoInPage(page,menuEntity);
        List<AdminMenuVo> records = list.getRecords();
        if(CollUtil.isNotEmpty(records)){
@@ -73,14 +78,9 @@
    @Override
    @Transactional
    public Result addMenu(AdminAddMenuDto adminAddMenuDto) {
        UserEntity userlogin = LoginUserUtil.getLoginUser();
        String menuName = adminAddMenuDto.getMenuName();
        if(StrUtil.isEmpty(menuName)){
            return Result.fail("请输入菜单名称");
        }
        Integer menuType = adminAddMenuDto.getMenuType();
        if(ObjectUtil.isEmpty(menuType)){
            return Result.fail("请输入菜单类型");
        }
        MenuEntity menuEntity = new MenuEntity();
        menuEntity.setMenuName(menuName);
        menuEntity.setMenuType(menuType);
@@ -145,14 +145,8 @@
            return Result.fail("菜单不存在");
        }
        String menuName = adminUpdateMenuDto.getMenuName();
        if(StrUtil.isEmpty(menuName)){
            return Result.fail("请输入菜单名称");
        }
        menuEntity.setMenuName(menuName);
        Integer menuType = adminUpdateMenuDto.getMenuType();
        if(ObjectUtil.isEmpty(menuType)){
            return Result.fail("请输入菜单类型");
        }
        menuEntity.setMenuType(menuType);
        String routeUrl = adminUpdateMenuDto.getRouteUrl();
        if(StrUtil.isNotEmpty(routeUrl)){
@@ -194,6 +188,7 @@
    @Override
    public Result getMenuInList() {
        UserEntity userlogin = LoginUserUtil.getLoginUser();
        List<AdminMenuVo> records = this.baseMapper.selectAdminMenuVoInListByParentId(MenuEntity.PARENTID_DEFAULT);
        if(CollUtil.isNotEmpty(records)){
            for(AdminMenuVo adminMenuVo : records){
@@ -217,6 +212,21 @@
                adminMenuVo.setChild(adminMenuVoChilds);
            }
        }
        if(CollUtil.isNotEmpty(records)){
            for(AdminMenuVo adminMenuVo : records){
                List<AdminMenuVo> child = adminMenuVo.getChild();
                if(CollUtil.isNotEmpty(child)){
                    List<Long> menuIds = new ArrayList<>();
                    for(AdminMenuVo adminMenuVoChild : child){
                        Long id = adminMenuVoChild.getId();
                        menuIds.add(id);
                    }
                    List<AdminMenuVo> adminMenuVoByOrderNum = this.baseMapper.selectAdminMenuVoOrderByOrderNum(menuIds);
                    adminMenuVo.setChild(adminMenuVoByOrderNum);
                }
            }
        }
        return Result.ok(records);
    }
}