1 files changed, 87 insertions, 0 deletions
diff --git a/openecomp-be/lib/openecomp-core-lib/openecomp-utilities-lib/src/main/java/org/openecomp/core/utilities/deserializers/RequirementDefinitionDeserializer.java b/openecomp-be/lib/openecomp-core-lib/openecomp-utilities-lib/src/main/java/org/openecomp/core/utilities/deserializers/RequirementDefinitionDeserializer.java
new file mode 100644
index 0000000000..b341a52c56
--- /dev/null
+++ b/openecomp-be/lib/openecomp-core-lib/openecomp-utilities-lib/src/main/java/org/openecomp/core/utilities/deserializers/RequirementDefinitionDeserializer.java
@@ -0,0 +1,87 @@
+package org.openecomp.core.utilities.deserializers;
+
+import com.google.gson.JsonArray;
+import com.google.gson.JsonDeserializationContext;
+import com.google.gson.JsonDeserializer;
+import com.google.gson.JsonElement;
+import com.google.gson.JsonObject;
+import com.google.gson.JsonParseException;
+import com.google.gson.JsonPrimitive;
+import org.openecomp.sdc.tosca.datatypes.model.RequirementDefinition;
+
+import java.lang.reflect.Type;
+import java.util.Objects;
+import java.util.Optional;
+
+public class RequirementDefinitionDeserializer implements JsonDeserializer<RequirementDefinition> {
+
+  private static final String CAPABILITY = "capability";
+  private static final String NODE = "node";
+  private static final String RELATIONSHIP = "relationship";
+  private static final String OCCURRENCES = "occurrences";
+
+  @Override
+  public RequirementDefinition deserialize(JsonElement jsonElement, Type type,
+                                           JsonDeserializationContext jsonDeserializationContext)
+      throws JsonParseException {
+
+    JsonObject jsonObject = jsonElement.getAsJsonObject();
+
+    RequirementDefinition requirementDefinition = new RequirementDefinition();
+    setRequirementValues(jsonObject, requirementDefinition);
+
+    Optional<Object[]> occurrences = handleOccurrences(jsonObject);
+    occurrences.ifPresent(requirementDefinition::setOccurrences);
+
+    return requirementDefinition;
+  }
+
+
+  private void setRequirementValues(JsonObject jsonObject,
+                                    RequirementDefinition requirementDefinition) {
+
+    JsonElement capabilityElement = jsonObject.get(CAPABILITY);
+    if (!Objects.isNull(capabilityElement)) {
+      requirementDefinition.setCapability(capabilityElement.getAsString());
+    }
+
+    JsonElement nodeElement = jsonObject.get(NODE);
+    if (!Objects.isNull(nodeElement)) {
+      requirementDefinition.setNode(nodeElement.getAsString());
+    }
+
+    JsonElement relationshipElement = jsonObject.get(RELATIONSHIP);
+    if (!Objects.isNull(relationshipElement)) {
+      requirementDefinition.setRelationship(relationshipElement.getAsString());
+    }
+  }
+
+  private Optional<Object[]> handleOccurrences(JsonObject jsonObject) {
+
+    JsonElement occurrencesElement = jsonObject.get(OCCURRENCES);
+
+    if(Objects.isNull(occurrencesElement)){
+      return Optional.empty();
+    }
+
+    JsonArray occurrences = occurrencesElement.getAsJsonArray();
+    Object[] fixedOccurrences = new Object[occurrences.size()];
+
+    for (int i = 0; i < occurrences.size(); i++) {
+      JsonElement currElement = occurrences.get(i);
+
+      // values inside occurrences array can be either String or Integer
+      if (currElement.isJsonPrimitive()) {
+        JsonPrimitive jsonPrimitive = currElement.getAsJsonPrimitive();
+
+        if (jsonPrimitive.isNumber()) {
+          fixedOccurrences[i] = jsonPrimitive.getAsNumber().intValue();
+        } else {
+          fixedOccurrences[i] = jsonPrimitive.getAsString();
+        }
+      }
+    }
+
+    return Optional.of(fixedOccurrences);
+  }
+}