apps/openstacktelemetry/app/src/main/java/org/onosproject/openstacktelemetry/web/OpenstackTelemetryWebResource.java

/*
 * Copyright 2018-present Open Networking Foundation
 *
 * 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.
 */
package org.onosproject.openstacktelemetry.web;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ArrayNode;
import com.fasterxml.jackson.databind.node.ObjectNode;
import com.google.common.collect.Sets;
import org.apache.commons.lang3.exception.ExceptionUtils;
import org.onosproject.codec.JsonCodec;
import org.onosproject.openstacktelemetry.api.FlowInfo;
import org.onosproject.openstacktelemetry.api.StatsFlowRule;
import org.onosproject.openstacktelemetry.api.StatsFlowRuleAdminService;
import org.onosproject.openstacktelemetry.codec.FlowInfoJsonCodec;
import org.onosproject.rest.AbstractWebResource;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.ws.rs.Consumes;
import javax.ws.rs.DELETE;
import javax.ws.rs.GET;
import javax.ws.rs.POST;
import javax.ws.rs.Path;
import javax.ws.rs.PathParam;
import javax.ws.rs.Produces;
import javax.ws.rs.core.Context;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import javax.ws.rs.core.UriBuilder;
import javax.ws.rs.core.UriInfo;
import java.io.InputStream;
import java.util.Set;

import static com.fasterxml.jackson.databind.SerializationFeature.INDENT_OUTPUT;
import static javax.ws.rs.core.Response.created;
import static org.onlab.util.Tools.readTreeFromStream;

/**
 * Handles REST API call of openstack telemetry.
 */

@Path("telemetry")
public class OpenstackTelemetryWebResource extends AbstractWebResource {

    private final Logger log = LoggerFactory.getLogger(getClass());

    private final ObjectNode root = mapper().createObjectNode();

    private static final String JSON_NODE_FLOW_RULE = "rules";
    private static final String FLOW_RULE_ID = "STATS_FLOW_RULE_ID";

    private final StatsFlowRuleAdminService
                    statsFlowRuleService = get(StatsFlowRuleAdminService.class);

    @Context
    private UriInfo uriInfo;

    /**
     * Creates a flow rule for metric.
     *
     * @param input openstack flow rule JSON input stream
     * @return 201 CREATED if the JSON is correct,
     *         400 BAD_REQUEST if the JSON is malformed.
     */
    @POST
    @Consumes(MediaType.APPLICATION_JSON)
    @Produces(MediaType.APPLICATION_JSON)
    public Response createBulkFlowRule(InputStream input) {
        log.info("CREATE BULK FLOW RULE: {}", input.toString());

        readNodeConfiguration(input).forEach(flowRule -> {
                log.debug("FlowRule: {}", flowRule.toString());
                statsFlowRuleService.createStatFlowRule(flowRule);
            });

        UriBuilder locationBuilder = uriInfo.getBaseUriBuilder()
                                            .path(JSON_NODE_FLOW_RULE)
                                            .path(FLOW_RULE_ID);

        return created(locationBuilder.build()).build();
    }

    /**
     * Delete flow rules.
     *
     * @param input openstack flow rule JSON input stream
     * @return 200 OK if processing is correct.
     */
    @DELETE
    @Consumes(MediaType.APPLICATION_JSON)
    @Produces(MediaType.APPLICATION_JSON)
    public Response deleteBulkFlowRule(InputStream input) {
        log.info("DELETE BULK FLOW RULE: {}", input.toString());

        readNodeConfiguration(input).forEach(flowRule -> {
            log.debug("FlowRule: {}", flowRule.toString());
            statsFlowRuleService.deleteStatFlowRule(flowRule);
        });

        return ok(root).build();
    }

    /**
     * Get flow rules which is installed on ONOS.
     *
     * @return 200 OK
     */
    public Response readBulkFlowRule() {
        log.info("READ BULK FLOW RULE");

        return ok(root).build();
    }

    /**
     * Get flow information list.
     *
     * @return Flow information list
     */
    @GET
    @Path("list")
    @Produces(MediaType.APPLICATION_JSON)
    public Response getFlowInfoBulk() {
        log.info("GET BULK FLOW RULE");

        Set<FlowInfo> flowInfoSet;
        flowInfoSet = statsFlowRuleService.getOverlayFlowInfos();

        log.info("\n\n======================================================\n" +
                 "FlowInfo Set: \n{}" +
                 "\n\n======================================================\n",
                 flowInfoSet);

        JsonCodec<FlowInfo> flowInfoCodec = new FlowInfoJsonCodec();

        ObjectNode nodeJson;
        int idx = 0;
        for (FlowInfo flowInfo: flowInfoSet) {
            nodeJson = flowInfoCodec.encode(flowInfo, this);
            root.put("FlowInfo" + String.valueOf(idx++), nodeJson.toString());
        }
        return ok(root).build();
    }

    @GET
    @Path("list/{srcIpPrefix}/{dstIpPrefix}")
    @Produces(MediaType.APPLICATION_JSON)
    public Response getFlowRule(@PathParam("srcIpPrefix") String srcIpPrefix,
                                @PathParam("dstIpPrefix") String dstIpPrefix) {
        return ok(root).build();
    }

    private Set<StatsFlowRule> readNodeConfiguration(InputStream input) {
        log.info("Input JSON Data: \n\t\t{}", input.toString());
        Set<StatsFlowRule> flowRuleSet = Sets.newHashSet();
        try {
            JsonNode jsonTree = readTreeFromStream(mapper().enable(INDENT_OUTPUT), input);
            ArrayNode nodes = (ArrayNode) jsonTree.path(JSON_NODE_FLOW_RULE);
            nodes.forEach(node -> {
                try {
                    ObjectNode objectNode = node.deepCopy();
                    log.debug("ObjectNode: {}", objectNode.toString());
                    StatsFlowRule statsFlowRule = codec(StatsFlowRule.class)
                                                        .decode(objectNode, this);
                    log.debug("StatsFlowRule: {}", statsFlowRule.toString());
                    flowRuleSet.add(statsFlowRule);
                } catch (Exception ex) {
                    log.error("Exception Stack:\n{}", ExceptionUtils.getStackTrace(ex));
                    throw new IllegalArgumentException();
                }
            });
        } catch (Exception ex) {
            log.error("Exception Stack:\n{}", ExceptionUtils.getStackTrace(ex));
            throw new IllegalArgumentException(ex);
        }

        return flowRuleSet;
    }
}