From f17a3980d0dfec11c1db69503e930fa20b6684f8 Mon Sep 17 00:00:00 2001 From: xiaoyong931011 <15274802129@163.com> Date: Thu, 07 Jul 2022 11:17:39 +0800 Subject: [PATCH] 20220606 --- src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/UserServiceImpl.java | 14 ++++++++++++++ 1 files changed, 14 insertions(+), 0 deletions(-) diff --git a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/UserServiceImpl.java b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/UserServiceImpl.java index 55bdc41..ffe82d9 100644 --- a/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/UserServiceImpl.java +++ b/src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/UserServiceImpl.java @@ -401,9 +401,23 @@ @Override public Result viewHomePage() { + UserEntity loginUser = LoginUserUtil.getLoginUser(); Long companyId = LoginUserUtil.getCompanyId(); + AdminHomePageInfoVo adminHomePageInfoVo = new AdminHomePageInfoVo(); + QueryWrapper<UserRoleEntity> userRoleQueryWrapper = new QueryWrapper<>(); + userRoleQueryWrapper.eq("user_id",loginUser.getId()); + List<UserRoleEntity> userRoleEntities = userRoleMapper.selectList(userRoleQueryWrapper); + if(CollUtil.isNotEmpty(userRoleEntities)){ + List<String> roleNames = new ArrayList<>(); + for(UserRoleEntity userRoleEntity : userRoleEntities){ + RoleEntity roleEntity = roleMapper.selectById(userRoleEntity.getRoleId()); + roleNames.add(roleEntity.getRoleName()); + } + adminHomePageInfoVo.setRoleNames(roleNames); + } + QueryWrapper<ColumnEntity> columnQueryWrapper = new QueryWrapper<>(); columnQueryWrapper.eq("company_id",companyId); adminHomePageInfoVo.setColumnNum(columnMapper.selectCount(columnQueryWrapper)); -- Gitblit v1.9.1