From 1facfdc9361d888d72c959bac4d8fb05dcad02d7 Mon Sep 17 00:00:00 2001
From: xiaoyong931011 <15274802129@163.com>
Date: Thu, 07 Jul 2022 18:27:06 +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 |   83 ++++++++++++++++++++++++++++++-----------
 1 files changed, 61 insertions(+), 22 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 2341744..7541027 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
@@ -43,11 +49,11 @@
         UserEntity userlogin = LoginUserUtil.getLoginUser();
         Page<AdminUserVo> page = new Page<>(adminUserDto.getPageNum(), adminUserDto.getPageSize());
         UserEntity userEntity = new UserEntity();
-        long belongId = userlogin.getBelongId() == null ? UserEntity.USER_BELONG_TOP : userlogin.getBelongId();
-        if(belongId > UserEntity.USER_BELONG_TOP){
-            userEntity.setBelongId(belongId);
+        long companyId = userlogin.getCompanyId() == null ? UserEntity.USER_BELONG_TOP : userlogin.getCompanyId();
+        if(companyId > UserEntity.USER_BELONG_TOP){
+            userEntity.setCompanyId(companyId);
             userEntity.setRoleName(RoleEntity.ROLE_ADMIN_NAME);
-        }else if(belongId == UserEntity.USER_BELONG_TOP){
+        }else if(companyId == UserEntity.USER_BELONG_TOP){
             userEntity.setRoleName(RoleEntity.ROLE_SUPER_NAME);
         }
         if(StrUtil.isNotEmpty(adminUserDto.getUsername())){
@@ -82,24 +88,14 @@
     public Result addUser(AdminAddUserDto adminAddUserDto) {
         UserEntity userlogin = LoginUserUtil.getLoginUser();
         String username = adminAddUserDto.getUsername();
-        if(StrUtil.isEmpty(username)){
-            return Result.fail("请输入用户名");
-        }
         QueryWrapper<UserEntity> objectQueryWrapper = new QueryWrapper<>();
         objectQueryWrapper.eq("username",username);
-//        objectQueryWrapper.eq("belong_id",userlogin.getBelongId());
         List<UserEntity> userEntities = this.baseMapper.selectList(objectQueryWrapper);
         if(CollUtil.isNotEmpty(userEntities)){
             return Result.fail("用户名重复");
         }
         String roleIds = adminAddUserDto.getRoleIds();
-        if(StrUtil.isEmpty(roleIds)){
-            return Result.fail("请选择用户角色");
-        }
         String phone = adminAddUserDto.getPhone();
-        if(StrUtil.isEmpty(phone)){
-            return Result.fail("请输入联系电话");
-        }
         Long belongId = adminAddUserDto.getBelongId() == null ? UserEntity.USER_BELONG_TOP:adminAddUserDto.getBelongId();
         if(belongId == UserEntity.USER_BELONG_TOP){
             return Result.fail("请选择所属公司");
@@ -115,7 +111,7 @@
         }
         userEntity.setPassword(SecureUtil.md5(UserEntity.PASSWORD_DEFAULT));
         userEntity.setStatus(UserEntity.STATUS_ENABLE);
-        userEntity.setBelongId(belongId);
+        userEntity.setCompanyId(belongId);
         int insert = this.baseMapper.insert(userEntity);
         if(insert > 0){
             saveUserRole(roleIds,userEntity.getId());
@@ -175,7 +171,7 @@
         adminSeeUserInfoVo.setNickname(userEntity.getNickname());
         adminSeeUserInfoVo.setPhone(userEntity.getPhone());
         adminSeeUserInfoVo.setEmail(userEntity.getEmail());
-        adminSeeUserInfoVo.setBelongId(userEntity.getBelongId());
+        adminSeeUserInfoVo.setBelongId(userEntity.getCompanyId());
 
         List<AdminRoleVo> adminRoleVos = new ArrayList<>();
         QueryWrapper<UserRoleEntity> objectQueryWrapper = new QueryWrapper<>();
@@ -217,7 +213,7 @@
         if(ObjectUtil.isEmpty(userEntity)){
             return Result.fail("用户不存在");
         }
-        userEntity.setBelongId(belongId);
+        userEntity.setCompanyId(belongId);
         userEntity.setUsername(username);
         userEntity.setPhone(phone);
         if(StrUtil.isNotEmpty(adminUpdateUserDto.getNickname())){
@@ -261,11 +257,20 @@
         List<UserRoleEntity> userRoleEntities = userRoleMapper.selectList(userRoleQueryWrapper);
         if(CollUtil.isNotEmpty(userRoleEntities)){
             List menuIds = new ArrayList<Long>();
-            for(UserRoleEntity userRoleEntity : userRoleEntities){
-                List<RoleMenuEntity> roleMenuEntities = roleMenuMapper.selectListByRoleId(userRoleEntity.getRoleId());
-                if(CollUtil.isNotEmpty(roleMenuEntities)){
-                    for(RoleMenuEntity roleMenuEntity : roleMenuEntities){
-                        menuIds.add(roleMenuEntity.getMenuId());
+            if(UserEntity.USERNAME_DEFAULT.equals(loginUser.getUsername())){
+                List<MenuEntity> menuEntities = menuMapper.selectList(new QueryWrapper<>());
+                if(CollUtil.isNotEmpty(menuEntities)){
+                    for(MenuEntity menuEntity : menuEntities){
+                        menuIds.add(menuEntity.getId());
+                    }
+                }
+            }else{
+                for(UserRoleEntity userRoleEntity : userRoleEntities){
+                    List<RoleMenuEntity> roleMenuEntities = roleMenuMapper.selectListByRoleId(userRoleEntity.getRoleId());
+                    if(CollUtil.isNotEmpty(roleMenuEntities)){
+                        for(RoleMenuEntity roleMenuEntity : roleMenuEntities){
+                            menuIds.add(roleMenuEntity.getMenuId());
+                        }
                     }
                 }
             }
@@ -394,6 +399,40 @@
         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);
+        articleQueryWrapper.eq("del_status",ArticleEntity.DELETE_STATUS_NO);
+        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