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