com.diboot.iam.service.impl.IamUserServiceImpl Maven / Gradle / Ivy
/*
* Copyright (c) 2015-2020, www.dibo.ltd ([email protected]).
*
* Licensed under the Apache License, Version 2.0 (the "License"); you may not
* use this file except in compliance with the License. You may obtain a copy of
* the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
* WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
* License for the specific language governing permissions and limitations under
* the License.
*/
package com.diboot.iam.service.impl;
import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.toolkit.Wrappers;
import com.diboot.core.config.BaseConfig;
import com.diboot.core.config.Cons;
import com.diboot.core.event.OperationEvent;
import com.diboot.core.exception.BusinessException;
import com.diboot.core.service.impl.BaseServiceImpl;
import com.diboot.core.util.V;
import com.diboot.core.vo.LabelValue;
import com.diboot.core.vo.Pagination;
import com.diboot.core.vo.Status;
import com.diboot.iam.auth.IamCustomize;
import com.diboot.iam.dto.IamUserFormDTO;
import com.diboot.iam.entity.*;
import com.diboot.iam.mapper.IamUserMapper;
import com.diboot.iam.service.*;
import com.diboot.iam.util.IamSecurityUtils;
import com.diboot.iam.vo.IamUserVO;
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.ApplicationEventPublisher;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import java.util.*;
import java.util.stream.Collectors;
/**
* 系统用户相关Service实现
* @author [email protected]
* @version 2.0
* @date 2019-12-17
*/
@SuppressWarnings("JavaDoc")
@Service
@Slf4j
public class IamUserServiceImpl extends BaseServiceImpl implements IamUserService {
@Autowired
private IamUserRoleService iamUserRoleService;
@Autowired
private IamAccountService iamAccountService;
@Autowired
private IamOrgService iamOrgService;
@Autowired
private IamRoleService iamRoleService;
@Autowired
private IamUserPositionService iamUserPositionService;
@Autowired(required = false)
private IamCustomize iamCustomize;
@Autowired
private ApplicationEventPublisher applicationEventPublisher;
@Override
@Transactional(rollbackFor = Exception.class)
public boolean createUserRelatedInfo(IamUserFormDTO userFormDTO) {
// 创建用户信息
this.createEntity(userFormDTO);
// 如果提交的有账号信息，则新建账号信息
if (V.notEmpty(userFormDTO.getUsername())) {
// 新建account账号
this.createAccount(userFormDTO);
// 批量创建角色关联关系
iamUserRoleService.createUserRoleRelations(userFormDTO.getUserType(), userFormDTO.getId(), userFormDTO.getRoleIdList());
}
if (V.notEmpty(userFormDTO.getUserPositionList())) {
iamUserPositionService.updateUserPositionRelations(userFormDTO.getUserType(), userFormDTO.getId(), userFormDTO.getUserPositionList());
}
return true;
}
@Override
@Transactional(rollbackFor = Exception.class)
public boolean updateUserRelatedInfo(IamUserFormDTO userFormDTO) {
// 更新用户信息
this.updateEntity(userFormDTO);
if (userFormDTO.getUserPositionList() != null) {
iamUserPositionService.updateUserPositionRelations(userFormDTO.getUserType(), userFormDTO.getId(), userFormDTO.getUserPositionList());
}
IamAccount iamAccount = iamAccountService.getSingleEntity(
Wrappers.lambdaQuery()
.eq(IamAccount::getUserType, IamUser.class.getSimpleName())
.eq(IamAccount::getUserId, userFormDTO.getId())
);
if (iamAccount == null) {
if (V.isEmpty(userFormDTO.getUsername())){
return true;
} else {
// 新建account账号
this.createAccount(userFormDTO);
// 批量创建角色关联关系
iamUserRoleService.createUserRoleRelations(userFormDTO.getUserType(), userFormDTO.getId(), userFormDTO.getRoleIdList());
}
} else {
if (V.isEmpty(userFormDTO.getUsername())) {
// 删除账号
this.deleteAccount(userFormDTO.getId());
// 删除角色关联关系
iamUserRoleService.updateUserRoleRelations(iamAccount.getUserType(), iamAccount.getUserId(), Collections.emptyList());
} else {
// 更新账号
iamAccount.setAuthAccount(userFormDTO.getUsername()).setStatus(userFormDTO.getAccountStatus());
// 用户离职，状态停用
if(Cons.ENABLE_STATUS.I.name().equals(userFormDTO.getStatus())){
iamAccount.setStatus(Cons.ENABLE_STATUS.I.name());
// 对外发布用户离职的事件
applicationEventPublisher.publishEvent(new OperationEvent(IamUser.class.getSimpleName(), userFormDTO));
}
// 设置密码
if (V.notEmpty(userFormDTO.getPassword())){
iamAccount.setAuthSecret(userFormDTO.getPassword());
iamCustomize.encryptPwd(iamAccount);
}
iamAccountService.updateEntity(iamAccount);
// 批量更新角色关联关系
iamUserRoleService.updateUserRoleRelations(iamAccount.getUserType(), iamAccount.getUserId(), userFormDTO.getRoleIdList());
}
}
return true;
}
@Override
@Transactional(rollbackFor = Exception.class)
public boolean deleteUserAndRelatedInfo(String id) {
if (!exists(IamUser::getId, id)){
throw new BusinessException(Status.FAIL_OPERATION, "exception.business.userService.deleteRecordNonExist");
}
// 删除用户信息
this.deleteEntity(id);
// 删除账号及角色信息
this.deleteAccount(id);
// 删除岗位信息
this.deleteUserPositions(id);
return true;
}
@Override
public List filterDuplicateUserNums(List userNumList) {
if (V.isEmpty(userNumList)) {
return Collections.emptyList();
}
int totalSize = userNumList.size(), batchSize = BaseConfig.getBatchSize();
Set uniqueUserNumSet = new HashSet<>(totalSize);
int startInx = 0;
while (startInx < totalSize) {
List existUserNumList = this.checkUserNumDuplicate(
// 截取子列表：[0, batchSize)、[batchSize, 2*batchSize)...[n*batchSize, totalSize)
userNumList.subList(startInx, Math.min(startInx + batchSize, totalSize))
);
if (V.notEmpty(existUserNumList)) {
uniqueUserNumSet.addAll(existUserNumList);
}
startInx += batchSize;
}
return new ArrayList<>(uniqueUserNumSet);
}
@Override
public boolean isUserNumExists(String id, String userNum) {
if(V.isEmpty(userNum)){
return true;
}
LambdaQueryWrapper wrapper = Wrappers.lambdaQuery()
.select(IamUser::getUserNum)
.eq(IamUser::getUserNum, userNum);
if (V.notEmpty(id)){
wrapper.ne(IamUser::getId, id);
}
return exists(wrapper);
}
@Override
public List getUserIdsByManagerId(String managerId) {
if(managerId == null){
return null;
}
List orgIds = iamOrgService.getOrgIdsByManagerId(managerId);
if(V.isEmpty(orgIds)){
return Collections.emptyList();
}
List iamUserIds = getValuesOfField(
Wrappers.lambdaQuery().in(IamUser::getOrgId, orgIds),
IamUser::getId
);
return iamUserIds;
}
@Override
public String getUserLeaderId(String userId) {
IamUser iamUser = getEntity(userId);
if (V.isEmpty(iamUser)) {
return null;
}
IamOrg iamOrg = iamOrgService.getEntity(iamUser.getOrgId());
if (V.isEmpty(iamOrg) || V.isEmpty(iamOrg.getManagerId())) {
return null;
}
return iamOrg.getManagerId();
}
@Override
public List getUsersByRoleIds(List roleIds) {
List ids = iamUserRoleService.getUserIdsByRoleIds(roleIds);
return getEntityListByIds(ids);
}
@Override
public List getUsersByRoleCode(String roleCode) {
if (V.isEmpty(roleCode)) {
return Collections.emptyList();
}
return getUsersByRoleCodes(Arrays.asList(roleCode));
}
@Override
public List getUsersByRoleCodes(List roleCodes) {
if (V.isEmpty(roleCodes)) {
return Collections.emptyList();
}
List roleIds = iamRoleService.getValuesOfField(
Wrappers.lambdaQuery().select(IamRole::getId)
.in(IamRole::getCode, roleCodes),
IamRole::getId
);
if (V.isEmpty(roleIds)) {
return Collections.emptyList();
}
List ids = iamUserRoleService.getUserIdsByRoleIds(roleIds);
return getEntityListByIds(ids);
}
@Override
public Map getLabelValueMap(List ids) {
LambdaQueryWrapper queryWrapper = new QueryWrapper().lambda()
.select(IamUser::getRealname, IamUser::getId, IamUser::getUserNum)
.in(IamUser::getId, ids);
// 返回构建条件
return getEntityList(queryWrapper).stream().collect(
Collectors.toMap(ent -> ent.getId(),
ent -> new LabelValue(ent.getRealname(), ent.getId()).setExt(ent.getUserNum())));
}
@Override
public List getUserViewList(QueryWrapper queryWrapper, Pagination pagination, String orgId) {
List orgIds = new ArrayList<>();
// 获取当前部门及所有下属部门的人员列表
if (V.notEmpty(orgId)) {
orgIds.add(orgId);
// 获取所有下级部门列表
orgIds.addAll(iamOrgService.getChildOrgIds(orgId));
// 相应部门下岗位相关用户
LambdaQueryWrapper queryUserIds = Wrappers.lambdaQuery()
.eq(IamUserPosition::getUserType, IamUser.class.getSimpleName())
.in(IamUserPosition::getOrgId, orgIds);
List userIds = iamUserPositionService.getValuesOfField(queryUserIds, IamUserPosition::getUserId);
queryWrapper.and(query -> query.in(Cons.ColumnName.org_id.name(), orgIds)
.or().in(V.notEmpty(userIds), Cons.FieldName.id.name(), userIds));
}
// 查询指定页的数据
List voList = getViewObjectList(queryWrapper, pagination, IamUserVO.class);
if (V.isEmpty(orgIds)) {
return voList;
}
for (IamUserVO user : voList) {
List userPositionList = user.getUserPositionList();
if (V.notEmpty(userPositionList)) {
user.setUserPositionList(userPositionList.stream().filter(p -> orgIds.contains(p.getOrgId())).collect(Collectors.toList()));
}
}
return voList;
}
/**
* 刷新用户电话邮箱头像等信息
* @return
*/
@Override
public void refreshUserInfo(IamUser currentUser) {
IamUser latestInfo = getEntity(currentUser.getId());
currentUser
.setRealname(latestInfo.getRealname())
.setStatus(latestInfo.getStatus())
.setAvatarUrl(latestInfo.getAvatarUrl())
.setUserNum(latestInfo.getUserNum())
.setGender(latestInfo.getGender())
.setBirthdate(latestInfo.getBirthdate())
.setEmail(latestInfo.getEmail())
.setMobilePhone(latestInfo.getMobilePhone())
.setOrgId(latestInfo.getOrgId());
}
/**
* 检查重复用户编号
* @param userNumList
* @return
*/
private List checkUserNumDuplicate(List userNumList) {
if (V.isEmpty(userNumList)) {
return Collections.emptyList();
}
List iamUserNums = getValuesOfField(
Wrappers.lambdaQuery().in(IamUser::getUserNum, userNumList),
IamUser::getUserNum
);
return iamUserNums;
}
protected void createAccount(IamUserFormDTO userFormDTO) {
// 创建账号信息
IamAccount iamAccount = new IamAccount()
.setTenantId(userFormDTO.getTenantId())
.setUserType(IamUser.class.getSimpleName())
.setUserId(userFormDTO.getId())
.setAuthAccount(userFormDTO.getUsername())
.setAuthSecret(userFormDTO.getPassword())
.setAuthType(userFormDTO.getAuthType())
.setStatus(userFormDTO.getAccountStatus());
// 用户离职，账号停用
if(Cons.ENABLE_STATUS.I.name().equals(userFormDTO.getStatus())){
iamAccount.setStatus(Cons.ENABLE_STATUS.I.name());
}
// 保存账号
iamAccountService.createEntity(iamAccount);
}
private void deleteAccount(String userId) {
if (V.equals(userId, IamSecurityUtils.getCurrentUserId())) {
throw new BusinessException("exception.business.userService.deleteSelfAccount");
}
// 删除账号信息
iamAccountService.deleteEntities(
Wrappers.lambdaQuery()
.eq(IamAccount::getUserType, IamUser.class.getSimpleName())
.eq(IamAccount::getUserId, userId)
);
// 删除用户角色关联关系列表
iamUserRoleService.deleteUserRoleRelations(IamUser.class.getSimpleName(), userId);
}
private void deleteUserPositions(String userId) {
// 删除岗位信息
iamUserPositionService.deleteEntities(
Wrappers.lambdaQuery()
.eq(IamUserPosition::getUserType, IamUser.class.getSimpleName())
.eq(IamUserPosition::getUserId, userId)
);
}
/**
* 判断员工编号是否存在
* @param iamUser
* @return
*/
@Override
public void beforeCreate(IamUser iamUser){
if(isUserNumExists(null, iamUser.getUserNum())){
log.warn("保存用户异常: 员工编号{} 已存在，请重新设置！", iamUser.getUserNum());
throw new BusinessException(Status.FAIL_VALIDATION, "exception.business.userService.userNumExist", iamUser.getUserNum());
}
}
/**
* 判断员工编号是否存在
* @param iamUser
* @return
*/
@Override
public void beforeUpdate(IamUser iamUser){
if(isUserNumExists(iamUser.getId(), iamUser.getUserNum())){
log.warn("保存用户异常: 员工编号{} 已存在，请重新设置！", iamUser.getUserNum());
throw new BusinessException(Status.FAIL_VALIDATION, "exception.business.userService.userNumExist", iamUser.getUserNum());
}
}
}