pcep/ietf-stateful02/src/main/java/org/opendaylight/protocol/pcep/crabbe/initiated00/PcinitiateMessageParser.java

   1 /*
   2  * Copyright (c) 2014 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.crabbe.initiated00;
   9
  10 import io.netty.buffer.ByteBuf;
  11
  12 import java.util.List;
  13
  14 import org.opendaylight.protocol.pcep.spi.AbstractMessageParser;
  15 import org.opendaylight.protocol.pcep.spi.ObjectHandlerRegistry;
  16 import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated._00.rev140113.Pcinitiate;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated._00.rev140113.PcinitiateBuilder;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated._00.rev140113.pcinitiate.message.PcinitiateMessageBuilder;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated._00.rev140113.pcinitiate.message.pcinitiate.message.Requests;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated._00.rev140113.pcinitiate.message.pcinitiate.message.RequestsBuilder;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.bandwidth.object.Bandwidth;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.endpoints.object.EndpointsObj;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.explicit.route.object.Ero;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.Metrics;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.MetricsBuilder;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lspa.object.Lspa;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.metric.object.Metric;
  31
  32 import com.google.common.collect.Lists;
  33
  34 /**
  35  * Parser for {@link Pcinitiate}
  36  */
  37 public class PcinitiateMessageParser extends AbstractMessageParser {
  38         public static final int TYPE = 12;
  39
  40         public PcinitiateMessageParser(final ObjectHandlerRegistry registry) {
  41                 super(registry);
  42         }
  43
  44         @Override
  45         public void serializeMessage(final Message message, final ByteBuf buffer) {
  46                 if (!(message instanceof Pcinitiate)) {
  47                         throw new IllegalArgumentException("Wrong instance of Message. Passed instance of " + message.getClass()
  48                                         + ". Needed PcinitiateMessage.");
  49                 }
  50                 final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated._00.rev140113.pcinitiate.message.PcinitiateMessage init = ((Pcinitiate) message).getPcinitiateMessage();
  51                 for (final Requests req : init.getRequests()) {
  52                         buffer.writeBytes(serializeObject(req.getEndpointsObj()));
  53                         buffer.writeBytes(serializeObject(req.getLspa()));
  54                         if (req.getEro() != null) {
  55                                 buffer.writeBytes(serializeObject(req.getEro()));
  56                         }
  57                         if (req.getBandwidth() != null) {
  58                                 buffer.writeBytes(serializeObject(req.getBandwidth()));
  59                         }
  60                         if (req.getMetrics() != null && !req.getMetrics().isEmpty()) {
  61                                 for (final Metrics m : req.getMetrics()) {
  62                                         buffer.writeBytes(serializeObject(m.getMetric()));
  63                                 }
  64                         }
  65                 }
  66         }
  67
  68         @Override
  69         protected Message validate(final List<Object> objects, final List<Message> errors) throws PCEPDeserializerException {
  70                 if (objects == null) {
  71                         throw new IllegalArgumentException("Passed list can't be null.");
  72                 }
  73                 final PcinitiateMessageBuilder builder = new PcinitiateMessageBuilder();
  74                 final List<Requests> reqs = Lists.newArrayList();
  75                 Requests req = null;
  76                 while (!objects.isEmpty()) {
  77                         req = this.getValidRequest(objects);
  78                         if (req == null) {
  79                                 break;
  80                         }
  81                         reqs.add(req);
  82                 }
  83                 builder.setRequests(reqs);
  84                 return new PcinitiateBuilder().setPcinitiateMessage(builder.build()).build();
  85         }
  86
  87         private Requests getValidRequest(final List<Object> objects) {
  88                 final RequestsBuilder builder = new RequestsBuilder();
  89                 builder.setEndpointsObj((EndpointsObj) objects.get(0));
  90                 objects.remove(0);
  91
  92                 builder.setLspa((Lspa) objects.get(0));
  93                 objects.remove(0);
  94                 final List<Metrics> metrics = Lists.newArrayList();
  95
  96                 Object obj;
  97                 State state = State.Init;
  98                 while (!objects.isEmpty() && !state.equals(State.End)) {
  99                         obj = objects.get(0);
 100
 101                         switch (state) {
 102                         case Init:
 103                                 state = State.EroIn;
 104                                 if (obj instanceof Ero) {
 105                                         builder.setEro((Ero) obj);
 106                                         break;
 107                                 }
 108                         case EroIn:
 109                                 state = State.BandwidthIn;
 110                                 if (obj instanceof Bandwidth) {
 111                                         builder.setBandwidth((Bandwidth) obj);
 112                                         break;
 113                                 }
 114                         case BandwidthIn:
 115                                 state = State.MetricIn;
 116                                 if (obj instanceof Metric) {
 117                                         metrics.add(new MetricsBuilder().setMetric((Metric) obj).build());
 118                                         state = State.BandwidthIn;
 119                                         break;
 120                                 }
 121                         case MetricIn:
 122                                 state = State.End;
 123                                 break;
 124                         case End:
 125                                 break;
 126                         }
 127                         if (!state.equals(State.End)) {
 128                                 objects.remove(0);
 129                         }
 130                 }
 131                 builder.setMetrics(metrics);
 132                 return builder.build();
 133         }
 134
 135         private enum State {
 136                 Init, EroIn, BandwidthIn, MetricIn, End
 137         }
 138
 139         @Override
 140         public int getMessageType() {
 141                 return TYPE;
 142         }
 143 }