|
|
@@ -55,16 +55,27 @@ public class FrameworkServiceImpl implements IFrameworkService {
|
|
|
if (!Objects.nonNull(employeeDO)){
|
|
|
return ResponseDTO.dataErrorParam("数据不存在!请检查");
|
|
|
}
|
|
|
- List<DepartmentDTO> departmentDTOS = dRepository.queryDepartmentByDepartmentId(employeeDO.getDepartmentId());
|
|
|
+ List<DepartmentDTO> departmentDTOSChild = dRepository.queryDepartmentByDepartmentId(employeeDO.getDepartmentId());
|
|
|
+ Set<Long> ids = departmentDTOSChild.stream().map(DepartmentDTO::getParentId)
|
|
|
+ .collect(Collectors.toSet());
|
|
|
+ List<DepartmentDTO> departmentDTOSParent = dRepository.supportsDepartmentQueryByIds(ids);
|
|
|
List<FrameworkDTO> frameworkDTOS = new ArrayList<>();
|
|
|
- for (DepartmentDTO departmentDTO : departmentDTOS) {
|
|
|
+ for (DepartmentDTO departmentDTOChild : departmentDTOSChild) {
|
|
|
FrameworkDTO frameworkDTO = new FrameworkDTO();
|
|
|
- if (departmentDTO.getDepartmentId().equals(employeeDO.getDepartmentId())){
|
|
|
- frameworkDTO.setId(employeeDO.getEmployeeId());
|
|
|
- frameworkDTO.setName(employeeDO.getActualName());
|
|
|
+ if (departmentDTOChild.getParentId() == 0){
|
|
|
+ frameworkDTO.setId(departmentDTOChild.getDepartmentId());
|
|
|
+ frameworkDTO.setName(departmentDTOChild.getName());
|
|
|
frameworkDTO.setNodeType("USER");
|
|
|
- frameworkDTO.setParentId(employeeDO.getDepartmentId());
|
|
|
- frameworkDTO.setParentName(departmentDTO.getName());
|
|
|
+ } else {
|
|
|
+ for (DepartmentDTO departmentDTOParent : departmentDTOSParent) {
|
|
|
+ if (departmentDTOParent.getDepartmentId().equals(departmentDTOChild.getParentId())){
|
|
|
+ frameworkDTO.setId(departmentDTOChild.getDepartmentId());
|
|
|
+ frameworkDTO.setName(departmentDTOChild.getName());
|
|
|
+ frameworkDTO.setNodeType("USER");
|
|
|
+ frameworkDTO.setParentId(departmentDTOParent.getDepartmentId());
|
|
|
+ frameworkDTO.setParentName(departmentDTOParent.getName());
|
|
|
+ }
|
|
|
+ }
|
|
|
}
|
|
|
frameworkDTOS.add(frameworkDTO);
|
|
|
}
|