BUG-139: PCEP TLVs extensions implementation
[bgpcep.git] / pcep / ietf-stateful07 / src / main / java / org / opendaylight / protocol / pcep / sync / optimizations / SyncOptimizationsLspObjectParser.java
1 /*
2  * Copyright (c) 2015 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.sync.optimizations;
9
10 import com.google.common.base.Preconditions;
11 import io.netty.buffer.ByteBuf;
12 import org.opendaylight.protocol.pcep.ietf.initiated00.CInitiated00LspObjectParser;
13 import org.opendaylight.protocol.pcep.spi.TlvRegistry;
14 import org.opendaylight.protocol.pcep.spi.VendorInformationTlvRegistry;
15 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.LspDbVersionTlv;
16 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.Tlvs1;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.Tlvs1Builder;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.controller.pcep.sync.optimizations.rev150714.lsp.db.version.tlv.LspDbVersion;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.lsp.Tlvs;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.ietf.stateful.rev131222.lsp.object.lsp.TlvsBuilder;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Tlv;
22
23 public class SyncOptimizationsLspObjectParser extends CInitiated00LspObjectParser {
24
25     public SyncOptimizationsLspObjectParser(final TlvRegistry tlvReg, final VendorInformationTlvRegistry viTlvReg) {
26         super(tlvReg, viTlvReg);
27     }
28
29     @Override
30     public void serializeTlvs(final Tlvs tlvs, final ByteBuf body) {
31         super.serializeTlvs(tlvs, body);
32         serializeAugmentation(tlvs.getAugmentation(Tlvs1.class), body);
33     }
34
35     private void serializeAugmentation(final Tlvs1 tlv, final ByteBuf body) {
36         if (tlv == null) {
37             return;
38         }
39         Preconditions.checkArgument(tlv instanceof LspDbVersionTlv, "TLV object is not instance of LspDbVersionTlv.");
40         final LspDbVersionTlv dbVersion = tlv;
41         serializeTlv(dbVersion.getLspDbVersion(), body);
42     }
43
44     @Override
45     public void addTlv(final TlvsBuilder tbuilder, final Tlv tlv) {
46         super.addTlv(tbuilder, tlv);
47         final Tlvs1Builder syncOptTlvsBuilder = new Tlvs1Builder();
48         if (tbuilder.getAugmentation(Tlvs1.class) != null) {
49             final Tlvs1 t = tbuilder.getAugmentation(Tlvs1.class);
50             if (t.getLspDbVersion() != null) {
51                 syncOptTlvsBuilder.setLspDbVersion(t.getLspDbVersion());
52             }
53         }
54         if (tlv instanceof LspDbVersion) {
55             syncOptTlvsBuilder.setLspDbVersion((LspDbVersion) tlv);
56         }
57         tbuilder.addAugmentation(Tlvs1.class, syncOptTlvsBuilder.build());
58     }
59 }