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 |   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