2 files changed, 313 insertions, 0 deletions
diff --git a/src/main/java/org/onap/pomba/contextaggregator/datatypes/AggregatedModels.java b/src/main/java/org/onap/pomba/contextaggregator/datatypes/AggregatedModels.java
new file mode 100644
index 0000000..5648b91
--- /dev/null
+++ b/src/main/java/org/onap/pomba/contextaggregator/datatypes/AggregatedModels.java
@@ -0,0 +1,170 @@
+/*
+ * ============LICENSE_START===================================================
+ * Copyright (c) 2018 Amdocs
+ * ============================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *        http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=====================================================
+ */
+package org.onap.pomba.contextaggregator.datatypes;
+
+import java.util.HashMap;
+import java.util.Map;
+import java.util.Map.Entry;
+import java.util.UUID;
+import org.onap.pomba.common.datatypes.ModelContext;
+import org.onap.pomba.contextaggregator.config.EventHeaderConfig;
+import com.google.gson.Gson;
+import com.google.gson.GsonBuilder;
+import com.google.gson.annotations.Expose;
+import com.google.gson.annotations.SerializedName;
+
+public class AggregatedModels {
+
+    @Expose
+    @SerializedName("event-header")
+    Header entityHeader;
+    @Expose
+    @SerializedName("entity")
+    POAEntity poaEntity;
+
+
+    /**
+     * Creates an event with an entity header and entity containing the models and poa-event from Dmaap
+     *
+     * @param headerConfig
+     * @param jsonContextMap
+     */
+    public AggregatedModels(EventHeaderConfig headerConfig, Map<String, String> jsonContextMap, POAEvent event) {
+        entityHeader = new Header(headerConfig);
+
+        Gson gson = new GsonBuilder().create();
+        Map<String, ModelContext> contextMap = new HashMap<>();
+        for (Entry<String, String> entry : jsonContextMap.entrySet()) {
+            ModelContext context = null;
+            if (entry.getValue().isEmpty()) {
+                context = new ModelContext();
+                context.setVf(null);
+            } else {
+                context = gson.fromJson(entry.getValue(), ModelContext.class);
+            }
+            contextMap.put(entry.getKey(), context);
+        }
+
+        poaEntity = new POAEntity(contextMap, event);
+    }
+
+
+    /**
+     * Returns this instance as a JSON payload
+     *
+     * @return
+     */
+    public String generateJsonPayload() {
+        Gson gson = new GsonBuilder().create();
+        String payload = gson.toJson(this);
+        return payload;
+    }
+
+    public Header getEntityHeader() {
+        return entityHeader;
+    }
+
+
+
+    /**
+     * Entity header class for JSON serialization
+     */
+    private class Header {
+        @Expose
+        private String id;
+        @Expose
+        private String domain;
+        @Expose
+        @SerializedName("source-name")
+        private String sourceName;
+        @Expose
+        @SerializedName("event-type")
+        private String eventType;
+        @Expose
+        @SerializedName("entity-type")
+        private String entityType;
+        @Expose
+        @SerializedName("top-entity-type")
+        private String topEntityType;
+        @Expose
+        @SerializedName("topic-name")
+        private String topicName;
+        @Expose
+        @SerializedName("event-id")
+        private String eventId;
+
+        public Header(EventHeaderConfig config) {
+            id = UUID.randomUUID().toString();
+            domain = config.getDomain();
+            sourceName = config.getSourceName();
+            eventType = config.getEventType();
+            entityType = config.getEntityType();
+            topEntityType = config.getTopicEntityType();
+            topicName = config.getTopicName();
+            eventId = UUID.randomUUID().toString();
+        }
+
+
+        public String getId() {
+            return id;
+        }
+
+        public String getDomain() {
+            return domain;
+        }
+
+        public String getSourceName() {
+            return sourceName;
+        }
+
+        public String getEventType() {
+            return eventType;
+        }
+
+        public String getEntityType() {
+            return entityType;
+        }
+
+        public String getTopEntityType() {
+            return topEntityType;
+        }
+
+        public String getTopicName() {
+            return topicName;
+        }
+
+        public String getEventId() {
+            return eventId;
+        }
+    }
+
+
+    private class POAEntity {
+        @Expose
+        @SerializedName("poa-event")
+        POAEvent event;
+        @Expose
+        @SerializedName("context-list")
+        private Map<String, ModelContext> contextMap;
+
+        public POAEntity(Map<String, ModelContext> contextMap, POAEvent event) {
+            this.contextMap = contextMap;
+            this.event = event;
+        }
+    }
+}
diff --git a/src/main/java/org/onap/pomba/contextaggregator/datatypes/POAEvent.java b/src/main/java/org/onap/pomba/contextaggregator/datatypes/POAEvent.java
new file mode 100644
index 0000000..d54ece4
--- /dev/null
+++ b/src/main/java/org/onap/pomba/contextaggregator/datatypes/POAEvent.java
@@ -0,0 +1,143 @@
+/*
+ * ============LICENSE_START===================================================
+ * Copyright (c) 2018 Amdocs
+ * ============================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *        http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ * ============LICENSE_END=====================================================
+ */
+package org.onap.pomba.contextaggregator.datatypes;
+
+import org.onap.pomba.contextaggregator.exception.ContextAggregatorError;
+import org.onap.pomba.contextaggregator.exception.ContextAggregatorException;
+
+public class POAEvent {
+
+    private String serviceInstanceId;
+    private String modelVersionId;
+    private String modelInvariantId;
+    private String customerId;
+    private String serviceType;
+    private String xFromAppId;
+    private String xTransactionId;
+
+    public POAEvent() {}
+
+    public String getServiceInstanceId() {
+        return serviceInstanceId;
+    }
+
+    public void setServiceInstanceId(String serviceInstanceId) {
+        this.serviceInstanceId = serviceInstanceId;
+    }
+
+    public String getModelVersionId() {
+        return modelVersionId;
+    }
+
+    public void setModelVersionId(String modelVersionId) {
+        this.modelVersionId = modelVersionId;
+    }
+
+    public String getModelInvariantId() {
+        return modelInvariantId;
+    }
+
+    public void setModelInvariantId(String modelInvariantId) {
+        this.modelInvariantId = modelInvariantId;
+    }
+
+    public String getCustomerId() {
+        return customerId;
+    }
+
+    public void setCustomerId(String customerId) {
+        this.customerId = customerId;
+    }
+
+    public String getServiceType() {
+        return serviceType;
+    }
+
+    public void setServiceType(String serviceType) {
+        this.serviceType = serviceType;
+    }
+
+    public String getxFromAppId() {
+        return xFromAppId;
+    }
+
+    public void setxFromAppId(String xFromAppId) {
+        this.xFromAppId = xFromAppId;
+    }
+
+    public String getxTransactionId() {
+        return xTransactionId;
+    }
+
+    public void setxTransactionId(String xTransactionId) {
+        this.xTransactionId = xTransactionId;
+    }
+
+    public boolean validate() throws ContextAggregatorException {
+        final String missing = " is missing";
+
+        // serviceInstanceId
+        if (getServiceInstanceId() == null || getServiceInstanceId().isEmpty()) {
+            throw new ContextAggregatorException(ContextAggregatorError.INVALID_EVENT_RECEIVED,
+                    "serviceInstanceId" + missing);
+        }
+
+        // modelVersionId
+        if (getModelVersionId() == null || getModelVersionId().isEmpty()) {
+            throw new ContextAggregatorException(ContextAggregatorError.INVALID_EVENT_RECEIVED,
+                    "modelVersionId" + missing);
+        }
+
+        // modelInvariantId
+        if (getModelInvariantId() == null || getModelInvariantId().isEmpty()) {
+            throw new ContextAggregatorException(ContextAggregatorError.INVALID_EVENT_RECEIVED,
+                    "modelInvariantId" + missing);
+        }
+
+        // customerId
+        if (getCustomerId() == null || getCustomerId().isEmpty()) {
+            throw new ContextAggregatorException(ContextAggregatorError.INVALID_EVENT_RECEIVED, "customerId" + missing);
+        }
+
+        // serviceType
+        if (getServiceType() == null || getServiceType().isEmpty()) {
+            throw new ContextAggregatorException(ContextAggregatorError.INVALID_EVENT_RECEIVED,
+                    "serviceType" + missing);
+        }
+
+        // X-FromAppId
+        if (getxFromAppId() == null || getxFromAppId().isEmpty()) {
+            throw new ContextAggregatorException(ContextAggregatorError.INVALID_EVENT_RECEIVED, "xFromAppId" + missing);
+        }
+
+        // X-TransactionId
+        if (getxTransactionId() == null || getxTransactionId().isEmpty()) {
+            throw new ContextAggregatorException(ContextAggregatorError.INVALID_EVENT_RECEIVED,
+                    "xTransactionId" + missing);
+        }
+
+        return true;
+    }
+
+    @Override
+    public String toString() {
+        return "POAEvent [serviceInstanceId=" + serviceInstanceId + ", modelVersionId=" + modelVersionId
+                + ", modelInvariantId=" + modelInvariantId + ", customerId=" + customerId + ", serviceType="
+                + serviceType + ", xFromAppId=" + xFromAppId + ", xTransactionId=" + xTransactionId + "]";
+    }
+}