|
@@ -7,19 +7,19 @@ import lombok.AllArgsConstructor;
|
|
|
import org.springblade.common.utils.SnowFlakeUtil;
|
|
|
import org.springblade.control.dto.ProjectCostBudgetDTO;
|
|
|
import org.springblade.control.dto.ProjectCostBudgetStatsDTO;
|
|
|
-import org.springblade.control.entity.ControlProjectInfo;
|
|
|
-import org.springblade.control.entity.ProjectCostBudget;
|
|
|
-import org.springblade.control.entity.ProjectCostBudgetStats;
|
|
|
+import org.springblade.control.entity.*;
|
|
|
import org.springblade.control.mapper.ProjectCostBudgetMapper;
|
|
|
import org.springblade.control.mapper.ProjectCostBudgetStatsMapper;
|
|
|
import org.springblade.control.service.IProjectCostBudgetService;
|
|
|
import org.springblade.control.service.IProjectCostBudgetStatsService;
|
|
|
import org.springblade.control.service.IProjectInfoService;
|
|
|
+import org.springblade.control.service.IProjectProcessService;
|
|
|
import org.springblade.control.vo.ProjectCostBudgetStatsVO;
|
|
|
import org.springblade.control.vo.ProjectCostBudgetVO;
|
|
|
import org.springblade.core.log.exception.ServiceException;
|
|
|
import org.springblade.core.mp.base.BaseServiceImpl;
|
|
|
import org.springblade.core.mp.support.Query;
|
|
|
+import org.springblade.manager.entity.ProjectInfo;
|
|
|
import org.springframework.stereotype.Service;
|
|
|
import org.springframework.transaction.annotation.Transactional;
|
|
|
|
|
@@ -39,6 +39,8 @@ import java.util.stream.Collectors;
|
|
|
public class ProjectCostBudgetStatsServiceImpl extends BaseServiceImpl<ProjectCostBudgetStatsMapper, ProjectCostBudgetStats> implements IProjectCostBudgetStatsService {
|
|
|
private final IProjectCostBudgetService budgetService;
|
|
|
private final IProjectInfoService projectInfoService;
|
|
|
+ private final IProjectProcessService processService;
|
|
|
+
|
|
|
|
|
|
/**
|
|
|
* 新增项目成本预算
|
|
@@ -68,10 +70,12 @@ public class ProjectCostBudgetStatsServiceImpl extends BaseServiceImpl<ProjectCo
|
|
|
ProjectCostBudgetStatsVO vo = baseMapper.getBudgetStatsById(id);
|
|
|
//查询具体明细并分组存入
|
|
|
List<ProjectCostBudget> list = budgetService.budgetListByProjectId(vo.getProjectId());
|
|
|
- Map<Integer, List<ProjectCostBudget>> map = list.parallelStream().collect(Collectors.groupingBy(ProjectCostBudget::getUnitType));
|
|
|
- vo.setBuildUnit(map.get(1));
|
|
|
- vo.setSupervisorUnit(map.get(2));
|
|
|
- vo.setConstructUnit(map.get(3));
|
|
|
+ if (list != null && list.size() > 0) {
|
|
|
+ Map<Integer, List<ProjectCostBudget>> map = list.parallelStream().collect(Collectors.groupingBy(ProjectCostBudget::getUnitType));
|
|
|
+ vo.setBuildUnit(map.get(1));
|
|
|
+ vo.setSupervisorUnit(map.get(2));
|
|
|
+ vo.setConstructUnit(map.get(3));
|
|
|
+ }
|
|
|
return vo;
|
|
|
}
|
|
|
|
|
@@ -121,4 +125,36 @@ public class ProjectCostBudgetStatsServiceImpl extends BaseServiceImpl<ProjectCo
|
|
|
vo.setPreviewList(list);
|
|
|
return vo;
|
|
|
}
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 获取未做预算的项目列表
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public List<ControlProjectInfo> getProjectList() {
|
|
|
+ return baseMapper.getProjectList();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 获取费用分类
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public List<DictInfo> getCostTypeDict() {
|
|
|
+ return baseMapper.getCostTypeDict();
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 获取项目环节
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public List<ProjectProcess> getProcessList(Long projectId) {
|
|
|
+ return baseMapper.getProcessList(projectId);
|
|
|
+ }
|
|
|
+
|
|
|
+ /**
|
|
|
+ * 获取项目环节
|
|
|
+ */
|
|
|
+ @Override
|
|
|
+ public List<DictInfo> getPostDict() {
|
|
|
+ return baseMapper.getPostDict();
|
|
|
+ }
|
|
|
}
|