Support querying of model by type

[sdc.git] / catalog-dao / src / main / java / org / openecomp / sdc / be / resources / data / ModelData.java

diff --git a/catalog-dao/src/main/java/org/openecomp/sdc/be/resources/data/ModelData.java b/catalog-dao/src/main/java/org/openecomp/sdc/be/resources/data/ModelData.java
index 026b1d7..5f7b15e 100644 (file)
--- a/catalog-dao/src/main/java/org/openecomp/sdc/be/resources/data/ModelData.java
+++ b/catalog-dao/src/main/java/org/openecomp/sdc/be/resources/data/ModelData.java
@@ -23,6 +23,7 @@ import java.util.Map;
 import lombok.Getter;
 import org.openecomp.sdc.be.dao.graph.datatype.GraphNode;
 import org.openecomp.sdc.be.dao.neo4j.GraphPropertiesDictionary;
+import org.openecomp.sdc.be.datatypes.enums.ModelTypeEnum;
 import org.openecomp.sdc.be.datatypes.enums.NodeTypeEnum;
 
 @Getter
@@ -30,17 +31,20 @@ public class ModelData extends GraphNode {
 
     private final String name;
     private final String uniqueId;
+    private final String modelType;
 
-    public ModelData(final String name, final String uniqueId) {
+    public ModelData(final String name, final String uniqueId, final ModelTypeEnum modelType) {
         super(NodeTypeEnum.Model);
         this.name = name;
         this.uniqueId = uniqueId;
+        this.modelType = modelType.getValue();
     }
 
     public ModelData(final Map<String, Object> properties) {
         super(NodeTypeEnum.Model);
         name = (String) properties.get(GraphPropertiesDictionary.NAME.getProperty());
         uniqueId = (String) properties.get(GraphPropertiesDictionary.UNIQUE_ID.getProperty());
+        modelType = (String) properties.get(GraphPropertiesDictionary.MODEL_TYPE.getProperty());
     }
 
     @Override
@@ -48,6 +52,7 @@ public class ModelData extends GraphNode {
         final Map<String, Object> map = new HashMap<>();
         addIfExists(map, GraphPropertiesDictionary.NAME, name);
         addIfExists(map, GraphPropertiesDictionary.UNIQUE_ID, uniqueId);
+        addIfExists(map, GraphPropertiesDictionary.MODEL_TYPE, modelType);
         return map;
     }