pcep/ietf-stateful02/src/main/java/org/opendaylight/protocol/pcep/ietf/stateful02/Stateful02PCReportMessageParser.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.ietf.stateful02;
   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.protocol.pcep.spi.PCEPErrors;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.Pcrpt;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.PcrptBuilder;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.lsp.object.Lsp;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.pcrpt.message.PcrptMessageBuilder;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.pcrpt.message.pcrpt.message.Reports;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.pcrpt.message.pcrpt.message.ReportsBuilder;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.pcrpt.message.pcrpt.message.reports.Path;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.pcrpt.message.pcrpt.message.reports.PathBuilder;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.bandwidth.object.Bandwidth;
  31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.explicit.route.object.Ero;
  32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.Metrics;
  33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.MetricsBuilder;
  34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lspa.object.Lspa;
  35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.metric.object.Metric;
  36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.reported.route.object.Rro;
  37
  38 import com.google.common.collect.Lists;
  39
  40 /**
  41  * Parser for {@link Pcrpt}
  42  */
  43 public final class Stateful02PCReportMessageParser extends AbstractMessageParser {
  44
  45         public static final int TYPE = 10;
  46
  47         public Stateful02PCReportMessageParser(final ObjectRegistry registry) {
  48                 super(registry);
  49         }
  50
  51         @Override
  52         public void serializeMessage(final Message message, final ByteBuf out) {
  53                 if (!(message instanceof Pcrpt)) {
  54                         throw new IllegalArgumentException("Wrong instance of Message. Passed instance of " + message.getClass()
  55                                         + ". Nedded PcrptMessage.");
  56                 }
  57                 final Pcrpt msg = (Pcrpt) message;
  58                 final List<Reports> reports = msg.getPcrptMessage().getReports();
  59                 ByteBuf buffer = Unpooled.buffer();
  60                 for (final Reports report : reports) {
  61                         buffer.writeBytes(serializeObject(report.getLsp()));
  62                         final Path p = report.getPath();
  63                         if (p != null) {
  64                                 if (p.getEro() != null) {
  65                                         buffer.writeBytes(serializeObject(p.getEro()));
  66                                 }
  67                                 if (p.getLspa() != null) {
  68                                         buffer.writeBytes(serializeObject(p.getLspa()));
  69                                 }
  70                                 if (p.getBandwidth() != null) {
  71                                         buffer.writeBytes(serializeObject(p.getBandwidth()));
  72                                 }
  73                                 if (p.getRro() != null) {
  74                                         buffer.writeBytes(serializeObject(p.getRro()));
  75                                 }
  76                                 if (p.getMetrics() != null && !p.getMetrics().isEmpty()) {
  77                                         for (final Metrics m : p.getMetrics()) {
  78                                                 buffer.writeBytes(serializeObject(m.getMetric()));
  79                                         }
  80                                 }
  81                         }
  82                 }
  83                 MessageUtil.formatMessage(TYPE, buffer, out);
  84         }
  85
  86         @Override
  87         public Message validate(final List<Object> objects, final List<Message> errors) throws PCEPDeserializerException {
  88                 if (objects == null) {
  89                         throw new IllegalArgumentException("Passed list can't be null.");
  90                 }
  91                 if (objects.isEmpty()) {
  92                         throw new PCEPDeserializerException("Pcrpt message cannot be empty.");
  93                 }
  94
  95                 final List<Reports> reports = Lists.newArrayList();
  96
  97                 while (!objects.isEmpty()) {
  98                         final Reports report = getValidReports(objects, errors);
  99                         if (reports != null) {
 100                                 reports.add(report);
 101                         }
 102                 }
 103                 return new PcrptBuilder().setPcrptMessage(new PcrptMessageBuilder().setReports(reports).build()).build();
 104         }
 105
 106         private Reports getValidReports(final List<Object> objects, final List<Message> errors) {
 107                 final ReportsBuilder builder = new ReportsBuilder();
 108                 if (objects.get(0) instanceof Lsp) {
 109                         builder.setLsp((Lsp) objects.get(0));
 110                         objects.remove(0);
 111                 } else {
 112                         errors.add(createErrorMsg(PCEPErrors.LSP_MISSING));
 113                         return null;
 114                 }
 115                 if (!objects.isEmpty()) {
 116                         final PathBuilder pBuilder = new PathBuilder();
 117                         parsePath(objects, pBuilder);
 118                         builder.setPath(pBuilder.build());
 119                 }
 120                 return builder.build();
 121         }
 122
 123         private void parsePath(final List<Object> objects, final PathBuilder builder) {
 124                 final List<Metrics> pathMetrics = Lists.newArrayList();
 125                 Object obj;
 126                 State state = State.Init;
 127                 while (!objects.isEmpty() && !state.equals(State.End)) {
 128                         obj = objects.get(0);
 129                         switch (state) {
 130                         case Init:
 131                                 state = State.EroIn;
 132                                 if (obj instanceof Ero) {
 133                                         builder.setEro((Ero) obj);
 134                                         break;
 135                                 }
 136                         case EroIn:
 137                                 state = State.LspaIn;
 138                                 if (obj instanceof Lspa) {
 139                                         builder.setLspa((Lspa) obj);
 140                                         break;
 141                                 }
 142                         case LspaIn:
 143                                 state = State.BandwidthIn;
 144                                 if (obj instanceof Bandwidth) {
 145                                         builder.setBandwidth((Bandwidth) obj);
 146                                         break;
 147                                 }
 148                         case BandwidthIn:
 149                                 state = State.RroIn;
 150                                 if (obj instanceof Rro) {
 151                                         builder.setRro((Rro) obj);
 152                                         break;
 153                                 }
 154                         case RroIn:
 155                                 state = State.MetricIn;
 156                                 if (obj instanceof Metric) {
 157                                         pathMetrics.add(new MetricsBuilder().setMetric((Metric) obj).build());
 158                                         state = State.RroIn;
 159                                         break;
 160                                 }
 161                         case MetricIn:
 162                                 state = State.End;
 163                                 break;
 164                         case End:
 165                                 break;
 166                         }
 167                         if (!state.equals(State.End)) {
 168                                 objects.remove(0);
 169                         }
 170                 }
 171                 if (!pathMetrics.isEmpty()) {
 172                         builder.setMetrics(pathMetrics);
 173                 }
 174         }
 175
 176         private enum State {
 177                 Init, EroIn, LspaIn, BandwidthIn, RroIn, MetricIn, End
 178         }
 179
 180         @Override
 181         public int getMessageType() {
 182                 return TYPE;
 183         }
 184 }