Fixed failing build after yangtools update (List fields are by default not null).
[bgpcep.git] / bgp / parser-impl / src / main / java / org / opendaylight / protocol / bgp / parser / impl / message / update / CommunitiesAttributeParser.java
index 3ed49eb3d220de6899ef9464cc9e5e891eb70059..9e513bc4eef0b86852a0c0b7288eda1b18d36229 100644 (file)
@@ -1,27 +1,62 @@
+/*
+ * Copyright (c) 2013 Cisco Systems, Inc. and others.  All rights reserved.
+ *
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v1.0 which accompanies this distribution,
+ * and is available at http://www.eclipse.org/legal/epl-v10.html
+ */
 package org.opendaylight.protocol.bgp.parser.impl.message.update;
 
+import com.google.common.base.Preconditions;
+import com.google.common.collect.Lists;
+import io.netty.buffer.ByteBuf;
+import io.netty.buffer.Unpooled;
 import java.util.List;
-
 import org.opendaylight.protocol.bgp.parser.BGPDocumentedException;
 import org.opendaylight.protocol.bgp.parser.spi.AttributeParser;
-import org.opendaylight.protocol.util.ByteArray;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.path.attributes.Communities;
-import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.update.PathAttributesBuilder;
+import org.opendaylight.protocol.bgp.parser.spi.AttributeSerializer;
+import org.opendaylight.protocol.bgp.parser.spi.AttributeUtil;
+import org.opendaylight.protocol.util.ReferenceCache;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.PathAttributes;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.path.attributes.Communities;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.update.PathAttributesBuilder;
+import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.types.rev130919.Community;
+import org.opendaylight.yangtools.yang.binding.DataObject;
 
-import com.google.common.collect.Lists;
+public final class CommunitiesAttributeParser implements AttributeParser, AttributeSerializer {
+
+    public static final int TYPE = 8;
+    public static final int ATTR_LENGTH = 4;
+
+    private final ReferenceCache refCache;
 
-final class CommunitiesAttributeParser implements AttributeParser {
-       static final int TYPE = 8;
+    public CommunitiesAttributeParser(final ReferenceCache refCache) {
+        this.refCache = Preconditions.checkNotNull(refCache);
+    }
 
-       @Override
-       public void parseAttribute(final byte[] bytes, final PathAttributesBuilder builder) throws BGPDocumentedException {
-               final List<Communities> set = Lists.newArrayList();
-               int i = 0;
-               while (i < bytes.length) {
-                       set.add((Communities) CommunitiesParser.parseCommunity(ByteArray.subByte(bytes, i, CommunitiesParser.COMMUNITY_LENGTH)));
-                       i += CommunitiesParser.COMMUNITY_LENGTH;
-               }
+    @Override
+    public void parseAttribute(final ByteBuf buffer, final PathAttributesBuilder builder) throws BGPDocumentedException {
+        final List<Communities> set = Lists.newArrayList();
+        while (buffer.isReadable()) {
+            set.add((Communities) CommunitiesParser.parseCommunity(this.refCache, buffer.slice(buffer.readerIndex(),
+                    CommunitiesParser.COMMUNITY_LENGTH)));
+            buffer.skipBytes(CommunitiesParser.COMMUNITY_LENGTH);
+        }
+        builder.setCommunities(set);
+    }
 
-               builder.setCommunities(set);
-       }
-}
\ No newline at end of file
+    @Override
+    public void serializeAttribute(final DataObject attribute, final ByteBuf byteAggregator) {
+        final PathAttributes pathAttributes = (PathAttributes) attribute;
+        final List<Communities> communities = pathAttributes.getCommunities();
+        if (communities == null || communities.isEmpty()) {
+            return;
+        }
+        final ByteBuf communitiesBuffer = Unpooled.buffer();
+        for (final Community community : communities) {
+            communitiesBuffer.writeShort(community.getAsNumber().getValue().shortValue());
+            communitiesBuffer.writeShort(community.getSemantics().shortValue());
+        }
+        AttributeUtil.formatAttribute(AttributeUtil.OPTIONAL | AttributeUtil.TRANSITIVE, TYPE, communitiesBuffer, byteAggregator);
+    }
+}