BUG-612 : switched ERO to ByteBuf.

[bgpcep.git] / pcep / impl / src / main / java / org / opendaylight / protocol / pcep / impl / object / PCEPIncludeRouteObjectParser.java

diff --git a/pcep/impl/src/main/java/org/opendaylight/protocol/pcep/impl/object/PCEPIncludeRouteObjectParser.java b/pcep/impl/src/main/java/org/opendaylight/protocol/pcep/impl/object/PCEPIncludeRouteObjectParser.java
index 60e1cc3b78f4a09d89d7f2390a067300afb481c5..0196394ee2e4f2416169b0927b58d0e237df4967 100644 (file)
--- a/pcep/impl/src/main/java/org/opendaylight/protocol/pcep/impl/object/PCEPIncludeRouteObjectParser.java
+++ b/pcep/impl/src/main/java/org/opendaylight/protocol/pcep/impl/object/PCEPIncludeRouteObjectParser.java
@@ -7,41 +7,66 @@
  */
 package org.opendaylight.protocol.pcep.impl.object;
 
+import io.netty.buffer.ByteBuf;
+
+import java.util.ArrayList;
 import java.util.List;
 
-import org.opendaylight.protocol.pcep.PCEPDeserializerException;
-import org.opendaylight.protocol.pcep.PCEPObject;
-import org.opendaylight.protocol.pcep.impl.PCEPEROSubobjectParser;
-import org.opendaylight.protocol.pcep.impl.PCEPObjectParser;
-import org.opendaylight.protocol.pcep.object.PCEPIncludeRouteObject;
-import org.opendaylight.protocol.pcep.subobject.ExplicitRouteSubobject;
+import org.opendaylight.protocol.pcep.spi.EROSubobjectRegistry;
+import org.opendaylight.protocol.pcep.spi.ObjectUtil;
+import org.opendaylight.protocol.pcep.spi.PCEPDeserializerException;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.Object;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.ObjectHeader;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.include.route.object.Iro;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.include.route.object.IroBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.include.route.object.iro.Subobject;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.include.route.object.iro.SubobjectBuilder;
+
+import com.google.common.base.Preconditions;
+import com.google.common.collect.Lists;
 
 /**
- * Parser for {@link org.opendaylight.protocol.pcep.object.PCEPIncludeRouteObject
- * PCEPIncludeRouteObject}
+ * Parser for {@link Iro}
  */
-public class PCEPIncludeRouteObjectParser implements PCEPObjectParser {
+public class PCEPIncludeRouteObjectParser extends AbstractEROWithSubobjectsParser {
 
-       @Override
-       public PCEPObject parse(byte[] bytes, boolean processed, boolean ignored) throws PCEPDeserializerException {
-               if (bytes == null || bytes.length == 0)
-                       throw new IllegalArgumentException("Byte array is mandatory. Can't be null or empty.");
+       public static final int CLASS = 10;
+
+       public static final int TYPE = 1;
 
-               final List<ExplicitRouteSubobject> subobjects = PCEPEROSubobjectParser.parse(bytes);
-               if (subobjects.isEmpty())
-                       throw new PCEPDeserializerException("Empty Include Route Object.");
+       public PCEPIncludeRouteObjectParser(final EROSubobjectRegistry subobjReg) {
+               super(subobjReg);
+       }
 
-               return new PCEPIncludeRouteObject(subobjects, processed, ignored);
+       @Override
+       public Iro parseObject(final ObjectHeader header, final ByteBuf bytes) throws PCEPDeserializerException {
+               Preconditions.checkArgument(bytes != null && bytes.isReadable(), "Array of bytes is mandatory. Can't be null or empty.");
+               final IroBuilder builder = new IroBuilder();
+               builder.setIgnore(header.isIgnore());
+               builder.setProcessingRule(header.isProcessingRule());
+               final List<Subobject> subs = new ArrayList<>();
+               for (final org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.explicit.route.object.ero.Subobject s : parseSubobjects(bytes)) {
+                       subs.add(new SubobjectBuilder().setSubobjectType(s.getSubobjectType()).build());
+               }
+               builder.setSubobject(subs);
+               return builder.build();
        }
 
        @Override
-       public byte[] put(PCEPObject obj) {
-               if (!(obj instanceof PCEPIncludeRouteObject))
-                       throw new IllegalArgumentException("Wrong instance of PCEPObject. Passed " + obj.getClass() + ". Needed PCEPIncludeRouteObject.");
+       public byte[] serializeObject(final Object object) {
+               if (!(object instanceof Iro)) {
+                       throw new IllegalArgumentException("Wrong instance of PCEPObject. Passed " + object.getClass() + ". Needed IncludeRouteObject.");
+               }
+               final Iro iro = ((Iro) object);
 
-               assert !(((PCEPIncludeRouteObject) obj).getSubobjects().isEmpty()) : "Empty Include Route Object.";
+               assert !(iro.getSubobject().isEmpty()) : "Empty Include Route Object.";
 
-               return PCEPEROSubobjectParser.put(((PCEPIncludeRouteObject) obj).getSubobjects());
-       }
+               final List<org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.explicit.route.object.ero.Subobject> subs = Lists.newArrayList();
 
+               for (final Subobject s : iro.getSubobject()) {
+                       subs.add(new org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.pcep.types.rev131005.explicit.route.object.ero.SubobjectBuilder().setLoose(
+                                       false).setSubobjectType(s.getSubobjectType()).build());
+               }
+               return ObjectUtil.formatSubobject(TYPE, CLASS, object.isProcessingRule(), object.isIgnore(), serializeSubobject(subs));
+       }
 }