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