org.openecomp.sdnc.dmaapclient.SdncFlatJsonDmaapConsumer.java Source code

Introduction

Here is the source code for org.openecomp.sdnc.dmaapclient.SdncFlatJsonDmaapConsumer.java

Source

/*-
 * ============LICENSE_START=======================================================
 * openECOMP : SDN-C
 * ================================================================================
 * 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.sdnc.dmaapclient;

import java.io.BufferedReader;
import java.io.DataOutputStream;
import java.io.File;
import java.io.FileReader;
import java.io.IOException;
import java.io.InputStreamReader;
import java.net.Authenticator;
import java.net.HttpURLConnection;
import java.net.PasswordAuthentication;
import java.net.URL;
import java.util.HashMap;
import java.util.Iterator;
import java.util.Map;

import javax.net.ssl.HostnameVerifier;
import javax.net.ssl.HttpsURLConnection;
import javax.net.ssl.SSLSession;

import org.apache.commons.codec.binary.Base64;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.ObjectMapper;
import com.fasterxml.jackson.databind.node.ObjectNode;

public class SdncFlatJsonDmaapConsumer extends SdncDmaapConsumer {

    private static final Logger LOG = LoggerFactory.getLogger(SdncFlatJsonDmaapConsumer.class);

    private static final String DMAAPLISTENERROOT = "DMAAPLISTENERROOT";
    private static final String SDNC_ENDPOINT = "SDNC.endpoint";

    @Override
    public void processMsg(String msg) throws InvalidMessageException {

        processMsg(msg, null);
    }

    public void processMsg(String msg, String mapDirName) throws InvalidMessageException {

        if (msg == null) {
            throw new InvalidMessageException("Null message");
        }

        ObjectMapper oMapper = new ObjectMapper();
        JsonNode instarRootNode = null;
        ObjectNode sdncRootNode = null;

        String instarMsgName = null;

        try {
            instarRootNode = oMapper.readTree(msg);
        } catch (Exception e) {
            throw new InvalidMessageException("Cannot parse json object", e);
        }

        Iterator<Map.Entry<String, JsonNode>> instarFields = instarRootNode.fields();

        while (instarFields.hasNext()) {
            Map.Entry<String, JsonNode> entry = instarFields.next();

            instarMsgName = entry.getKey();
            instarRootNode = entry.getValue();
            break;
        }

        Map<String, String> fieldMap = loadMap(instarMsgName, mapDirName);

        if (fieldMap == null) {
            throw new InvalidMessageException("Unable to process message - cannot load field mappings");
        }

        if (!fieldMap.containsKey(SDNC_ENDPOINT)) {
            throw new InvalidMessageException("No SDNC endpoint known for message " + instarMsgName);
        }

        String sdncEndpoint = fieldMap.get(SDNC_ENDPOINT);

        sdncRootNode = oMapper.createObjectNode();
        ObjectNode inputNode = oMapper.createObjectNode();

        for (String fromField : fieldMap.keySet()) {

            if (!SDNC_ENDPOINT.equals(fromField)) {
                JsonNode curNode = instarRootNode.get(fromField);
                if (curNode != null) {
                    String fromValue = curNode.textValue();

                    inputNode.put(fieldMap.get(fromField), fromValue);
                }
            }
        }
        sdncRootNode.put("input", inputNode);

        try {
            String rpcMsgbody = oMapper.writeValueAsString(sdncRootNode);
            String odlUrlBase = getProperty("sdnc.odl.url-base");
            String odlUser = getProperty("sdnc.odl.user");
            String odlPassword = getProperty("sdnc.odl.password");

            if ((odlUrlBase != null) && (odlUrlBase.length() > 0)) {
                SdncOdlConnection conn = SdncOdlConnection.newInstance(odlUrlBase + sdncEndpoint, odlUser,
                        odlPassword);

                conn.send("POST", "application/json", rpcMsgbody);
            } else {
                LOG.info("POST message body would be:\n" + rpcMsgbody);
            }
        } catch (Exception e) {

        }

    }

    private Map<String, String> loadMap(String msgType, String mapDirName) {
        Map<String, String> results = new HashMap<String, String>();

        if (mapDirName == null) {
            String rootdir = System.getenv(DMAAPLISTENERROOT);

            if ((rootdir == null) || (rootdir.length() == 0)) {
                rootdir = "/opt/app/dmaap-listener";
            }

            mapDirName = rootdir + "/lib";

        }

        String mapFilename = mapDirName + "/" + msgType + ".map";

        File mapFile = new File(mapFilename);

        if (!mapFile.canRead()) {
            LOG.error("Cannot read map file (" + mapFilename + ")");
            return (null);
        }

        try {
            BufferedReader mapReader = new BufferedReader(new FileReader(mapFile));

            String curLine = null;

            while ((curLine = mapReader.readLine()) != null) {
                curLine = curLine.trim();

                if ((curLine.length() > 0) && (!curLine.startsWith("#"))) {

                    if (curLine.contains("=>")) {
                        String[] entry = curLine.split("=>");
                        if (entry.length == 2) {
                            results.put(entry[0].trim(), entry[1].trim());
                        }
                    }
                }
            }
            mapReader.close();
        } catch (Exception e) {
            LOG.error("Caught exception reading map " + mapFilename, e);
            return (null);
        }

        return (results);
    }

}