pcep/ietf-stateful/src/main/java/org/opendaylight/protocol/pcep/ietf/initiated/InitiatedPCInitiateMessageParser.java

   1 /*
   2  * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
   3  *
   4  * This program and the accompanying materials are made available under the
   5  * terms of the Eclipse Public License v1.0 which accompanies this distribution,
   6  * and is available at http://www.eclipse.org/legal/epl-v10.html
   7  */
   8 package org.opendaylight.protocol.pcep.ietf.initiated;
   9
  10 import static com.google.common.base.Preconditions.checkArgument;
  11
  12 import io.netty.buffer.ByteBuf;
  13 import io.netty.buffer.Unpooled;
  14 import java.util.ArrayList;
  15 import java.util.List;
  16 import java.util.Queue;
  17 import org.opendaylight.protocol.pcep.spi.AbstractMessageParser;
  18 import org.opendaylight.protocol.pcep.spi.MessageUtil;
  19 import org.opendaylight.protocol.pcep.spi.ObjectRegistry;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.initiated.rev200720.Pcinitiate;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.initiated.rev200720.PcinitiateBuilder;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.initiated.rev200720.pcinitiate.message.PcinitiateMessageBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.initiated.rev200720.pcinitiate.message.pcinitiate.message.Requests;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.initiated.rev200720.pcinitiate.message.pcinitiate.message.RequestsBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.lsp.object.Lsp;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev200720.srp.object.Srp;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.Message;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.Object;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.bandwidth.object.Bandwidth;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.endpoints.object.EndpointsObj;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.explicit.route.object.Ero;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.include.route.object.Iro;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.lsp.attributes.Metrics;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.lsp.attributes.MetricsBuilder;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.lspa.object.Lspa;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.metric.object.Metric;
  37
  38 /**
  39  * Parser for {@link Pcinitiate}.
  40  */
  41 public class InitiatedPCInitiateMessageParser extends AbstractMessageParser {
  42
  43     public static final int TYPE = 12;
  44
  45     public InitiatedPCInitiateMessageParser(final ObjectRegistry registry) {
  46         super(registry);
  47     }
  48
  49     @Override
  50     public void serializeMessage(final Message message, final ByteBuf out) {
  51         checkArgument(message instanceof Pcinitiate,
  52             "Wrong instance of Message. Passed instance of %s. Need PcinitiateMessage.", message.getClass());
  53         final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.initiated.rev200720
  54             .pcinitiate.message.PcinitiateMessage init = ((Pcinitiate) message).getPcinitiateMessage();
  55         final ByteBuf buffer = Unpooled.buffer();
  56         for (final Requests req : init.getRequests()) {
  57             serializeRequest(req, buffer);
  58         }
  59         MessageUtil.formatMessage(TYPE, buffer, out);
  60     }
  61
  62     protected void serializeRequest(final Requests req, final ByteBuf buffer) {
  63         serializeObject(req.getSrp(), buffer);
  64         serializeObject(req.getLsp(), buffer);
  65         serializeObject(req.getEndpointsObj(), buffer);
  66         serializeObject(req.getEro(), buffer);
  67         serializeObject(req.getLspa(), buffer);
  68         serializeObject(req.getBandwidth(), buffer);
  69         for (final Metrics m : req.nonnullMetrics()) {
  70             serializeObject(m.getMetric(), buffer);
  71         }
  72         serializeObject(req.getIro(), buffer);
  73     }
  74
  75     @Override
  76     protected Message validate(final Queue<Object> objects, final List<Message> errors) {
  77         checkArgument(objects != null, "Passed list can't be null.");
  78         final PcinitiateMessageBuilder builder = new PcinitiateMessageBuilder();
  79         final List<Requests> reqs = new ArrayList<>();
  80         while (!objects.isEmpty()) {
  81             reqs.add(this.getValidRequest(objects));
  82         }
  83         builder.setRequests(reqs);
  84         return new PcinitiateBuilder().setPcinitiateMessage(builder.build()).build();
  85     }
  86
  87     protected Requests getValidRequest(final Queue<Object> objects) {
  88         final RequestsBuilder builder = new RequestsBuilder()
  89             .setSrp((Srp) objects.remove())
  90             .setLsp((Lsp) objects.remove());
  91
  92         final List<Metrics> metrics = new ArrayList<>();
  93         State state = State.INIT;
  94         for (Object obj = objects.peek(); obj != null; obj = objects.peek()) {
  95             state = insertObject(state, obj, builder, metrics);
  96             if (state == State.END) {
  97                 break;
  98             }
  99
 100             objects.remove();
 101         }
 102
 103         builder.setMetrics(metrics);
 104         return builder.build();
 105     }
 106
 107     private static State insertObject(final State state, final Object obj, final RequestsBuilder builder,
 108             final List<Metrics> metrics) {
 109         switch (state) {
 110             case INIT:
 111                 if (obj instanceof EndpointsObj) {
 112                     builder.setEndpointsObj((EndpointsObj) obj);
 113                     return State.ENDPOINTS_IN;
 114                 }
 115                 // fall through
 116             case ENDPOINTS_IN:
 117                 if (obj instanceof Ero) {
 118                     builder.setEro((Ero) obj);
 119                     return State.ERO_IN;
 120                 }
 121                 // fall through
 122             case ERO_IN:
 123                 if (obj instanceof Lspa) {
 124                     builder.setLspa((Lspa) obj);
 125                     return State.LSPA_IN;
 126                 }
 127                 // fall through
 128             case LSPA_IN:
 129                 if (obj instanceof Bandwidth) {
 130                     builder.setBandwidth((Bandwidth) obj);
 131                     return State.BANDWIDTH_IN;
 132                 }
 133                 // fall through
 134             case BANDWIDTH_IN:
 135                 if (obj instanceof Metric) {
 136                     metrics.add(new MetricsBuilder().setMetric((Metric) obj).build());
 137                     return State.BANDWIDTH_IN;
 138                 }
 139                 // fall through
 140             case METRIC_IN:
 141                 if (obj instanceof Iro) {
 142                     builder.setIro((Iro) obj);
 143                     return State.IRO_IN;
 144                 }
 145                 // fall through
 146             case IRO_IN:
 147             case END:
 148                 return State.END;
 149             default:
 150                 return state;
 151         }
 152     }
 153
 154     private enum State {
 155         INIT, ENDPOINTS_IN, ERO_IN, LSPA_IN, BANDWIDTH_IN, METRIC_IN, IRO_IN, END
 156     }
 157 }