| /* |
| * Copyright 2015 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.codec.impl; |
| |
| import org.onosproject.codec.CodecContext; |
| import org.onosproject.codec.JsonCodec; |
| import org.onosproject.net.DeviceId; |
| import org.onosproject.net.Link; |
| import org.onosproject.net.intent.Constraint; |
| import org.onosproject.net.intent.constraint.AnnotationConstraint; |
| import org.onosproject.net.intent.constraint.BandwidthConstraint; |
| import org.onosproject.net.intent.constraint.LambdaConstraint; |
| import org.onosproject.net.intent.constraint.LatencyConstraint; |
| import org.onosproject.net.intent.constraint.LinkTypeConstraint; |
| import org.onosproject.net.intent.constraint.ObstacleConstraint; |
| import org.onosproject.net.intent.constraint.WaypointConstraint; |
| |
| import com.fasterxml.jackson.databind.node.ArrayNode; |
| import com.fasterxml.jackson.databind.node.ObjectNode; |
| |
| import static com.google.common.base.Preconditions.checkNotNull; |
| |
| /** |
| * Constraint JSON codec. |
| */ |
| public final class ConstraintCodec extends JsonCodec<Constraint> { |
| |
| /** |
| * Encodes a latency constraint. |
| * |
| * @param constraint latency constraint to encode |
| * @param context code context |
| * @return JSON ObjectNode representing the constraint |
| */ |
| private ObjectNode encodeLatencyConstraint(Constraint constraint, |
| CodecContext context) { |
| checkNotNull(constraint, "Duration constraint cannot be null"); |
| final LatencyConstraint latencyConstraint = |
| (LatencyConstraint) constraint; |
| return context.mapper().createObjectNode() |
| .put("latencyMillis", latencyConstraint.latency().toMillis()); |
| } |
| |
| /** |
| * Encodes an obstacle constraint. |
| * |
| * @param constraint obstacle constraint to encode |
| * @param context code context |
| * @return JSON ObjectNode representing the constraint |
| */ |
| private ObjectNode encodeObstacleConstraint(Constraint constraint, |
| CodecContext context) { |
| checkNotNull(constraint, "Obstacle constraint cannot be null"); |
| final ObstacleConstraint obstacleConstraint = |
| (ObstacleConstraint) constraint; |
| |
| final ObjectNode result = context.mapper().createObjectNode(); |
| final ArrayNode jsonObstacles = result.putArray("obstacles"); |
| |
| for (DeviceId did : obstacleConstraint.obstacles()) { |
| jsonObstacles.add(did.toString()); |
| } |
| |
| return result; |
| } |
| |
| /** |
| * Encodes a waypoint constraint. |
| * |
| * @param constraint waypoint constraint to encode |
| * @param context code context |
| * @return JSON ObjectNode representing the constraint |
| */ |
| private ObjectNode encodeWaypointConstraint(Constraint constraint, |
| CodecContext context) { |
| checkNotNull(constraint, "Waypoint constraint cannot be null"); |
| final WaypointConstraint waypointConstraint = |
| (WaypointConstraint) constraint; |
| |
| final ObjectNode result = context.mapper().createObjectNode(); |
| final ArrayNode jsonWaypoints = result.putArray("waypoints"); |
| |
| for (DeviceId did : waypointConstraint.waypoints()) { |
| jsonWaypoints.add(did.toString()); |
| } |
| |
| return result; |
| } |
| |
| /** |
| * Encodes a annotation constraint. |
| * |
| * @param constraint annotation constraint to encode |
| * @param context code context |
| * @return JSON ObjectNode representing the constraint |
| */ |
| private ObjectNode encodeAnnotationConstraint(Constraint constraint, |
| CodecContext context) { |
| checkNotNull(constraint, "Annotation constraint cannot be null"); |
| final AnnotationConstraint annotationConstraint = |
| (AnnotationConstraint) constraint; |
| return context.mapper().createObjectNode() |
| .put("key", annotationConstraint.key()) |
| .put("threshold", annotationConstraint.threshold()); |
| } |
| |
| /** |
| * Encodes a bandwidth constraint. |
| * |
| * @param constraint bandwidth constraint to encode |
| * @param context code context |
| * @return JSON ObjectNode representing the constraint |
| */ |
| private ObjectNode encodeBandwidthConstraint(Constraint constraint, |
| CodecContext context) { |
| checkNotNull(constraint, "Bandwidth constraint cannot be null"); |
| final BandwidthConstraint bandwidthConstraint = |
| (BandwidthConstraint) constraint; |
| return context.mapper().createObjectNode() |
| .put("bandwidth", bandwidthConstraint.bandwidth().toDouble()); |
| } |
| |
| /** |
| * Encodes a lambda constraint. |
| * |
| * @param constraint lambda constraint to encode |
| * @param context code context |
| * @return JSON ObjectNode representing the constraint |
| */ |
| private ObjectNode encodeLambdaConstraint(Constraint constraint, |
| CodecContext context) { |
| checkNotNull(constraint, "Lambda constraint cannot be null"); |
| final LambdaConstraint lambdaConstraint = |
| (LambdaConstraint) constraint; |
| |
| return context.mapper().createObjectNode() |
| .put("lambda", lambdaConstraint.lambda().toInt()); |
| } |
| |
| /** |
| * Encodes a link type constraint. |
| * |
| * @param constraint link type constraint to encode |
| * @param context code context |
| * @return JSON ObjectNode representing the constraint |
| */ |
| private ObjectNode encodeLinkTypeConstraint(Constraint constraint, |
| CodecContext context) { |
| checkNotNull(constraint, "Link type constraint cannot be null"); |
| |
| final LinkTypeConstraint linkTypeConstraint = |
| (LinkTypeConstraint) constraint; |
| |
| final ObjectNode result = context.mapper().createObjectNode() |
| .put("inclusive", linkTypeConstraint.isInclusive()); |
| |
| final ArrayNode jsonTypes = result.putArray("types"); |
| |
| if (linkTypeConstraint.types() != null) { |
| for (Link.Type type : linkTypeConstraint.types()) { |
| jsonTypes.add(type.name()); |
| } |
| } |
| |
| return result; |
| } |
| |
| @Override |
| public ObjectNode encode(Constraint constraint, CodecContext context) { |
| checkNotNull(constraint, "Constraint cannot be null"); |
| |
| final ObjectNode result; |
| if (constraint instanceof BandwidthConstraint) { |
| result = encodeBandwidthConstraint(constraint, context); |
| } else if (constraint instanceof LambdaConstraint) { |
| result = encodeLambdaConstraint(constraint, context); |
| } else if (constraint instanceof LinkTypeConstraint) { |
| result = encodeLinkTypeConstraint(constraint, context); |
| } else if (constraint instanceof AnnotationConstraint) { |
| result = encodeAnnotationConstraint(constraint, context); |
| } else if (constraint instanceof LatencyConstraint) { |
| result = encodeLatencyConstraint(constraint, context); |
| } else if (constraint instanceof ObstacleConstraint) { |
| result = encodeObstacleConstraint(constraint, context); |
| } else if (constraint instanceof WaypointConstraint) { |
| result = encodeWaypointConstraint(constraint, context); |
| } else { |
| result = context.mapper().createObjectNode(); |
| } |
| |
| result.put("type", constraint.getClass().getSimpleName()); |
| return result; |
| } |
| } |