| package com.doumee.biz.system.impl; | 
|   | 
| import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; | 
| import com.baomidou.mybatisplus.core.conditions.update.UpdateWrapper; | 
| import com.doumee.biz.system.SystemDictDataBiz; | 
| import com.doumee.biz.system.SystemUserBiz; | 
| import com.doumee.core.annotation.excel.ExcelImporter; | 
| import com.doumee.core.exception.BusinessException; | 
| import com.doumee.core.constants.ResponseStatus; | 
| import com.doumee.core.model.LoginUserInfo; | 
| import com.doumee.core.utils.Constants; | 
| import com.doumee.core.utils.Utils; | 
| import com.doumee.dao.system.SystemDepartmentUserMapper; | 
| import com.doumee.dao.system.dto.*; | 
| import com.doumee.dao.system.dto.UpdatePwdDto; | 
| import com.doumee.dao.system.model.*; | 
| import com.doumee.service.system.*; | 
| import org.apache.commons.lang3.RandomStringUtils; | 
| import org.apache.commons.lang3.StringUtils; | 
| import org.apache.shiro.SecurityUtils; | 
| import org.springframework.beans.factory.annotation.Autowired; | 
| import org.springframework.stereotype.Service; | 
| import org.springframework.transaction.annotation.Transactional; | 
| import org.springframework.util.CollectionUtils; | 
| import org.springframework.web.multipart.MultipartFile; | 
|   | 
| import java.util.Date; | 
| import java.util.List; | 
| import java.util.Map; | 
| import java.util.stream.Collectors; | 
|   | 
| @Service | 
| public class SystemUserBizImpl implements SystemUserBiz { | 
|   | 
|     @Autowired | 
|     private SystemUserService systemUserService; | 
|   | 
|     @Autowired | 
|     private SystemUserRoleService systemUserRoleService; | 
|   | 
|     @Autowired | 
|     private SystemDepartmentUserService systemDepartmentUserService; | 
|   | 
|     @Autowired | 
|     private SystemPositionUserService systemPositionUserService; | 
|   | 
|     @Autowired | 
|     private SystemDepartmentService systemDepartmentService; | 
|   | 
|     @Autowired | 
|     private SystemDictDataBiz systemDictDataBiz; | 
|   | 
|     @Transactional(rollbackFor = {Exception.class,BusinessException.class}) | 
|     @Override | 
|     public void deleteById(Integer id) { | 
|         SystemUser user = systemUserService.findById(id); | 
|         if (user == null) { | 
|             return; | 
|         } | 
|         if (user.getFixed()) { | 
|             throw new BusinessException(ResponseStatus.NOT_ALLOWED.getCode(), "请勿删除" + user.getUsername() + ",因为这是固定用户"); | 
|         } | 
|         UpdateWrapper<SystemDepartmentUser> wrapper = new UpdateWrapper<>(); | 
|         wrapper.lambda() | 
|                 .eq(SystemDepartmentUser::getUserId,id) | 
|                 .set(SystemDepartmentUser::getDeleted,true); | 
|         systemDepartmentUserService.update(wrapper); | 
|         systemUserService.deleteById(id); | 
|     } | 
|   | 
|     @Override | 
|     @Transactional( rollbackFor = {Exception.class, BusinessException.class}) | 
|     public void deleteByIdInBatch(List<Integer> ids) { | 
|         if (CollectionUtils.isEmpty(ids)) { | 
|             return; | 
|         } | 
|         for (Integer id : ids) { | 
|             this.deleteById(id); | 
|         } | 
|     } | 
|   | 
|     @Override | 
|     public void updatePwd(UpdatePwdDto dto) { | 
|         SystemUser user = systemUserService.findById(dto.getUserId()); | 
|         if (user.getDeleted()) { | 
|             throw new BusinessException(ResponseStatus.DATA_EMPTY.getCode(), "用户不存在或已被删除"); | 
|         } | 
|         // 验证原始密码 | 
|         if (!user.getPassword().equals(Utils.Secure.encryptPassword(dto.getOldPwd(), user.getSalt()))) { | 
|             throw new BusinessException(ResponseStatus.PWD_INCORRECT.getCode(), "原始密码不正确"); | 
|         } | 
|         // 修改密码 | 
|         SystemUser newUser = new SystemUser(); | 
|         newUser.setId(dto.getUserId()); | 
|         newUser.setPassword(Utils.Secure.encryptPassword(dto.getNewPwd(), user.getSalt())); | 
|         systemUserService.updateById(newUser); | 
|     } | 
|   | 
|     @Override | 
|     public void resetPwd(ResetSystemUserPwdDTO dto) { | 
|         // 查询用户 | 
|         SystemUser systemUser = systemUserService.findById(dto.getId()); | 
|         if (systemUser == null || systemUser.getDeleted()) { | 
|             throw new BusinessException(ResponseStatus.DATA_EMPTY.getCode(), "用户不存在或已被删除"); | 
|         } | 
|         // 修改密码 | 
|         SystemUser updateUserDto = new SystemUser(); | 
|         updateUserDto.setId(dto.getId()); | 
|         updateUserDto.setUpdateUser(dto.getOperaUserId()); | 
|         updateUserDto.setPassword(Utils.Secure.encryptPassword(dto.getPassword(), systemUser.getSalt())); | 
|         systemUserService.updateById(updateUserDto); | 
|     } | 
|   | 
|     @Override | 
|     @Transactional(rollbackFor = {Exception.class,BusinessException.class}) | 
|     public void create(CreateSystemUserDTO systemUser) { | 
|         // 验证用户名 | 
|         SystemUser queryUserDto = new SystemUser(); | 
|         queryUserDto.setUsername(systemUser.getUsername()); | 
|         queryUserDto.setDeleted(Boolean.FALSE); | 
|         SystemUser user = systemUserService.findOne(queryUserDto); | 
|         if (user != null) { | 
|             throw new BusinessException(ResponseStatus.DATA_EXISTS.getCode(), "用户名【"+systemUser.getUsername()+"】不能重复"); | 
|         } | 
|         // 验证工号 | 
|        if (StringUtils.isNotBlank(systemUser.getEmpNo())) { | 
|             queryUserDto = new SystemUser(); | 
|             queryUserDto.setDeleted(Boolean.FALSE); | 
|             queryUserDto.setEmpNo(systemUser.getEmpNo()); | 
|             user = systemUserService.findOne(queryUserDto); | 
|             if (user != null) { | 
|                 throw new BusinessException(ResponseStatus.DATA_EXISTS.getCode(), "工号【"+systemUser.getEmpNo()+"】已存在"); | 
|             } | 
|         } | 
|         // 生成密码盐 | 
|         String salt = RandomStringUtils.randomAlphabetic(6); | 
|         // 生成密码 | 
|         systemUser.setPassword(Utils.Secure.encryptPassword(systemDictDataBiz.queryByCode(Constants.SYSTEM,Constants.INITIAL_PASSWORD).getCode(), salt)); | 
|         systemUser.setSalt(salt); | 
|         systemUser.setSource(Constants.ZERO); | 
|   | 
|         // 创建用户记录 | 
|         systemUser.setType(Constants.ZERO); | 
|         Integer userId = systemUserService.create(systemUser); | 
|         // 设置部门 | 
|         if (systemUser.getDepartmentId() != null) { | 
|             SystemDepartmentUser systemDepartmentUser = new SystemDepartmentUser(); | 
|             systemDepartmentUser.setDepartmentId(systemUser.getDepartmentId()); | 
|             systemDepartmentUser.setUserId(userId); | 
|             systemDepartmentUser.setOperaUser(systemUser.getCreateUser()); | 
|             systemDepartmentUser.setOperaTime(new Date()); | 
|             systemDepartmentUserService.create(systemDepartmentUser); | 
|         }else { | 
|             SystemDepartmentUser systemDepartmentUser = new SystemDepartmentUser(); | 
|             QueryWrapper<SystemDepartment> wrapper = new QueryWrapper<>(); | 
|             wrapper.lambda().isNull(SystemDepartment::getParentId).last("limit 1"); | 
|             SystemDepartment systemDepartment = systemDepartmentService.findOne(wrapper); | 
|             systemDepartmentUser.setDepartmentId(systemDepartment.getId()); | 
|             systemDepartmentUser.setUserId(userId); | 
|             systemDepartmentUser.setOperaUser(systemUser.getCreateUser()); | 
|             systemDepartmentUser.setOperaTime(new Date()); | 
|             systemDepartmentUserService.create(systemDepartmentUser); | 
|         } | 
|         // 设置岗位 | 
| //        if (systemUser.getPositionIds() != null && systemUser.getPositionIds().size() > 0) { | 
| //            for (Integer positionId : systemUser.getPositionIds()) { | 
| //                SystemPositionUser systemPositionUser = new SystemPositionUser(); | 
| //                systemPositionUser.setPositionId(positionId); | 
| //                systemPositionUser.setUserId(userId); | 
| //                systemPositionUser.setOperaUser(systemUser.getCreateUser()); | 
| //                systemPositionUser.setOperaTime(new Date()); | 
| //                systemPositionUserService.create(systemPositionUser); | 
| //            } | 
| //        } | 
|     } | 
|   | 
|     @Override | 
|     public void updateById(CreateSystemUserDTO systemUser) { | 
|         // 验证用户名 | 
|         SystemUser queryUserDto = new SystemUser(); | 
|         SystemUser user = null; | 
|         if (StringUtils.isNotBlank(systemUser.getUsername())){ | 
|   | 
|             queryUserDto.setUsername(systemUser.getUsername()); | 
|             queryUserDto.setDeleted(Boolean.FALSE); | 
|             user = systemUserService.findOne(queryUserDto); | 
|             if (user != null && !user.getId().equals(systemUser.getId())) { | 
|                 throw new BusinessException(ResponseStatus.DATA_EXISTS.getCode(), "用户名【"+systemUser.getUsername()+"】已存在"); | 
|             } | 
|         } | 
|   | 
|         // 验证工号 | 
|         if (StringUtils.isNotBlank(systemUser.getEmpNo())) { | 
|             queryUserDto = new SystemUser(); | 
|             queryUserDto.setEmpNo(systemUser.getEmpNo()); | 
|             queryUserDto.setDeleted(Boolean.FALSE); | 
|             user = systemUserService.findOne(queryUserDto); | 
|             if (user != null && !user.getId().equals(systemUser.getId())) { | 
|                 throw new BusinessException(ResponseStatus.DATA_EXISTS.getCode(), "工号【"+systemUser.getEmpNo()+"】已存在"); | 
|             } | 
|         } | 
|         // 修改用户 | 
|         systemUserService.updateById(systemUser); | 
|         // 设置部门 | 
|   | 
|         if (systemUser.getDepartmentId() != null) { | 
|             SystemDepartmentUser deleteDepartmentDto = new SystemDepartmentUser(); | 
|             deleteDepartmentDto.setUserId(systemUser.getId()); | 
|             systemDepartmentUserService.delete(deleteDepartmentDto); | 
|             SystemDepartmentUser systemDepartmentUser = new SystemDepartmentUser(); | 
|             systemDepartmentUser.setDepartmentId(systemUser.getDepartmentId()); | 
|             systemDepartmentUser.setUserId(systemUser.getId()); | 
|             systemDepartmentUser.setOperaUser(systemUser.getUpdateUser()); | 
|             systemDepartmentUser.setOperaTime(new Date()); | 
|             systemDepartmentUserService.create(systemDepartmentUser); | 
|         } | 
|         // 设置岗位 | 
|     /*    SystemPositionUser deletePositionDto = new SystemPositionUser(); | 
|         deletePositionDto.setUserId(systemUser.getId()); | 
|         systemPositionUserService.delete(deletePositionDto); | 
|         if (systemUser.getPositionIds() != null && systemUser.getPositionIds().size() > 0) { | 
|             for (Integer positionId : systemUser.getPositionIds()) { | 
|                 SystemPositionUser systemPositionUser = new SystemPositionUser(); | 
|                 systemPositionUser.setPositionId(positionId); | 
|                 systemPositionUser.setUserId(systemUser.getId()); | 
|                 systemPositionUser.setOperaUser(systemUser.getUpdateUser()); | 
|                 systemPositionUser.setOperaTime(new Date()); | 
|                 systemPositionUserService.create(systemPositionUser); | 
|             } | 
|         }*/ | 
|     } | 
|   | 
|     @Override | 
|     @Transactional(rollbackFor = {Exception.class,BusinessException.class}) | 
|     public void createUserRole(CreateUserRoleDTO dto) { | 
|         // 删除关联角色 | 
|         SystemUserRole deleteDto = new SystemUserRole(); | 
|         deleteDto.setUserId(dto.getUserId()); | 
|         systemUserRoleService.delete(deleteDto); | 
|         // 新增新的角色 | 
|         for (Integer roleId : dto.getRoleIds()) { | 
|             SystemUserRole newUserRole = new SystemUserRole(); | 
|             newUserRole.setUserId(dto.getUserId()); | 
|             newUserRole.setRoleId(roleId); | 
|             newUserRole.setCreateUser(dto.getCreateUser()); | 
|             systemUserRoleService.create(newUserRole); | 
|         } | 
|     } | 
|   | 
|     @Override | 
|     public Integer importSystemUserBatch(MultipartFile file) { | 
|   | 
|         LoginUserInfo loginUserInfo = (LoginUserInfo) SecurityUtils.getSubject().getPrincipal(); | 
|   | 
|         try { | 
|             ExcelImporter ie = new ExcelImporter(file,1,0); | 
|             List<ImportSystemUserDTO> dataList = ie.getDataList(ImportSystemUserDTO.class, null); | 
|             if (CollectionUtils.isEmpty(dataList)){ | 
|                 throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(),"对不起,录入数据为空!"); | 
|             } | 
|             List<String> collect = dataList.stream() | 
|                     .map(s -> s.getDepartmentName()) | 
|                     .filter(s->StringUtils.isNotBlank(s)) | 
|                     .distinct() | 
|                     .collect(Collectors.toList()); | 
|             if (!CollectionUtils.isEmpty(collect)){ | 
|                 QueryWrapper<SystemDepartment> wrapper = new QueryWrapper<>(); | 
|                 wrapper.lambda().in(SystemDepartment::getName,collect); | 
|                 List<SystemDepartment> systemDepartments = systemDepartmentService.findList(wrapper); | 
|                 Map<Integer, String> collect1 = systemDepartments.stream().collect(Collectors.toMap(s -> s.getId(), s -> s.getName())); | 
|   | 
|                 dataList.forEach(s->{ | 
|                     CreateSystemUserDTO systemUser = new CreateSystemUserDTO(); | 
|                     systemUser.setDepartmentId(getKey(collect1,s.getDepartmentName())); | 
|                     systemUser.setUsername(s.getUsername()); | 
|                     systemUser.setRealname(s.getUsername()); | 
|                     systemUser.setEmpNo(s.getEmpNo()); | 
|                     systemUser.setMobile(s.getMobile()); | 
|                     systemUser.setPassword(s.getMobile()); | 
|                     systemUser.setCreateUser(loginUserInfo.getId()); | 
|                     systemUser.setUpdateUser(loginUserInfo.getId()); | 
|                     systemUser.setType(Constants.ZERO); | 
|   | 
|                     create(systemUser); | 
|                 }); | 
|   | 
|             }else { | 
|                 dataList.forEach(s->{ | 
|   | 
|                     CreateSystemUserDTO systemUser = new CreateSystemUserDTO(); | 
|                     QueryWrapper<SystemDepartment> wrapper = new QueryWrapper<>(); | 
|                     wrapper.lambda().isNull(SystemDepartment::getParentId).last("limit 1"); | 
|                     SystemDepartment systemDepartment = systemDepartmentService.findOne(wrapper); | 
|   | 
|                     systemUser.setDepartmentId(systemDepartment.getId()); | 
|                     systemUser.setUsername(s.getUsername()); | 
|                     systemUser.setRealname(s.getUsername()); | 
|                     systemUser.setEmpNo(s.getEmpNo()); | 
|                     systemUser.setMobile(s.getMobile()); | 
|                     systemUser.setPassword(s.getMobile()); | 
|                     systemUser.setCreateUser(loginUserInfo.getId()); | 
|                     systemUser.setUpdateUser(loginUserInfo.getId()); | 
|                     systemUser.setType(Constants.ZERO); | 
|                     create(systemUser); | 
|                     }); | 
|             } | 
|   | 
|         } catch (Exception e) { | 
|   | 
|             throw new BusinessException(ResponseStatus.BAD_REQUEST.getCode(),e.getMessage()); | 
|   | 
|         } | 
|         return null; | 
|     } | 
|   | 
|     private <K,V> K getKey(Map<K,V> map,V v){ | 
|   | 
|         for (Map.Entry<K,V>  entry: map.entrySet()){ | 
|   | 
|             if (v == null && entry.getValue() == null){ | 
|                 return entry.getKey(); | 
|             }else if ( v != null && v.equals(entry.getValue())){ | 
|                 return entry.getKey(); | 
|             } | 
|         } | 
|         return null; | 
|     } | 
| } |