tangzy
2021-08-11 c0bf52690604d78dcbff7bf465e088133f1d2eea
src/main/java/org/springblade/modules/system/service/impl/UserServiceImpl.java
@@ -43,6 +43,7 @@
import org.springblade.modules.system.entity.*;
import org.springblade.modules.system.excel.UserExcel;
import org.springblade.modules.system.mapper.UserMapper;
import org.springblade.modules.system.node.TreeNode;
import org.springblade.modules.system.service.IRoleService;
import org.springblade.modules.system.service.IUserDeptService;
import org.springblade.modules.system.service.IUserOauthService;
@@ -52,10 +53,7 @@
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.*;
import static org.springblade.common.constant.CommonConstant.DEFAULT_PARAM_PASSWORD;
@@ -140,7 +138,7 @@
   }
   @Override
   public IPage<User> selectUserPage(IPage<User> page, User user, Long deptId, String tenantId) {
   public IPage<UserVO> selectUserPage(IPage<UserVO> page, User user, Long deptId, String tenantId) {
      List<Long> deptIdList = SysCache.getDeptChildIds(deptId);
      return page.setRecords(baseMapper.selectUserPage(page, user, deptIdList, tenantId));
   }
@@ -300,46 +298,29 @@
   public void importUser(List<UserExcel> data, Boolean isCovered) {
      data.forEach(userExcel -> {
         User user = Objects.requireNonNull(BeanUtil.copy(userExcel, User.class));
         // 设置用户平台
         user.setUserType(Func.toInt(DictCache.getKey(DictEnum.USER_TYPE, userExcel.getUserTypeName()), 1));
         // 设置部门ID
         user.setDeptId(SysCache.getDeptIds(userExcel.getTenantId(), userExcel.getDeptName()));
         // 设置岗位ID
         user.setPostId(SysCache.getPostIds(userExcel.getTenantId(), userExcel.getPostName()));
         // 设置角色ID
         user.setRoleId(SysCache.getRoleIds(userExcel.getTenantId(), userExcel.getRoleName()));
         // 设置租户ID
         if (!AuthUtil.isAdministrator() || StringUtil.isBlank(user.getTenantId())) {
            user.setTenantId(AuthUtil.getTenantId());
         }
         // 覆盖数据
         if (isCovered) {
            // 查询用户是否存在
            User oldUser = UserCache.getUser(userExcel.getTenantId(), userExcel.getAccount());
            if (oldUser != null && oldUser.getId() != null) {
               user.setId(oldUser.getId());
               this.updateUser(user);
               return;
            }
         }
         // 获取默认密码配置
         user.setRoleId("1412226235153731586");
         //获取默认密码配置
         String initPassword = ParamCache.getValue(DEFAULT_PARAM_PASSWORD);
         user.setPassword(initPassword);
         this.submit(user);
      });
   }
   @Override
   public List<UserExcel> exportUser(Wrapper<User> queryWrapper) {
      List<UserExcel> userList = baseMapper.exportUser(queryWrapper);
      userList.forEach(user -> {
         user.setUserTypeName(DictCache.getValue(DictEnum.USER_TYPE, user.getUserType()));
         user.setRoleName(StringUtil.join(SysCache.getRoleNames(user.getRoleId())));
         user.setDeptName(StringUtil.join(SysCache.getDeptNames(user.getDeptId())));
         user.setPostName(StringUtil.join(SysCache.getPostNames(user.getPostId())));
      });
      return userList;
   }
//   @Override
//   public List<UserExcel> exportUser(Wrapper<User> queryWrapper) {
//      List<UserExcel> userList = baseMapper.exportUser(queryWrapper);
//      userList.forEach(user -> {
//         user.setUserTypeName(DictCache.getValue(DictEnum.USER_TYPE, user.getUserType()));
//         user.setRoleName(StringUtil.join(SysCache.getRoleNames(user.getRoleId())));
//         user.setDeptName(StringUtil.join(SysCache.getDeptNames(user.getDeptId())));
//         user.setPostName(StringUtil.join(SysCache.getPostNames(user.getPostId())));
//      });
//      return userList;
//   }
   @Override
   @Transactional(rollbackFor = Exception.class)
@@ -438,5 +419,51 @@
      return page.setRecords(baseMapper.selectUserPageSecurity(page, user, deptIdList, tenantId));
   }
   /**
    * 自定义用户列表(只有保安员),根据保安员查
    */
   @Override
   public List<User> selectUserPageSecurityUnit(IPage<User> page, User user, Long deptId, String tenantId) {
      List<Long> deptIdList = SysCache.getDeptChildIds(deptId);
      return baseMapper.selectUserPageSecurity(page, user, deptIdList, tenantId);
   }
   /**
    * 保安员列表,帅选无保安证
    * @param page
    * @param user
    * @param deptId
    * @param tenantId
    * @return
    */
   @Override
   public IPage<UserVO> selectUserPageSecurityApply(IPage<UserVO> page, UserVO user, Long deptId, String tenantId) {
      List<Long> deptIdList = SysCache.getDeptChildIds(deptId);
      return baseMapper.selectUserPageSecurityApply(page, user, null, tenantId);
   }
   /**
    * 保安员列表树 安员列表树,帅选无保安证,下拉tree
    * @param user 用户信息
    * @return
    */
   @Override
   public List<TreeNode> getSecurityApplyTree(UserVO user) {
      Map<Long,TreeNode> map = baseMapper.getSecurityApplyTree(user);
      List<TreeNode> tree = new ArrayList<>();
      map.forEach((id,treeNode) ->{
         if (map.containsKey(treeNode.getParentId())){
            map.get(treeNode.getParentId()).getChildren().add(treeNode);
         }else {
            tree.add(treeNode);
         }
      });
      return tree;
   }
   @Override
   public List<Map<Object, Object>> selectInr(String deptid) {
      return baseMapper.selectInr(deptid);
   }
}