io.netlibs.bgp.netty.protocol.update.CommunityPathAttributeCodecHandler.java Source code

Introduction

Here is the source code for io.netlibs.bgp.netty.protocol.update.CommunityPathAttributeCodecHandler.java

Source

/**
 *  Copyright 2012 Rainer Bieniek (Rainer.Bieniek@web.de)
 *
 *  Licensed under the Apache License, Version 2.0 (the "License");
 *  you may not use this file except in compliance with the License.
 *  You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 *  Unless required by applicable law or agreed to in writing, software
 *  distributed under the License is distributed on an "AS IS" BASIS,
 *  WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 *  See the License for the specific language governing permissions and
 *  limitations under the License.
 *
 * File: org.bgp4j.netty.protocol.update.CommunityPathAttributeCodecHandler.java
 */
package io.netlibs.bgp.netty.protocol.update;

import io.netlibs.bgp.protocol.BGPv4Constants;
import io.netlibs.bgp.protocol.attributes.CommunityMember;
import io.netlibs.bgp.protocol.attributes.CommunityPathAttribute;
import io.netty.buffer.ByteBuf;
import io.netty.buffer.Unpooled;

/**
 * @author Rainer Bieniek (Rainer.Bieniek@web.de)
 *
 */
public class CommunityPathAttributeCodecHandler extends PathAttributeCodecHandler<CommunityPathAttribute> {

    /*
     * (non-Javadoc)
     *
     * @see org.bgp4j.netty.protocol.update.PathAttributeCodecHandler#typeCode(org.bgp4j.netty.protocol.update.PathAttribute)
     */
    @Override
    public int typeCode(final CommunityPathAttribute attr) {
        return BGPv4Constants.BGP_PATH_ATTRIBUTE_TYPE_COMMUNITIES;
    }

    /*
     * (non-Javadoc)
     *
     * @see org.bgp4j.netty.protocol.update.PathAttributeCodecHandler#valueLength(org.bgp4j.netty.protocol.update.PathAttribute)
     */
    @Override
    public int valueLength(final CommunityPathAttribute attr) {
        int size = 0;

        if (attr.getMembers() != null) {
            size += 4 * attr.getMembers().size();
        }

        return size;
    }

    /*
     * (non-Javadoc)
     *
     * @see org.bgp4j.netty.protocol.update.PathAttributeCodecHandler#encodeValue(org.bgp4j.netty.protocol.update.PathAttribute)
     */
    @Override
    public ByteBuf encodeValue(final CommunityPathAttribute attr) {
        final ByteBuf buffer = Unpooled.buffer(this.valueLength(attr));

        if (attr.getMembers() != null) {
            for (final CommunityMember member : attr.getMembers()) {
                buffer.writeShort(member.getAsNumber());
                buffer.writeShort(member.getValue());
            }
        }

        return buffer;
    }

}