BUG-731: reduced complexity
[bgpcep.git] / pcep / ietf-stateful02 / src / main / java / org / opendaylight / protocol / pcep / ietf / stateful02 / Stateful02PCRequestMessageParser.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 com.google.common.base.Optional;
11 import com.google.common.collect.Lists;
12 import io.netty.buffer.ByteBuf;
13 import java.util.List;
14 import org.opendaylight.protocol.pcep.impl.message.PCEPRequestMessageParser;
15 import org.opendaylight.protocol.pcep.spi.ObjectRegistry;
16 import org.opendaylight.protocol.pcep.spi.PCEPErrors;
17 import org.opendaylight.protocol.pcep.spi.VendorInformationObjectRegistry;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.P2p1;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.P2p1Builder;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.lsp.object.Lsp;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.bandwidth.object.Bandwidth;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.classtype.object.ClassType;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.exclude.route.object.Xro;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.include.route.object.Iro;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.load.balancing.object.LoadBalancing;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.Metrics;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lsp.attributes.MetricsBuilder;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.lspa.object.Lspa;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.metric.object.Metric;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.of.object.Of;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcreq.message.pcreq.message.requests.SegmentComputation;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcreq.message.pcreq.message.requests.SegmentComputationBuilder;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcreq.message.pcreq.message.requests.segment.computation.P2p;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcreq.message.pcreq.message.requests.segment.computation.P2pBuilder;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcreq.message.pcreq.message.requests.segment.computation.p2p.ReportedRoute;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcreq.message.pcreq.message.requests.segment.computation.p2p.ReportedRouteBuilder;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.reported.route.object.Rro;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.rp.object.Rp;
41
42 /**
43  * Parser for Pcreq
44  */
45 public final class Stateful02PCRequestMessageParser extends PCEPRequestMessageParser {
46
47     public Stateful02PCRequestMessageParser(final ObjectRegistry registry, final VendorInformationObjectRegistry viReg) {
48         super(registry, viReg);
49     }
50
51     @Override
52     protected void serializeP2P(final ByteBuf buffer, final P2p p2p) {
53         serializeObject(p2p.getEndpointsObj(), buffer);
54         if (p2p.getAugmentation(P2p1.class) != null) {
55             serializeObject(p2p.getAugmentation(P2p1.class).getLsp(), buffer);
56         }
57         final ReportedRoute rr = p2p.getReportedRoute();
58         if (rr != null) {
59             serializeObject(rr.getRro(), buffer);
60             serializeObject(rr.getBandwidth(), buffer);
61         }
62         serializeObject(p2p.getLoadBalancing(), buffer);
63         serializeObject(p2p.getLspa(), buffer);
64         serializeObject(p2p.getBandwidth(), buffer);
65         if (p2p.getMetrics() != null) {
66             for (final Metrics m : p2p.getMetrics()) {
67                 serializeObject(m.getMetric(), buffer);
68             }
69         }
70         serializeObject(p2p.getIro(), buffer);
71         serializeObject(p2p.getRro(), buffer);
72         serializeObject(p2p.getXro(), buffer);
73         serializeObject(p2p.getOf(), buffer);
74         serializeObject(p2p.getClassType(), buffer);
75     }
76
77     @Override
78     protected SegmentComputation getSegmentComputation(final P2pBuilder builder, final List<Object> objects, final List<Message> errors,
79             final Rp rp) {
80         final List<Metrics> metrics = Lists.newArrayList();
81
82         State state = State.Init;
83         while (!objects.isEmpty() && state != State.End) {
84             Object obj = objects.get(0);
85
86             switch (state) {
87             case Init:
88                 state = State.LspIn;
89                 if (obj instanceof Lsp) {
90                     builder.addAugmentation(P2p1.class, new P2p1Builder().setLsp((Lsp) obj).build());
91                     break;
92                 }
93             case LspIn:
94                 state = State.ReportedIn;
95                 if (obj instanceof Rro) {
96                     final ReportedRouteBuilder rrBuilder = new ReportedRouteBuilder();
97                     rrBuilder.setRro((Rro) obj);
98                     objects.remove(0);
99                     obj = objects.get(0);
100                     if (obj instanceof Bandwidth) {
101                         rrBuilder.setBandwidth((Bandwidth) obj);
102                     }
103                     break;
104                 }
105             case ReportedIn:
106                 state = State.LoadBIn;
107                 if (obj instanceof LoadBalancing) {
108                     builder.setLoadBalancing((LoadBalancing) obj);
109                     break;
110                 }
111             case LoadBIn:
112                 state = State.LspaIn;
113                 if (obj instanceof Lspa) {
114                     builder.setLspa((Lspa) obj);
115                     break;
116                 }
117             case LspaIn:
118                 state = State.BandwidthIn;
119                 if (obj instanceof Bandwidth) {
120                     builder.setBandwidth((Bandwidth) obj);
121                     break;
122                 }
123             case BandwidthIn:
124                 state = State.MetricIn;
125                 if (obj instanceof Metric) {
126                     metrics.add(new MetricsBuilder().setMetric((Metric) obj).build());
127                     state = State.BandwidthIn;
128                     break;
129                 }
130             case MetricIn:
131                 state = State.IroIn;
132                 if (obj instanceof Iro) {
133                     builder.setIro((Iro) obj);
134                     break;
135                 }
136             case IroIn:
137                 state = State.RroIn;
138                 if (obj instanceof Rro) {
139                     builder.setRro((Rro) obj);
140                     break;
141                 }
142             case RroIn:
143                 state = State.XroIn;
144                 if (obj instanceof Xro) {
145                     builder.setXro((Xro) obj);
146                     break;
147                 }
148             case XroIn:
149                 state = State.OfIn;
150                 if (obj instanceof Of) {
151                     builder.setOf((Of) obj);
152                     break;
153                 }
154             case OfIn:
155                 state = State.CtIn;
156                 if (obj instanceof ClassType) {
157                     final ClassType classType = (ClassType) obj;
158                     if (!classType.isProcessingRule()) {
159                         errors.add(createErrorMsg(PCEPErrors.P_FLAG_NOT_SET, Optional.of(rp)));
160                     } else {
161                         builder.setClassType(classType);
162                     }
163                     break;
164                 }
165             case CtIn:
166                 state = State.End;
167                 break;
168             case End:
169                 break;
170             }
171             if (!state.equals(State.End)) {
172                 objects.remove(0);
173             }
174         }
175         if (!metrics.isEmpty()) {
176             builder.setMetrics(metrics);
177         }
178
179         if (rp.isReoptimization()
180                 && builder.getBandwidth() != null
181                 && !builder.getReportedRoute().getBandwidth().getBandwidth().equals(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.network.concepts.rev131125.Bandwidth(new byte[] { 0 }))
182                 && builder.getReportedRoute().getRro() == null) {
183             errors.add(createErrorMsg(PCEPErrors.RRO_MISSING, Optional.of(rp)));
184             return null;
185         }
186         return new SegmentComputationBuilder().setP2p(builder.build()).build();
187     }
188
189     private enum State {
190         Init, LspIn, ReportedIn, LoadBIn, LspaIn, BandwidthIn, MetricIn, IroIn, RroIn, XroIn, OfIn, CtIn, End
191     }
192 }