package com.doumee.service.aware;
|
|
import com.doumee.core.aware.DataPermissionMapping;
|
import com.doumee.core.aware.DefaultDataPermissionAware;
|
import com.doumee.core.constants.Constants;
|
import com.doumee.core.constants.DataPermissionConstants;
|
import com.doumee.dao.system.dto.QuerySystemDepartmentDTO;
|
import com.doumee.dao.system.model.SystemDepartmentUser;
|
import com.doumee.dao.system.vo.SystemDepartmentListVO;
|
import com.doumee.service.system.SystemDepartmentService;
|
import com.doumee.service.system.SystemDepartmentUserService;
|
import org.apache.commons.lang3.StringUtils;
|
import org.springframework.beans.BeanUtils;
|
import org.springframework.beans.factory.annotation.Autowired;
|
import org.springframework.stereotype.Component;
|
import org.springframework.util.CollectionUtils;
|
|
import java.util.ArrayList;
|
import java.util.Collections;
|
import java.util.List;
|
|
/**
|
* 部门数据权限控制
|
* @author Eva.Caesar Liu
|
* @date 2022/03/15 09:54
|
*/
|
@Component
|
public class DepartmentDataPermissionAware extends DefaultDataPermissionAware<SystemDepartmentListVO> {
|
|
@Autowired
|
private SystemDepartmentService systemDepartmentService;
|
|
@Autowired
|
private SystemDepartmentUserService systemDepartmentUserService;
|
|
@Override
|
public DataPermissionConstants.Module module() {
|
return DataPermissionConstants.Module.DEPARTMENT;
|
}
|
|
@Override
|
public List<SystemDepartmentListVO> defaultData(String userId) {
|
return onlyUser(userId);
|
}
|
|
/**
|
* 全部数据
|
*
|
* @return List<SystemDepartmentListVO>
|
*/
|
@DataPermissionMapping(value = DataPermissionConstants.Type.ALL, priority = 1)
|
public List<SystemDepartmentListVO> all() {
|
QuerySystemDepartmentDTO dto = new QuerySystemDepartmentDTO();
|
return this.getRootList(systemDepartmentService.findList(dto));
|
}
|
|
/**
|
* 自定义
|
*
|
* @param customData 自定义数据ID集
|
* @return List<SystemDepartmentListVO>
|
*/
|
@DataPermissionMapping(value = DataPermissionConstants.Type.DEPARTMENT_CUSTOM, priority = 2, injectCustomData = true)
|
public List<SystemDepartmentListVO> custom(String customData) {
|
if (StringUtils.isBlank(customData)) {
|
return Collections.emptyList();
|
}
|
List<Integer> ids = new ArrayList<>();
|
String[] stringIds = customData.split(",");
|
for(String stringId : stringIds) {
|
ids.add(Integer.valueOf(stringId));
|
}
|
QuerySystemDepartmentDTO dto = new QuerySystemDepartmentDTO();
|
dto.setIds(ids);
|
List<SystemDepartmentListVO> departmentListVo = systemDepartmentService.findList(dto);
|
return this.getRootList(departmentListVo);
|
}
|
|
/**
|
* 用户所属部门及其子孙部门
|
*
|
* @param userId 用户ID
|
* @return List<SystemDepartmentListVO>
|
*/
|
@DataPermissionMapping(value = DataPermissionConstants.Type.DEPARTMENT_CHILDREN, priority = 3, injectUser = true)
|
public List<SystemDepartmentListVO> children(String userId) {
|
return this.getRootList(getUserChildren(userId));
|
}
|
|
/**
|
* 用户所属部门及其子部门
|
*
|
* @param userId 用户ID
|
* @return List<SystemDepartmentListVO>
|
*/
|
@DataPermissionMapping(value = DataPermissionConstants.Type.DEPARTMENT_CHILD, priority = 4, injectUser = true)
|
public List<SystemDepartmentListVO> child(String userId) {
|
List<SystemDepartmentListVO> children = this.getRootList(getUserChildren(userId));
|
for (SystemDepartmentListVO root : children) {
|
if (CollectionUtils.isEmpty(root.getChildren())) {
|
continue;
|
}
|
for (SystemDepartmentListVO child : root.getChildren()) {
|
if (CollectionUtils.isEmpty(child.getChildren())) {
|
continue;
|
}
|
child.setHasChildren(Boolean.TRUE);
|
child.setChildren(null);
|
}
|
}
|
return children;
|
}
|
|
/**
|
* 仅用户所属部门
|
*
|
* @param userId 用户ID
|
* @return List<SystemDepartmentListVO>
|
*/
|
@DataPermissionMapping(value = DataPermissionConstants.Type.DEPARTMENT, priority = 5, injectUser = true)
|
public List<SystemDepartmentListVO> onlyUser(String userId) {
|
SystemDepartmentListVO userDepartment = this.getUserDepartment(userId);
|
if (userDepartment == null) {
|
return Collections.emptyList();
|
}
|
return new ArrayList<SystemDepartmentListVO>(){{
|
this.add(userDepartment);
|
}};
|
}
|
|
/**
|
* 获取根部门
|
*/
|
private List<SystemDepartmentListVO> getRootList(List<SystemDepartmentListVO> departments) {
|
if (CollectionUtils.isEmpty(departments)) {
|
return Collections.emptyList();
|
}
|
List<SystemDepartmentListVO> rootDepartments = new ArrayList<>();
|
// 添加根部门
|
for (SystemDepartmentListVO currentDepartment : departments) {
|
boolean hasParent = false;
|
for (SystemDepartmentListVO department: departments) {
|
if (department.getId().equals(currentDepartment.getParentId())) {
|
hasParent = true;
|
break;
|
}
|
}
|
if (!hasParent) {
|
rootDepartments.add(currentDepartment);
|
}
|
}
|
// 移除根部门
|
for (SystemDepartmentListVO rootDepartment : rootDepartments) {
|
departments.removeIf(department -> department.getId().equals(rootDepartment.getId()));
|
}
|
// 填充子部门
|
for (SystemDepartmentListVO child : rootDepartments) {
|
this.fillChildren(child, departments);
|
}
|
return rootDepartments;
|
}
|
|
/**
|
* 获取用户部门
|
*/
|
private SystemDepartmentListVO getUserDepartment (String userId) {
|
SystemDepartmentUser queryDto = new SystemDepartmentUser();
|
queryDto.setUserId(userId);
|
queryDto.setDeleted(Constants.ONE);
|
SystemDepartmentUser departmentUser = systemDepartmentUserService.findOne(queryDto);
|
if (departmentUser == null) {
|
return null;
|
}
|
QuerySystemDepartmentDTO dto = new QuerySystemDepartmentDTO();
|
dto.setId(departmentUser.getDepartmentId());
|
List<SystemDepartmentListVO> systemDepartments = systemDepartmentService.findList(dto);
|
if (systemDepartments.isEmpty()) {
|
return null;
|
}
|
return systemDepartments.get(0);
|
}
|
|
/**
|
* 获取用户子孙部门
|
*/
|
private List<SystemDepartmentListVO> getUserChildren (String userId) {
|
SystemDepartmentListVO userDepartment = this.getUserDepartment(userId);
|
if (userDepartment == null) {
|
return Collections.emptyList();
|
}
|
// 查询用户所在部门以下部门
|
List<SystemDepartmentListVO> departmentListVo = new ArrayList<>();
|
List<Integer> ids = systemDepartmentService.findChildren(userDepartment.getId());
|
if (!ids.isEmpty()) {
|
QuerySystemDepartmentDTO dto = new QuerySystemDepartmentDTO();
|
dto.setIds(ids);
|
departmentListVo = systemDepartmentService.findList(dto);
|
}
|
departmentListVo.add(userDepartment);
|
return departmentListVo;
|
}
|
|
/**
|
* 填充子部门
|
*/
|
private void fillChildren(SystemDepartmentListVO parent, List<SystemDepartmentListVO> departments) {
|
if (departments.size() == 0) {
|
return;
|
}
|
if (parent.getChildren() == null) {
|
parent.setChildren(new ArrayList<>());
|
}
|
List<Integer> handledIds = new ArrayList<>();
|
for (SystemDepartmentListVO department : departments) {
|
if (parent.getId().equals(department.getParentId())) {
|
SystemDepartmentListVO child = new SystemDepartmentListVO();
|
BeanUtils.copyProperties(department, child, "children");
|
child.setChildren(new ArrayList<>());
|
parent.getChildren().add(child);
|
handledIds.add(department.getId());
|
}
|
}
|
departments.removeIf(menu -> handledIds.contains(menu.getId()));
|
parent.setHasChildren(Boolean.TRUE);
|
if (parent.getChildren().size() > 0) {
|
parent.setHasChildren(Boolean.FALSE);
|
for (SystemDepartmentListVO child : parent.getChildren()) {
|
this.fillChildren(child, departments);
|
}
|
}
|
}
|
}
|
