1 files changed, 130 insertions, 0 deletions
diff --git a/dmaap-listener/src/main/java/org/onap/ccsdk/sli/northbound/dmaapclient/SdncDhcpEventConsumer.java b/dmaap-listener/src/main/java/org/onap/ccsdk/sli/northbound/dmaapclient/SdncDhcpEventConsumer.java
new file mode 100644
index 000000000..7b68ceb63
--- /dev/null
+++ b/dmaap-listener/src/main/java/org/onap/ccsdk/sli/northbound/dmaapclient/SdncDhcpEventConsumer.java
@@ -0,0 +1,130 @@
+package org.onap.ccsdk.sli.northbound.dmaapclient;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.FileNotFoundException;
+import java.io.IOException;
+import java.sql.SQLException;
+import java.util.Properties;
+import org.onap.ccsdk.sli.core.dblib.DBResourceManager;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+import com.fasterxml.jackson.databind.JsonNode;
+import com.fasterxml.jackson.databind.ObjectMapper;
+
+public class SdncDhcpEventConsumer extends SdncDmaapConsumerImpl {
+	private static final Logger LOG = LoggerFactory.getLogger(SdncDhcpEventConsumer.class);
+
+	private static final String MAC_ADDR_TAG = "macaddr";
+	private static final String MSG_NAME_TAG = "msg_name";
+	private static final String IP_ADDR_TAG = "yiaddr";
+
+	private static DBResourceManager jdbcDataSource = null;
+	private static final String SDNC_CONFIG_DIR = "SDNC_CONFIG_DIR";
+
+	private class MissingDhcpAttributeException extends InvalidMessageException {
+
+		public MissingDhcpAttributeException(String fieldName) {
+			super("Invalid DHCP event - missing " + fieldName + " attribute");
+		}
+	}
+
+	private static void setJdbcDataSource() throws IOException {
+
+		String propPath;
+		String propDir = System.getenv(SDNC_CONFIG_DIR);
+		if (propDir == null) {
+			propDir = "/opt/onap/sdnc/data/properties";
+		}
+		propPath = propDir + "/dblib.properties";
+		File propFile = new File(propPath);
+
+		if (!propFile.exists()) {
+
+			throw new FileNotFoundException("Missing configuration properties file : " + propFile);
+		}
+
+		Properties props = new Properties();
+		props.load(new FileInputStream(propFile));
+
+		setJdbcDataSource(new DBResourceManager(props));
+
+	}
+
+	static void setJdbcDataSource(DBResourceManager dbMgr) {
+
+		jdbcDataSource = dbMgr;
+
+		if (jdbcDataSource.isActive()) {
+			LOG.warn("DBLIB: JDBC DataSource has been initialized.");
+		} else {
+			LOG.warn("DBLIB: JDBC DataSource did not initialize successfully.");
+		}
+	}
+
+	@Override
+	public void processMsg(String msg) throws InvalidMessageException {
+		if (msg == null) {
+			throw new InvalidMessageException("Null message");
+		}
+
+		ObjectMapper oMapper = new ObjectMapper();
+
+		JsonNode dhcpRootNode;
+		String msgName;
+		String macAddr;
+		String ipAddr;
+
+		try {
+			dhcpRootNode = oMapper.readTree(msg);
+
+		} catch (IOException e) {
+			throw new InvalidMessageException("Cannot parse json object", e);
+		}
+
+		JsonNode msgNameNode = dhcpRootNode.get(MSG_NAME_TAG);
+		if (msgNameNode != null) {
+			msgName = msgNameNode.textValue();
+
+		} else {
+			throw new MissingDhcpAttributeException(MSG_NAME_TAG);
+		}
+
+		JsonNode macAddrNode = dhcpRootNode.get(MAC_ADDR_TAG);
+		if (macAddrNode != null) {
+			macAddr = macAddrNode.textValue();
+
+		} else {
+			throw new MissingDhcpAttributeException(MAC_ADDR_TAG);
+		}
+
+		JsonNode ipAddrNode = dhcpRootNode.get(IP_ADDR_TAG);
+		if (ipAddrNode != null) {
+			ipAddr = ipAddrNode.textValue();
+
+		} else {
+			throw new MissingDhcpAttributeException(IP_ADDR_TAG);
+		}
+
+		LOG.debug("Got DHCP event : msg name {}; mac addr {}; ip addr {}", msgName, macAddr, ipAddr);
+
+		if (jdbcDataSource == null) {
+			try {
+				setJdbcDataSource();
+			} catch (IOException e) {
+				LOG.error("Could not create JDBC connection", e);
+				return;
+			}
+		}
+
+		try {
+
+			jdbcDataSource.writeData("INSERT INTO DHCP_MAP(mac_addr, ip_addr) VALUES('" + macAddr + "','" + ipAddr + "') ON DUPLICATE KEY UPDATE ip_addr = '"+ipAddr+"'", null, null);
+
+		} catch (SQLException e) {
+			LOG.error("Could not insert DHCP event data into the database ", e);
+		}
+
+	}
+
+}