IllegalArgument Exception changed to Preconditions checks
[bgpcep.git] / pcep / ietf-stateful02 / src / main / java / org / opendaylight / protocol / pcep / ietf / stateful02 / Stateful02PCReplyMessageParser.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.base.Preconditions;
12 import com.google.common.collect.Lists;
13 import io.netty.buffer.ByteBuf;
14 import io.netty.buffer.Unpooled;
15 import java.util.List;
16 import org.opendaylight.protocol.pcep.impl.message.PCEPReplyMessageParser;
17 import org.opendaylight.protocol.pcep.spi.MessageUtil;
18 import org.opendaylight.protocol.pcep.spi.ObjectRegistry;
19 import org.opendaylight.protocol.pcep.spi.PCEPErrors;
20 import org.opendaylight.protocol.pcep.spi.VendorInformationObjectRegistry;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.Replies1;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.Replies1Builder;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.lsp.object.Lsp;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.message.rev131007.Pcrep;
25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Message;
26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.explicit.route.object.Ero;
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.pcrep.message.pcrep.message.Replies;
30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.RepliesBuilder;
31 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.Result;
32 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.FailureCase;
33 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.FailureCaseBuilder;
34 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.SuccessCase;
35 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.SuccessCaseBuilder;
36 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.failure._case.NoPath;
37 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.success._case.SuccessBuilder;
38 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.success._case.success.Paths;
39 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.pcrep.message.replies.result.success._case.success.PathsBuilder;
40 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.rp.object.Rp;
41
42 /**
43  * Parser for {@link Pcrep}
44  */
45 @Deprecated
46 public final class Stateful02PCReplyMessageParser extends PCEPReplyMessageParser {
47
48     public Stateful02PCReplyMessageParser(final ObjectRegistry registry, final VendorInformationObjectRegistry viReg) {
49         super(registry, viReg);
50     }
51
52     @Override
53     public void serializeMessage(final Message message, final ByteBuf out) {
54         Preconditions.checkArgument(message instanceof Pcrep, "Wrong instance of Message. Passed instance of %s. Need Pcrep.", message.getClass());
55         final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.pcrep.message.PcrepMessage repMsg = ((Pcrep) message).getPcrepMessage();
56         Preconditions.checkArgument(repMsg.getReplies() != null && !repMsg.getReplies().isEmpty(), "Replies cannot be null or empty.");
57         final ByteBuf buffer = Unpooled.buffer();
58         for (final Replies reply : repMsg.getReplies()) {
59             Preconditions.checkArgument(reply.getRp() != null, "Reply must contain RP object.");
60             serializeObject(reply.getRp(), buffer);
61             if (reply.getAugmentation(Replies1.class) != null && reply.getAugmentation(Replies1.class).getLsp() != null) {
62                 serializeObject(reply.getAugmentation(Replies1.class).getLsp(), buffer);
63             }
64             if (reply.getResult() != null) {
65                 if (reply.getResult() instanceof FailureCase) {
66                     final FailureCase f = ((FailureCase) reply.getResult());
67                     serializeObject(f.getNoPath(), buffer);
68                     serializeObject(f.getLspa(), buffer);
69                     serializeObject(f.getBandwidth(), buffer);
70                     if (f.getMetrics() != null && !f.getMetrics().isEmpty()) {
71                         for (final Metrics m : f.getMetrics()) {
72                             serializeObject(m.getMetric(), buffer);
73                         }
74                     }
75                     serializeObject(f.getIro(), buffer);
76                 } else {
77                     final SuccessCase s = (SuccessCase) reply.getResult();
78                     for (final Paths p : s.getSuccess().getPaths()) {
79                         serializeObject(p.getEro(), buffer);
80                         serializeObject(p.getLspa(), buffer);
81                         serializeObject(p.getOf(), buffer);
82                         serializeObject(p.getBandwidth(), buffer);
83                         if (p.getMetrics() != null && !p.getMetrics().isEmpty()) {
84                             for (final Metrics m : p.getMetrics()) {
85                                 serializeObject(m.getMetric(), buffer);
86                             }
87                         }
88                         serializeObject(p.getIro(), buffer);
89                     }
90                 }
91             }
92         }
93         MessageUtil.formatMessage(TYPE, buffer, out);
94     }
95
96     @Override
97     protected Replies getValidReply(final List<Object> objects, final List<Message> errors) {
98         if (!(objects.get(0) instanceof Rp)) {
99             errors.add(createErrorMsg(PCEPErrors.RP_MISSING, Optional.<Rp>absent()));
100             return null;
101         }
102         final Rp rp = (Rp) objects.get(0);
103         objects.remove(0);
104         Result res = null;
105         Lsp lsp = null;
106         if (objects.get(0) instanceof Lsp) {
107             lsp = (Lsp) objects.get(0);
108             objects.remove(0);
109         }
110         if (!objects.isEmpty()) {
111             if (objects.get(0) instanceof NoPath) {
112                 final NoPath noPath = (NoPath) objects.get(0);
113                 objects.remove(0);
114                 final FailureCaseBuilder builder = new FailureCaseBuilder();
115                 builder.setNoPath(noPath);
116                 while (!objects.isEmpty()) {
117                     parseAttributes(builder, objects);
118                 }
119                 res = builder.build();
120             } else if (objects.get(0) instanceof Ero) {
121                 final Ero ero = (Ero) objects.get(0);
122                 objects.remove(0);
123                 final SuccessBuilder builder = new SuccessBuilder();
124                 final List<Paths> paths = Lists.newArrayList();
125                 final PathsBuilder pBuilder = new PathsBuilder();
126                 pBuilder.setEro(ero);
127                 while (!objects.isEmpty()) {
128                     parsePath(pBuilder, objects);
129                     paths.add(pBuilder.build());
130                 }
131                 builder.setPaths(paths);
132                 res = new SuccessCaseBuilder().setSuccess(builder.build()).build();
133             }
134         }
135         return new RepliesBuilder().setRp(rp).addAugmentation(Replies1.class, new Replies1Builder().setLsp(lsp).build()).setResult(res).build();
136     }
137 }