bmp/bmp-parser-impl/src/main/java/org/opendaylight/protocol/bmp/parser/tlv/StatType010TlvHandler.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
   9 package org.opendaylight.protocol.bmp.parser.tlv;
  10
  11 import static java.util.Objects.requireNonNull;
  12 import static org.opendaylight.protocol.util.ByteBufWriteUtil.writeUnsignedByte;
  13 import static org.opendaylight.protocol.util.ByteBufWriteUtil.writeUnsignedLong;
  14 import static org.opendaylight.protocol.util.ByteBufWriteUtil.writeUnsignedShort;
  15
  16 import com.google.common.base.Preconditions;
  17 import io.netty.buffer.ByteBuf;
  18 import io.netty.buffer.Unpooled;
  19 import java.math.BigInteger;
  20 import org.opendaylight.protocol.bgp.parser.spi.AddressFamilyRegistry;
  21 import org.opendaylight.protocol.bgp.parser.spi.SubsequentAddressFamilyRegistry;
  22 import org.opendaylight.protocol.bmp.spi.parser.BmpDeserializationException;
  23 import org.opendaylight.protocol.bmp.spi.parser.BmpTlvParser;
  24 import org.opendaylight.protocol.bmp.spi.parser.BmpTlvSerializer;
  25 import org.opendaylight.protocol.bmp.spi.parser.TlvUtil;
  26 import org.opendaylight.protocol.util.ByteArray;
  27 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.yang.types.rev130715.Gauge64;
  28 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.Tlv;
  29 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.stat.tlvs.PerAfiSafiLocRibTlv;
  30 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.stat.tlvs.PerAfiSafiLocRibTlvBuilder;
  31
  32 public class StatType010TlvHandler implements BmpTlvParser, BmpTlvSerializer {
  33
  34     public static final int TYPE = 10;
  35     private final AddressFamilyRegistry afiRegistry;
  36     private final SubsequentAddressFamilyRegistry safiRegistry;
  37
  38     public StatType010TlvHandler(final AddressFamilyRegistry afiReg, SubsequentAddressFamilyRegistry safiReg) {
  39         this.afiRegistry = requireNonNull(afiReg, "AddressFamily cannot be null");
  40         this.safiRegistry = requireNonNull(safiReg, "SubsequentAddressFamily cannot be null");
  41     }
  42
  43     @Override
  44     public void serializeTlv(final Tlv tlv, final ByteBuf output) {
  45         Preconditions.checkArgument(tlv instanceof PerAfiSafiLocRibTlv, "PerAfiSafiLocRibInTlv is mandatory.");
  46         final ByteBuf buffer = Unpooled.buffer();
  47         writeUnsignedShort(this.afiRegistry.numberForClass(((PerAfiSafiLocRibTlv) tlv).getAfi()), buffer);
  48         writeUnsignedByte(this.safiRegistry.numberForClass(((PerAfiSafiLocRibTlv) tlv).getSafi()).shortValue(), buffer);
  49         writeUnsignedLong(((PerAfiSafiLocRibTlv) tlv).getCount().getValue(), buffer);
  50         TlvUtil.formatTlv(TYPE, buffer, output);
  51     }
  52
  53     @Override
  54     public Tlv parseTlv(final ByteBuf buffer) throws BmpDeserializationException {
  55         if (buffer == null) {
  56             return null;
  57         }
  58         return new PerAfiSafiLocRibTlvBuilder()
  59                 .setAfi(this.afiRegistry.classForFamily(buffer.readUnsignedShort()))
  60                 .setSafi(this.safiRegistry.classForFamily(buffer.readUnsignedByte()))
  61                 .setCount(new Gauge64(new BigInteger(ByteArray.readAllBytes(buffer)))).build();
  62     }
  63 }