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