Backend support for operation milestones with activities

[sdc.git] / common-be / src / main / java / org / openecomp / sdc / be / datatypes / elements / OperationDataDefinition.java

diff --git a/common-be/src/main/java/org/openecomp/sdc/be/datatypes/elements/OperationDataDefinition.java b/common-be/src/main/java/org/openecomp/sdc/be/datatypes/elements/OperationDataDefinition.java
index f93e41c..a4bc7f1 100644 (file)
--- a/common-be/src/main/java/org/openecomp/sdc/be/datatypes/elements/OperationDataDefinition.java
+++ b/common-be/src/main/java/org/openecomp/sdc/be/datatypes/elements/OperationDataDefinition.java
@@ -21,7 +21,9 @@
 package org.openecomp.sdc.be.datatypes.elements;
 
 import com.fasterxml.jackson.annotation.JsonCreator;
+import java.util.Map;
 import org.openecomp.sdc.be.datatypes.enums.JsonPresentationFields;
+import org.openecomp.sdc.be.datatypes.enums.MilestoneTypeEnum;
 import org.openecomp.sdc.be.datatypes.tosca.ToscaDataDefinition;
 
 import java.io.Serializable;
@@ -46,6 +48,7 @@ public class OperationDataDefinition extends ToscaDataDefinition implements Seri
     }
 
     public OperationDataDefinition(OperationDataDefinition p) {
+        setMilestones(p.getMilestones());
         setDescription(p.getDescription());
         setImplementation(p.getImplementation());
         setInputs(p.getInputs());
@@ -59,6 +62,15 @@ public class OperationDataDefinition extends ToscaDataDefinition implements Seri
         setWorkflowVersion(p.getWorkflowVersion());
     }
 
+    public void setMilestones(Map<String, MilestoneDataDefinition> milestones) {
+        setToscaPresentationValue(JsonPresentationFields.OPERATION_MILESTONES, milestones);
+    }
+
+    public Map<String, MilestoneDataDefinition>  getMilestones() {
+        return (Map<String, MilestoneDataDefinition> ) getToscaPresentationValue(
+            JsonPresentationFields.OPERATION_MILESTONES);
+    }
+
     public String getDescription() {
         return (String) getToscaPresentationValue(JsonPresentationFields.DESCRIPTION);
     }