diff --git a/src/main/java/cn/palmte/work/model/process/ProjectProcessDetail.java b/src/main/java/cn/palmte/work/model/process/ProjectProcessDetail.java index f85c53e..3e61c8c 100644 --- a/src/main/java/cn/palmte/work/model/process/ProjectProcessDetail.java +++ b/src/main/java/cn/palmte/work/model/process/ProjectProcessDetail.java @@ -2,6 +2,7 @@ package cn.palmte.work.model.process; import java.math.BigDecimal; import java.util.List; +import java.util.Optional; import cn.palmte.work.model.Project; import cn.palmte.work.model.ProjectBudgetIncomeDetail; @@ -21,7 +22,11 @@ public class ProjectProcessDetail { private Project project; private ProjectProcess process; private ProjectType projectType; + + private String projectTypeDesc; + private CooperationType cooperationType; + private String cooperationTypeDesc; private List attachments; @@ -35,4 +40,17 @@ public class ProjectProcessDetail { public List supplierMaterials; public List procurementDetails; + public void setCooperationType(CooperationType cooperationType) { + this.cooperationType = cooperationType; + this.cooperationTypeDesc = Optional.ofNullable(cooperationType) + .map(CooperationType::getDescription) + .orElse(null); + } + + public void setProjectType(ProjectType projectType) { + this.projectType = projectType; + this.projectTypeDesc = Optional.ofNullable(projectType). + map(ProjectType::getDescription) + .orElse(null); + } }