org.opendaylight.sfc.sbrest.json.SfgExporter Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of sfc-sb-rest Show documentation
There is a newer version: 0.10.4
/*
 * Copyright (c) 2014, 2017 Cisco Systems, Inc. and others. All rights reserved.
 *
 * This program and the accompanying materials are made available under the
 * terms of the Eclipse Public License v1.0 which accompanies this distribution,
 * and is available at http://www.eclipse.org/legal/epl-v10.html
 */
package org.opendaylight.sfc.sbrest.json;

import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.fasterxml.jackson.databind.node.ObjectNode;
import java.util.Locale;
import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.sfg.rev150214.service.function.group.entry.SfcServiceFunction;
import org.opendaylight.yang.gen.v1.urn.cisco.params.xml.ns.yang.sfc.sfg.rev150214.service.function.groups.ServiceFunctionGroup;
import org.opendaylight.yangtools.yang.binding.DataObject;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

class SfgExporter extends AbstractExporter implements Exporter {

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

    public static final String SERVICE_FUNCTION_GROUP = "service-function-group";
    public static final String SERVICE_FUNCTION = "service-function";
    public static final String NAME = "name";
    public static final String KEY = "key";
    public static final String TYPE = "type";
    public static final String REST_URI = "rest-uri";
    public static final String ALGORITHM = "algorithm";
    public static final String IP_MGMT_ADDRESS = "ip-mgmt-address";

    public static final String SERVICE_FUNCTION_TYPE_PREFIX = "service-function-type:";

    @Override
    public String exportJson(DataObject dataObject) {

        String ret = null;
        if (dataObject instanceof ServiceFunctionGroup) {
            ServiceFunctionGroup sfg = (ServiceFunctionGroup) dataObject;
            ObjectNode sfgNode = mapper.createObjectNode();
            sfgNode.put(NAME, sfg.getName());
            sfgNode.put(IP_MGMT_ADDRESS, ExporterUtil.convertIpAddress(sfg.getIpMgmtAddress()));
            sfgNode.put(ALGORITHM, sfg.getAlgorithm());
            if (sfg.getRestUri() != null) {
                sfgNode.put(REST_URI, sfg.getRestUri().getValue());
            }
            if (sfg.getType() != null) {
                sfgNode.put(TYPE, SERVICE_FUNCTION_TYPE_PREFIX + sfg.getType().getValue()
                        .toLowerCase(Locale.getDefault()));
            }

            // this should be revamped
            if (sfg.getSfcServiceFunction() != null) {
                ArrayNode sfArray = mapper.createArrayNode();
                for (SfcServiceFunction entry : sfg.getSfcServiceFunction()) {
                    ObjectNode objectNode = mapper.createObjectNode();
                    objectNode.put(NAME, entry.getName().getValue());
                    sfArray.add(objectNode);
                }
                sfgNode.putArray(SERVICE_FUNCTION).addAll(sfArray);
            }
            ArrayNode sfgArray = mapper.createArrayNode();
            sfgArray.add(sfgNode);
            try {
                Object sfObject = mapper.treeToValue(sfgArray, Object.class);
                ret = mapper.writeValueAsString(sfObject);
                ret = "{\"" + SERVICE_FUNCTION_GROUP + "\":" + ret + "}";
                LOG.debug("Created Service Function Group JSON: {}", ret);
            } catch (JsonProcessingException e) {
                LOG.error("Error during creation of JSON for Service Function {}", sfg.getName());
            }

        } else {
            throw new IllegalArgumentException("Argument is not an instance of ServiceFunction");
        }
        return ret;
    }

    @Override
    public String exportJsonNameOnly(DataObject dataObject) {
        String ret;
        if (dataObject instanceof ServiceFunctionGroup) {
            ServiceFunctionGroup obj = (ServiceFunctionGroup) dataObject;

            ObjectNode node = mapper.createObjectNode();
            node.put(NAME, obj.getName());
            ArrayNode sfArray = mapper.createArrayNode();
            sfArray.add(node);
            ret = "{\"" + SERVICE_FUNCTION_GROUP + "\":" + sfArray.toString() + "}";
        } else {
            throw new IllegalArgumentException("Argument is not an instance of ServiceFunction");
        }
        return ret;
    }
}