Bug-479: Implementation of Vendor-Information TLV

[bgpcep.git] / pcep / ietf-stateful02 / src / main / java / org / opendaylight / protocol / pcep / crabbe / initiated00 / PCEPOpenObjectParser.java

diff --git a/pcep/ietf-stateful02/src/main/java/org/opendaylight/protocol/pcep/crabbe/initiated00/PCEPOpenObjectParser.java b/pcep/ietf-stateful02/src/main/java/org/opendaylight/protocol/pcep/crabbe/initiated00/PCEPOpenObjectParser.java
index a1df09163be8176d1ba5567357be8788e9c1b49e..cc3e9f0180f6322e8482612cf9149bb0d9d78962 100644 (file)
--- a/pcep/ietf-stateful02/src/main/java/org/opendaylight/protocol/pcep/crabbe/initiated00/PCEPOpenObjectParser.java
+++ b/pcep/ietf-stateful02/src/main/java/org/opendaylight/protocol/pcep/crabbe/initiated00/PCEPOpenObjectParser.java
@@ -7,9 +7,10 @@
  */
 package org.opendaylight.protocol.pcep.crabbe.initiated00;
 
+import io.netty.buffer.ByteBuf;
 import org.opendaylight.protocol.pcep.ietf.stateful02.Stateful02OpenObjectParser;
-import org.opendaylight.protocol.pcep.spi.TlvHandlerRegistry;
-import org.opendaylight.protocol.util.ByteArray;
+import org.opendaylight.protocol.pcep.spi.TlvRegistry;
+import org.opendaylight.protocol.pcep.spi.VendorInformationTlvRegistry;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated._00.rev140113.Tlvs1;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated._00.rev140113.Tlvs1Builder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.initiated._00.rev140113.lsp.cleanup.tlv.LspCleanup;
@@ -17,67 +18,55 @@ import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.cra
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.Tlvs2Builder;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.crabbe.stateful._02.rev140110.stateful.capability.tlv.Stateful;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Tlv;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.Open;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.open.Tlvs;
 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.open.object.open.TlvsBuilder;
 
 /**
- * Parser for {@link Open}
+ * Parser for Open object
  */
 public final class PCEPOpenObjectParser extends Stateful02OpenObjectParser {
 
-       public PCEPOpenObjectParser(final TlvHandlerRegistry tlvReg) {
-               super(tlvReg);
-       }
+    public PCEPOpenObjectParser(TlvRegistry tlvReg, VendorInformationTlvRegistry viTlvReg) {
+        super(tlvReg, viTlvReg);
+    }
 
-       @Override
-       public void addTlv(final TlvsBuilder tbuilder, final Tlv tlv) {
-               super.addTlv(tbuilder, tlv);
-               final Tlvs2Builder statefulBuilder = new Tlvs2Builder();
-               if (tbuilder.getAugmentation(Tlvs2.class) != null) {
-                       final Tlvs2 t = tbuilder.getAugmentation(Tlvs2.class);
-                       if (t.getStateful() != null) {
-                               statefulBuilder.setStateful(t.getStateful());
-                       }
-               }
-               final Tlvs1Builder cleanupBuilder = new Tlvs1Builder();
-               if (tbuilder.getAugmentation(Tlvs1.class) != null) {
-                       final Tlvs1 t = tbuilder.getAugmentation(Tlvs1.class);
-                       if (t.getLspCleanup() != null) {
-                               cleanupBuilder.setLspCleanup(t.getLspCleanup());
-                       }
-               }
-               if (tlv instanceof Stateful) {
-                       statefulBuilder.setStateful((Stateful) tlv);
-               } else if (tlv instanceof LspCleanup) {
-                       cleanupBuilder.setLspCleanup((LspCleanup) tlv);
-               }
-               tbuilder.addAugmentation(Tlvs2.class, statefulBuilder.build());
-               tbuilder.addAugmentation(Tlvs1.class, cleanupBuilder.build());
-       }
+    @Override
+    public void addTlv(final TlvsBuilder tbuilder, final Tlv tlv) {
+        super.addTlv(tbuilder, tlv);
+        final Tlvs2Builder statefulBuilder = new Tlvs2Builder();
+        if (tbuilder.getAugmentation(Tlvs2.class) != null) {
+            final Tlvs2 t = tbuilder.getAugmentation(Tlvs2.class);
+            if (t.getStateful() != null) {
+                statefulBuilder.setStateful(t.getStateful());
+            }
+        }
+        final Tlvs1Builder cleanupBuilder = new Tlvs1Builder();
+        if (tbuilder.getAugmentation(Tlvs1.class) != null) {
+            final Tlvs1 t = tbuilder.getAugmentation(Tlvs1.class);
+            if (t.getLspCleanup() != null) {
+                cleanupBuilder.setLspCleanup(t.getLspCleanup());
+            }
+        }
+        if (tlv instanceof Stateful) {
+            statefulBuilder.setStateful((Stateful) tlv);
+        } else if (tlv instanceof LspCleanup) {
+            cleanupBuilder.setLspCleanup((LspCleanup) tlv);
+        }
+        tbuilder.addAugmentation(Tlvs2.class, statefulBuilder.build());
+        tbuilder.addAugmentation(Tlvs1.class, cleanupBuilder.build());
+    }
 
-       @Override
-       public byte[] serializeTlvs(final Tlvs tlvs) {
-               if (tlvs == null) {
-                       return new byte[0];
-               }
-               final byte[] prev = super.serializeTlvs(tlvs);
-               int finalLength = prev.length;
-               byte[] cleanupBytes = null;
-               if (tlvs.getAugmentation(Tlvs1.class) != null) {
-                       final Tlvs1 cleanupTlv = tlvs.getAugmentation(Tlvs1.class);
-                       if (cleanupTlv.getLspCleanup() != null) {
-                               cleanupBytes = serializeTlv(cleanupTlv.getLspCleanup());
-                               finalLength += cleanupBytes.length;
-                       }
-               }
-               final byte[] result = new byte[finalLength];
-               ByteArray.copyWhole(prev, result, 0);
-               int offset = prev.length;
-               if (cleanupBytes != null) {
-                       ByteArray.copyWhole(cleanupBytes, result, offset);
-                       offset += cleanupBytes.length;
-               }
-               return result;
-       }
+    @Override
+    public void serializeTlvs(final Tlvs tlvs, final ByteBuf body) {
+        if (tlvs == null) {
+            return;
+        }
+        super.serializeTlvs(tlvs, body);
+        if (tlvs.getAugmentation(Tlvs1.class) != null) {
+            final Tlvs1 cleanupTlv = tlvs.getAugmentation(Tlvs1.class);
+            if (cleanupTlv.getLspCleanup() != null) {
+                serializeTlv(cleanupTlv.getLspCleanup(), body);
+            }
+        }
+    }
 }