src/main/java/org.onosproject.xran/rest/LinkWebResource.java

/*
 * Copyright 2016-present Open Networking Laboratory
 *
 * 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.xran.rest;

import com.fasterxml.jackson.databind.JsonNode;
import com.fasterxml.jackson.databind.node.ObjectNode;
import com.google.common.collect.Lists;
import org.apache.commons.lang.exception.ExceptionUtils;
import org.onosproject.rest.AbstractWebResource;
import org.onosproject.xran.XranStore;
import org.onosproject.xran.annotations.Patch;
import org.onosproject.xran.controller.XranController;
import org.onosproject.xran.entities.RnibLink;
import org.openmuc.jasn1.ber.types.BerInteger;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import javax.ws.rs.Consumes;
import javax.ws.rs.DefaultValue;
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.QueryParam;
import javax.ws.rs.core.MediaType;
import javax.ws.rs.core.Response;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.Optional;
import java.util.concurrent.SynchronousQueue;

/**
 * Link web resource.
 */
@Path("links")
public class LinkWebResource extends AbstractWebResource {

    private static final Logger log =
            LoggerFactory.getLogger(LinkWebResource.class);

    /**
     * test.
     *
     * @param eciHex test
     * @param ue     test
     * @return test
     */
    @GET
    @Produces(MediaType.APPLICATION_JSON)
    public Response getLinksBetween(@DefaultValue("") @QueryParam("cell") String eciHex,
                             @DefaultValue("-1") @QueryParam("ue") long ue) {
        log.debug("GET LINKS CELL {} AND UE {}", eciHex, ue);

        List<RnibLink> list = Lists.newArrayList();
        if (!eciHex.isEmpty() && ue != -1) {
            RnibLink link = get(XranStore.class).getLinkBetweenCellIdUeId(eciHex, ue);
            if (link != null) {
                list.add(link);
            }
        } else if (!eciHex.isEmpty()) {
            list.addAll(get(XranStore.class).getLinksByCellId(eciHex));
        } else if (ue != -1) {
            list.addAll(get(XranStore.class).getLinksByUeId(ue));
        } else {
            list.addAll(get(XranStore.class).getLinks());
        }

        ObjectNode rootNode = mapper().createObjectNode();

        try {
            JsonNode jsonNode = mapper().readTree(list.toString());
            rootNode.put("links", jsonNode);
        } catch (IOException e) {
            log.error(ExceptionUtils.getFullStackTrace(e));
            e.printStackTrace();
        }

        return ok(rootNode.toString()).build();
    }

    /**
     * test.
     *
     * @param src    test
     * @param dst    test
     * @param stream test
     * @return test
     */
    @Patch
    @Path("{src},{dst}")
    @Consumes(MediaType.APPLICATION_JSON)
    public Response patchLinks(@PathParam("src") String src, @PathParam("dst") long dst, InputStream stream) {
        log.debug("Patch LINKS FROM {} to {}", src, dst);

        try {
            RnibLink link = get(XranStore.class).getLinkBetweenCellIdUeId(src, dst);
            if (link != null) {

                ObjectNode jsonTree = (ObjectNode) mapper().readTree(stream);

                JsonNode type = jsonTree.get("type");
                if (type != null) {
                    final SynchronousQueue<String>[] queue = new SynchronousQueue[1];
                    RnibLink.Type linkType = RnibLink.Type.getEnum(type.asText());
                    if (linkType.equals(RnibLink.Type.SERVING_PRIMARY)) {
                        List<RnibLink> linksByUeId = get(XranStore.class).getLinksByUeId(dst);
                        Optional<RnibLink> primary = linksByUeId.stream()
                                .filter(l -> l.getType().equals(RnibLink.Type.SERVING_PRIMARY))
                                .findFirst();
                        if (primary.isPresent()) {
                            queue[0] = get(XranController.class).sendHORequest(link, primary.get());
                            return Response.ok().entity(queue[0].take()).build();
                        }
                    }
                }

                JsonNode trafficpercent = jsonTree.get("trafficpercent");
                if (trafficpercent != null) {
                    JsonNode jsonNode = trafficpercent.get("traffic-percent-dl");
                    if (jsonNode != null) {
                        link.getTrafficPercent().setTrafficPercentDl(new BerInteger(jsonNode.asInt()));
                    }
                    jsonNode = trafficpercent.get("traffic-percent-ul");
                    if (jsonNode != null) {
                        link.getTrafficPercent().setTrafficPercentUl(new BerInteger(jsonNode.asInt()));
                    }
                    return Response.ok().build();
                }
            }
        } catch (Exception e) {
            log.error(ExceptionUtils.getFullStackTrace(e));
            e.printStackTrace();
            return Response.serverError().entity(ExceptionUtils.getFullStackTrace(e)).build();
        }

        return Response.noContent().build();
    }

    /**
     * test.
     *
     * @param src    test
     * @param dst    test
     * @param stream test
     * @return test
     */
    @POST
    @Path("{src},{dst}")
    @Consumes(MediaType.APPLICATION_JSON)
    public Response postLinks(@PathParam("src") String src, @PathParam("dst") long dst, InputStream stream) {
        log.debug("POST LINKS FROM {} to {}", src, dst);

        boolean b = false;
        try {
            ObjectNode jsonTree = (ObjectNode) mapper().readTree(stream);

            JsonNode type = jsonTree.get("type");

            if (type != null) {
                b = get(XranStore.class).createLinkBetweenCellIdUeId(src, dst, type.asText());
            }
        } catch (Exception e) {
            log.error(ExceptionUtils.getFullStackTrace(e));
            e.printStackTrace();
        }

        return ok(b).build();
    }

}