1 files changed, 211 insertions, 0 deletions
diff --git a/ajsc-aai/src/main/java/org/openecomp/aai/util/StoreNotificationEvent.java b/ajsc-aai/src/main/java/org/openecomp/aai/util/StoreNotificationEvent.java
new file mode 100644
index 0000000..89152ce
--- /dev/null
+++ b/ajsc-aai/src/main/java/org/openecomp/aai/util/StoreNotificationEvent.java
@@ -0,0 +1,211 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * org.openecomp.aai
+ * ================================================================================
+ * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * 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.openecomp.aai.util;
+
+import java.text.DateFormat;
+import java.text.SimpleDateFormat;
+import java.util.Date;
+import java.util.UUID;
+
+import org.apache.cxf.message.Exchange;
+import org.apache.cxf.message.Message;
+import org.apache.cxf.phase.PhaseInterceptorChain;
+import org.eclipse.persistence.dynamic.DynamicEntity;
+
+import org.eclipse.persistence.jaxb.dynamic.DynamicJAXBContext;
+import org.openecomp.aai.domain.notificationEvent.NotificationEvent;
+import org.openecomp.aai.exceptions.AAIException;
+
+public class StoreNotificationEvent {
+
+	/**
+	 * Instantiates a new store notification event.
+	 */
+	public StoreNotificationEvent() {}
+
+	/**
+	 * Store event.
+	 *
+	 * @param eh the eh
+	 * @param obj the obj
+	 * @throws AAIException the AAI exception
+	 */
+	public void storeEvent(NotificationEvent.EventHeader eh, Object obj) throws AAIException { 
+
+		if (obj == null) { 
+			throw new AAIException("AAI_7350");
+		}
+
+		org.openecomp.aai.domain.notificationEvent.ObjectFactory factory = new org.openecomp.aai.domain.notificationEvent.ObjectFactory();
+
+		org.openecomp.aai.domain.notificationEvent.NotificationEvent ne = factory.createNotificationEvent();
+
+		if (eh.getId() == null) { 
+			eh.setId(genDate2() + "-" + UUID.randomUUID().toString());
+		}
+		if (eh.getTimestamp() == null) { 
+			eh.setTimestamp(genDate());
+		}
+
+		// there's no default, but i think we want to put this in hbase?
+
+		if (eh.getEntityLink() == null) { 
+			eh.setEntityLink("UNK");
+		}
+
+		if (eh.getAction() == null) { 
+			eh.setAction("UNK");
+		}
+
+		if (eh.getEventType() == null) { 
+			eh.setEventType(AAIConfig.get("aai.notificationEvent.default.eventType", "UNK"));
+		}
+
+		if (eh.getDomain() == null) { 
+			eh.setDomain(AAIConfig.get("aai.notificationEvent.default.domain", "UNK"));
+		}
+
+		if (eh.getSourceName() == null) { 
+			eh.setSourceName(AAIConfig.get("aai.notificationEvent.default.sourceName", "UNK"));
+		}
+
+		if (eh.getSequenceNumber() == null) { 
+			eh.setSequenceNumber(AAIConfig.get("aai.notificationEvent.default.sequenceNumber", "UNK"));
+		}
+
+		if (eh.getSeverity() == null) { 
+			eh.setSeverity(AAIConfig.get("aai.notificationEvent.default.severity", "UNK"));
+		}
+
+		if (eh.getVersion() == null) { 
+			eh.setVersion(AAIConfig.get("aai.notificationEvent.default.version", "UNK"));
+		}
+
+		ne.setCambriaPartition(AAIConstants.UEB_PUB_PARTITION_AAI);
+		ne.setEventHeader(eh);
+		ne.setEntity(obj);
+
+		try {
+			Exchange message = PhaseInterceptorChain.getCurrentMessage().getExchange();
+			message.put("NOTIFICATION_EVENT", ne);
+		} catch (Exception e) {
+			throw new AAIException("AAI_7350", e);
+		}
+	}
+	
+	/**
+	 * Store dynamic event.
+	 *
+	 * @param notificationJaxbContext the notification jaxb context
+	 * @param notificationVersion the notification version
+	 * @param eventHeader the event header
+	 * @param obj the obj
+	 * @throws AAIException the AAI exception
+	 */
+	public void storeDynamicEvent(DynamicJAXBContext notificationJaxbContext, String notificationVersion, DynamicEntity eventHeader, DynamicEntity obj) throws AAIException { 
+
+		if (obj == null) { 
+			throw new AAIException("AAI_7350");
+		}
+
+		DynamicEntity notificationEvent = notificationJaxbContext.getDynamicType("inventory.aai.openecomp.org." + notificationVersion + ".NotificationEvent").newDynamicEntity();
+
+		if (eventHeader.get("id") == null) { 
+			eventHeader.set("id", genDate2() + "-" + UUID.randomUUID().toString());
+		}
+		if (eventHeader.get("timestamp") == null) { 
+			eventHeader.set("timestamp", genDate());
+		}
+
+		// there's no default, but i think we want to put this in hbase?
+
+		if (eventHeader.get("entityLink") == null) { 
+			eventHeader.set("entityLink", "UNK");
+		}
+
+		if (eventHeader.get("action") == null) { 
+			eventHeader.set("action", "UNK");
+		}
+
+		if (eventHeader.get("eventType") == null) { 
+			eventHeader.set("eventType", AAIConfig.get("aai.notificationEvent.default.eventType", "UNK"));
+		}
+
+		if (eventHeader.get("domain") == null) { 
+			eventHeader.set("domain", AAIConfig.get("aai.notificationEvent.default.domain", "UNK"));
+		}
+
+		if (eventHeader.get("sourceName") == null) { 
+			eventHeader.set("sourceName", AAIConfig.get("aai.notificationEvent.default.sourceName", "UNK"));
+		}
+
+		if (eventHeader.get("sequenceNumber") == null) { 
+			eventHeader.set("sequenceNumber", AAIConfig.get("aai.notificationEvent.default.sequenceNumber", "UNK"));
+		}
+
+		if (eventHeader.get("severity") == null) { 
+			eventHeader.set("severity", AAIConfig.get("aai.notificationEvent.default.severity", "UNK"));
+		}
+
+		if (eventHeader.get("version") == null) { 
+			eventHeader.set("version", AAIConfig.get("aai.notificationEvent.default.version", "UNK"));
+		}
+
+		if (notificationEvent.get("cambriaPartition") == null) {
+			notificationEvent.set("cambriaPartition", AAIConstants.UEB_PUB_PARTITION_AAI);
+		}
+
+		notificationEvent.set("eventHeader", eventHeader);
+		notificationEvent.set("entity", obj);
+
+		try {
+			Exchange message = PhaseInterceptorChain.getCurrentMessage().getExchange();
+			message.put("NOTIFICATION_EVENT", notificationEvent);
+			message.put("NOTIFICATION_EVENT_TYPE", "dynamic");
+			message.put("NOTIFICATION_JAXB_CONTEXT", notificationJaxbContext);
+		} catch (Exception e) {
+			throw new AAIException("AAI_7350", e);
+		}
+	}
+
+	/**
+	 * Gen date.
+	 *
+	 * @return the string
+	 */
+	public static String genDate() {
+		Date date = new Date();
+		DateFormat formatter = new SimpleDateFormat("YYYYMMdd-HH:mm:ss:SSS");
+		return formatter.format(date);
+	}
+	
+	/**
+	 * Gen date 2.
+	 *
+	 * @return the string
+	 */
+	public static String genDate2() {
+		Date date = new Date();
+		DateFormat formatter = new SimpleDateFormat("YYYYMMddHHmmss");
+		return formatter.format(date);
+	}
+
+}