|
@@ -99,30 +99,38 @@ public class ManageServiceImpl implements ManageService {
|
|
List<Long> companyIds = new ArrayList<>();
|
|
List<Long> companyIds = new ArrayList<>();
|
|
List<Long> costTypeIds = new ArrayList<>();
|
|
List<Long> costTypeIds = new ArrayList<>();
|
|
manages.stream().forEach(m -> {
|
|
manages.stream().forEach(m -> {
|
|
- deptIds.add(m.getDeptId());
|
|
|
|
|
|
+ if(Objects.nonNull(m.getDeptId())){
|
|
|
|
+ deptIds.add(m.getDeptId());
|
|
|
|
+ }
|
|
if(Objects.nonNull(m.getProjectId())){
|
|
if(Objects.nonNull(m.getProjectId())){
|
|
projectIds.add(m.getProjectId());
|
|
projectIds.add(m.getProjectId());
|
|
}
|
|
}
|
|
if(Objects.equals(m.getType(), ManageTypeEnum.COST)){
|
|
if(Objects.equals(m.getType(), ManageTypeEnum.COST)){
|
|
switch (m.getCompanyType()){
|
|
switch (m.getCompanyType()){
|
|
case COMPANY:
|
|
case COMPANY:
|
|
- companyIds.add(m.getCompanyId());
|
|
|
|
- break;
|
|
|
|
- case PROJECT:
|
|
|
|
- projectIds.add(m.getCompanyId());
|
|
|
|
|
|
+ if(Objects.nonNull(m.getCompanyId())){
|
|
|
|
+ companyIds.add(m.getCompanyId());
|
|
|
|
+ }
|
|
break;
|
|
break;
|
|
case DEPARTMENT:
|
|
case DEPARTMENT:
|
|
- deptIds.add(m.getCompanyId());
|
|
|
|
|
|
+ if(Objects.nonNull(m.getCompanyId())){
|
|
|
|
+ deptIds.add(m.getCompanyId());
|
|
|
|
+ }
|
|
break;
|
|
break;
|
|
default:
|
|
default:
|
|
break;
|
|
break;
|
|
}
|
|
}
|
|
}else {
|
|
}else {
|
|
- productIds.add(m.getProductId());
|
|
|
|
|
|
+ if(Objects.nonNull(m.getProductId())){
|
|
|
|
+ productIds.add(m.getProductId());
|
|
|
|
+ }
|
|
}
|
|
}
|
|
if(Objects.nonNull(m.getCostTypeId())){
|
|
if(Objects.nonNull(m.getCostTypeId())){
|
|
costTypeIds.add(m.getCostTypeId());
|
|
costTypeIds.add(m.getCostTypeId());
|
|
}
|
|
}
|
|
|
|
+ if(Objects.nonNull(m.getToProjectId())){
|
|
|
|
+ projectIds.add(m.getToProjectId());
|
|
|
|
+ }
|
|
});
|
|
});
|
|
//部门
|
|
//部门
|
|
List<Department> departments = departmentRepository.findByIds(deptIds);
|
|
List<Department> departments = departmentRepository.findByIds(deptIds);
|
|
@@ -159,19 +167,21 @@ public class ManageServiceImpl implements ManageService {
|
|
m.setDepartment(departmentMap.get(m.getDeptId()));
|
|
m.setDepartment(departmentMap.get(m.getDeptId()));
|
|
m.setProject(projectMap.get(m.getProjectId()));
|
|
m.setProject(projectMap.get(m.getProjectId()));
|
|
m.setProduct(producMap.get(m.getProductId()));
|
|
m.setProduct(producMap.get(m.getProductId()));
|
|
|
|
+ m.setToProject(projectMap.get(m.getToProjectId()));
|
|
if(Objects.nonNull(m.getCostTypeId())){
|
|
if(Objects.nonNull(m.getCostTypeId())){
|
|
m.setCostType(costTypeMap.get(m.getCostTypeId()));
|
|
m.setCostType(costTypeMap.get(m.getCostTypeId()));
|
|
}
|
|
}
|
|
if(Objects.nonNull(m.getCompanyType())){
|
|
if(Objects.nonNull(m.getCompanyType())){
|
|
switch (m.getCompanyType()){
|
|
switch (m.getCompanyType()){
|
|
case COMPANY:
|
|
case COMPANY:
|
|
- m.setCompany(companyMap.get(m.getCompanyId()));
|
|
|
|
- break;
|
|
|
|
- case PROJECT:
|
|
|
|
- m.setCompany(companyMap.get(m.getProjectId()));
|
|
|
|
|
|
+ if(Objects.nonNull(m.getCompanyId())){
|
|
|
|
+ m.setCompany(companyMap.get(m.getCompanyId()));
|
|
|
|
+ }
|
|
break;
|
|
break;
|
|
case DEPARTMENT:
|
|
case DEPARTMENT:
|
|
- m.setCompany(companyMap.get(m.getDeptId()));
|
|
|
|
|
|
+ if(Objects.nonNull(m.getCompanyId())){
|
|
|
|
+ m.setCompany(companyMap.get(m.getDeptId()));
|
|
|
|
+ }
|
|
break;
|
|
break;
|
|
default:
|
|
default:
|
|
break;
|
|
break;
|
|
@@ -217,17 +227,19 @@ public class ManageServiceImpl implements ManageService {
|
|
manage.setDepartment(departmentRepository.find(manage.getDeptId()));
|
|
manage.setDepartment(departmentRepository.find(manage.getDeptId()));
|
|
manage.setProject(projectRepository.find(manage.getProjectId()));
|
|
manage.setProject(projectRepository.find(manage.getProjectId()));
|
|
manage.setProduct(productRepository.find(manage.getProductId()));
|
|
manage.setProduct(productRepository.find(manage.getProductId()));
|
|
|
|
+ manage.setToProject(projectRepository.find(manage.getToProjectId()));
|
|
if(Objects.nonNull(manage.getCompanyType())){
|
|
if(Objects.nonNull(manage.getCompanyType())){
|
|
|
|
|
|
switch (manage.getCompanyType()){
|
|
switch (manage.getCompanyType()){
|
|
case COMPANY:
|
|
case COMPANY:
|
|
- manage.setCompany(companyRepository.find(manage.getCompanyId()));
|
|
|
|
- break;
|
|
|
|
- case PROJECT:
|
|
|
|
- manage.setCompany(projectRepository.find(manage.getCompanyId()));
|
|
|
|
|
|
+ if(Objects.nonNull(manage.getCompanyId())){
|
|
|
|
+ manage.setCompany(companyRepository.find(manage.getCompanyId()));
|
|
|
|
+ }
|
|
break;
|
|
break;
|
|
case DEPARTMENT:
|
|
case DEPARTMENT:
|
|
- manage.setCompany(departmentRepository.find(manage.getDeptId()));
|
|
|
|
|
|
+ if(Objects.nonNull(manage.getCompanyId())){
|
|
|
|
+ manage.setCompany(departmentRepository.find(manage.getDeptId()));
|
|
|
|
+ }
|
|
break;
|
|
break;
|
|
default:
|
|
default:
|
|
break;
|
|
break;
|