From 679b072c926361039c5b9f47cf99f015a1eefd01 Mon Sep 17 00:00:00 2001 From: Helius <wangdoubleone@gmail.com> Date: Thu, 07 Jul 2022 14:18:54 +0800 Subject: [PATCH] Merge branch 'master' of http://120.27.238.55:7000/r/farmer-cms --- src/main/java/com/xcong/farmer/cms/modules/system/service/Impl/UserServiceImpl.java | 39 +++++++++++++++++++++++++++++++++++++++ 1 files changed, 39 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 54e461c..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 @@ -36,6 +36,12 @@ @Resource private RoleMenuMapper roleMenuMapper; @Resource + private ColumnMapper columnMapper; + @Resource + private ArticleMapper articleMapper; + @Resource + private MessageBoardMapper messageBoardMapper; + @Resource private MenuMapper menuMapper; @Override @@ -393,6 +399,39 @@ return Result.ok(adminUserInfoVo); } + @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)); + + QueryWrapper<ArticleEntity> articleQueryWrapper = new QueryWrapper<>(); + articleQueryWrapper.eq("company_id",companyId); + adminHomePageInfoVo.setArticleNum(articleMapper.selectCount(articleQueryWrapper)); + + QueryWrapper<MessageBoardEntity> messageQueryWrapper = new QueryWrapper<>(); + messageQueryWrapper.eq("company_id",companyId); + adminHomePageInfoVo.setMessageNum(messageBoardMapper.selectCount(messageQueryWrapper)); + return Result.ok(adminHomePageInfoVo); + } + private void saveUserRole(String roleIds,Long userId) { if (StrUtil.isNotEmpty(roleIds)) { String[] roleIdList = roleIds.split(StringPool.COMMA); -- Gitblit v1.9.1