Support Outputs during Import Service

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

diff --git a/common-be/src/main/java/org/openecomp/sdc/be/datatypes/elements/AttributeDataDefinition.java b/common-be/src/main/java/org/openecomp/sdc/be/datatypes/elements/AttributeDataDefinition.java
index c16efe9..77e219b 100644 (file)
--- a/common-be/src/main/java/org/openecomp/sdc/be/datatypes/elements/AttributeDataDefinition.java
+++ b/common-be/src/main/java/org/openecomp/sdc/be/datatypes/elements/AttributeDataDefinition.java
@@ -40,6 +40,7 @@ public class AttributeDataDefinition extends ToscaDataDefinition {
     private String outputPath;
     private String instanceUniqueId;
     private String attributeId;
+    private String model;
     private String parentUniqueId;
 
     public AttributeDataDefinition(final AttributeDataDefinition attributeDataDefinition) {
@@ -53,9 +54,11 @@ public class AttributeDataDefinition extends ToscaDataDefinition {
         this.setValue(attributeDataDefinition.getValue());
         this.setStatus(attributeDataDefinition.getStatus());
         this.setEntry_schema(attributeDataDefinition.getEntry_schema());
+        this.setSchema(attributeDataDefinition.getSchema());
         this.setOutputPath(attributeDataDefinition.getOutputPath());
         this.setInstanceUniqueId(attributeDataDefinition.getInstanceUniqueId());
         this.setAttributeId(attributeDataDefinition.getAttributeId());
+        this.setModel(attributeDataDefinition.getModel());
         this.setParentUniqueId(attributeDataDefinition.getParentUniqueId());
         this.setOutputId(attributeDataDefinition.getOutputId());
         if (CollectionUtils.isNotEmpty(attributeDataDefinition.getGetOutputValues())) {
@@ -81,6 +84,14 @@ public class AttributeDataDefinition extends ToscaDataDefinition {
         setToscaPresentationValue(JsonPresentationFields.OWNER_ID, ownerId);
     }
 
+    public String getName() {
+        return (String) getToscaPresentationValue(JsonPresentationFields.NAME);
+    }
+
+    public void setName(final String name) {
+        setToscaPresentationValue(JsonPresentationFields.NAME, name);
+    }
+
     @Override
     public String getType() {
         return (String) getToscaPresentationValue(JsonPresentationFields.TYPE);
@@ -119,10 +130,36 @@ public class AttributeDataDefinition extends ToscaDataDefinition {
     }
 
     public SchemaDefinition getSchema() {
-        return null;
+        return (SchemaDefinition) getToscaPresentationValue(JsonPresentationFields.SCHEMA);
+    }
+
+    public void setSchema(final SchemaDefinition schema) {
+        setToscaPresentationValue(JsonPresentationFields.SCHEMA, schema);
     }
 
     public String getParentUniqueId() {
         return getOwnerId();
     }
+
+    public boolean isGetOutputAttribute() {
+        return this.getGetOutputValues() != null && !this.getGetOutputValues().isEmpty();
+    }
+
+    public boolean typeEquals(Object obj) {
+        if (this == obj) {
+            return true;
+        }
+        if (obj == null) {
+            return false;
+        }
+        if (getClass() != obj.getClass()) {
+            return false;
+        }
+        AttributeDataDefinition other = (AttributeDataDefinition) obj;
+        if (this.getType() == null) {
+            return other.getType() == null;
+        }
+        return false;
+    }
+
 }