IllegalArgument Exception changed to Preconditions checks
[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.rev131126.Pcinitiate;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.PcinitiateBuilder;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.pcinitiate.message.PcinitiateMessageBuilder;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.pcinitiate.message.pcinitiate.message.Requests;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated.rev131126.pcinitiate.message.pcinitiate.message.RequestsBuilder;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.Lsp;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.srp.object.Srp;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.bandwidth.object.Bandwidth;
29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.endpoints.object.EndpointsObj;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.explicit.route.object.Ero;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.include.route.object.Iro;
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
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.rev131126.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 State insertObject(final State state, final Object obj, final RequestsBuilder builder, final List<Metrics> metrics) {
110         switch (state) {
111         case INIT:
112             if (obj instanceof EndpointsObj) {
113                 builder.setEndpointsObj((EndpointsObj) obj);
114                 return State.ENDPOINTS_IN;
115             }
116         case ENDPOINTS_IN:
117             if (obj instanceof Ero) {
118                 builder.setEro((Ero) obj);
119                 return State.ERO_IN;
120             }
121         case ERO_IN:
122             if (obj instanceof Lspa) {
123                 builder.setLspa((Lspa) obj);
124                 return State.LSPA_IN;
125             }
126         case LSPA_IN:
127             if (obj instanceof Bandwidth) {
128                 builder.setBandwidth((Bandwidth) obj);
129                 return State.BANDWIDTH_IN;
130             }
131         case BANDWIDTH_IN:
132             if (obj instanceof Metric) {
133                 metrics.add(new MetricsBuilder().setMetric((Metric) obj).build());
134                 return State.BANDWIDTH_IN;
135             }
136         case METRIC_IN:
137             if (obj instanceof Iro) {
138                 builder.setIro((Iro) obj);
139                 return State.IRO_IN;
140             }
141         case IRO_IN:
142         case END:
143             return State.END;
144         default:
145             return state;
146         }
147     }
148
149     private enum State {
150         INIT, ENDPOINTS_IN, ERO_IN, LSPA_IN, BANDWIDTH_IN, METRIC_IN, IRO_IN, END
151     }
152 }