Add new revision for pcep types model
[bgpcep.git] / pcep / ietf-stateful07 / src / main / java / org / opendaylight / protocol / pcep / ietf / initiated00 / CInitiated00PCInitiateMessageParser.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.initiated00;
9
10 import com.google.common.base.Preconditions;
11 import com.google.common.collect.Lists;
12 import io.netty.buffer.ByteBuf;
13 import io.netty.buffer.Unpooled;
14 import java.util.List;
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.rev181109.Pcinitiate;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev181109.PcinitiateBuilder;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev181109.pcinitiate.message.PcinitiateMessageBuilder;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev181109.pcinitiate.message.pcinitiate.message.Requests;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev181109.pcinitiate.message.pcinitiate.message.RequestsBuilder;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.lsp.object.Lsp;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev181109.srp.object.Srp;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.Message;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.Object;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.bandwidth.object.Bandwidth;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.endpoints.object.EndpointsObj;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.explicit.route.object.Ero;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.include.route.object.Iro;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.lsp.attributes.Metrics;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.lsp.attributes.MetricsBuilder;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.lspa.object.Lspa;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev181109.metric.object.Metric;
36
37 /**
38  * Parser for {@link Pcinitiate}
39  */
40 public class CInitiated00PCInitiateMessageParser extends AbstractMessageParser {
41
42     public static final int TYPE = 12;
43
44     public CInitiated00PCInitiateMessageParser(final ObjectRegistry registry) {
45         super(registry);
46     }
47
48     @Override
49     public void serializeMessage(final Message message, final ByteBuf out) {
50         Preconditions.checkArgument(message instanceof Pcinitiate, "Wrong instance of Message. Passed instance of %s. Need PcinitiateMessage.", message.getClass());
51         final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev181109.pcinitiate.message.PcinitiateMessage init = ((Pcinitiate) message).getPcinitiateMessage();
52         final ByteBuf buffer = Unpooled.buffer();
53         for (final Requests req : init.getRequests()) {
54             serializeRequest(req, buffer);
55         }
56         MessageUtil.formatMessage(TYPE, buffer, out);
57     }
58
59     protected void serializeRequest(final Requests req, final ByteBuf buffer) {
60         serializeObject(req.getSrp(), buffer);
61         serializeObject(req.getLsp(), buffer);
62         serializeObject(req.getEndpointsObj(), buffer);
63         serializeObject(req.getEro(), buffer);
64         serializeObject(req.getLspa(), buffer);
65         serializeObject(req.getBandwidth(), buffer);
66         if (req.getMetrics() != null) {
67             for (final Metrics m : req.getMetrics()) {
68                 serializeObject(m.getMetric(), buffer);
69             }
70         }
71         serializeObject(req.getIro(), buffer);
72     }
73
74     @Override
75     protected Message validate(final List<Object> objects, final List<Message> errors) throws PCEPDeserializerException {
76         Preconditions.checkArgument(objects != null, "Passed list can't be null.");
77         final PcinitiateMessageBuilder builder = new PcinitiateMessageBuilder();
78         final List<Requests> reqs = Lists.newArrayList();
79         while (!objects.isEmpty()) {
80             reqs.add(this.getValidRequest(objects));
81         }
82         builder.setRequests(reqs);
83         return new PcinitiateBuilder().setPcinitiateMessage(builder.build()).build();
84     }
85
86     protected Requests getValidRequest(final List<Object> objects) {
87         final RequestsBuilder builder = new RequestsBuilder();
88         builder.setSrp((Srp) objects.get(0));
89         objects.remove(0);
90
91         builder.setLsp((Lsp) objects.get(0));
92         objects.remove(0);
93
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             state = insertObject(state, obj, builder, metrics);
101             if (!state.equals(State.END)) {
102                 objects.remove(0);
103             }
104         }
105         builder.setMetrics(metrics);
106         return builder.build();
107     }
108
109     private static State insertObject(final State state, final Object obj, final RequestsBuilder builder,
110             final List<Metrics> metrics) {
111         switch (state) {
112         case INIT:
113             if (obj instanceof EndpointsObj) {
114                 builder.setEndpointsObj((EndpointsObj) obj);
115                 return State.ENDPOINTS_IN;
116             }
117         case ENDPOINTS_IN:
118             if (obj instanceof Ero) {
119                 builder.setEro((Ero) obj);
120                 return State.ERO_IN;
121             }
122         case ERO_IN:
123             if (obj instanceof Lspa) {
124                 builder.setLspa((Lspa) obj);
125                 return State.LSPA_IN;
126             }
127         case LSPA_IN:
128             if (obj instanceof Bandwidth) {
129                 builder.setBandwidth((Bandwidth) obj);
130                 return State.BANDWIDTH_IN;
131             }
132         case BANDWIDTH_IN:
133             if (obj instanceof Metric) {
134                 metrics.add(new MetricsBuilder().setMetric((Metric) obj).build());
135                 return State.BANDWIDTH_IN;
136             }
137         case METRIC_IN:
138             if (obj instanceof Iro) {
139                 builder.setIro((Iro) obj);
140                 return State.IRO_IN;
141             }
142         case IRO_IN:
143         case END:
144             return State.END;
145         default:
146             return state;
147         }
148     }
149
150     private enum State {
151         INIT, ENDPOINTS_IN, ERO_IN, LSPA_IN, BANDWIDTH_IN, METRIC_IN, IRO_IN, END
152     }
153 }