bgp/bmp-parser-impl/src/main/java/org/opendaylight/protocol/bmp/parser/message/InitiationHandler.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.message;
  10
  11 import com.google.common.base.Preconditions;
  12 import com.google.common.collect.ImmutableList;
  13 import io.netty.buffer.ByteBuf;
  14 import org.opendaylight.protocol.bmp.spi.parser.AbstractBmpMessageWithTlvParser;
  15 import org.opendaylight.protocol.bmp.spi.parser.BmpDeserializationException;
  16 import org.opendaylight.protocol.bmp.spi.parser.BmpTlvRegistry;
  17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev150512.InitiationMessage;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev150512.InitiationMessageBuilder;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev150512.Tlv;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev150512.description.tlv.DescriptionTlv;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev150512.initiation.Tlvs;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev150512.initiation.TlvsBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev150512.name.tlv.NameTlv;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev150512.string.informations.StringInformation;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev150512.string.informations.StringInformationBuilder;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev150512.string.tlv.StringTlv;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev150512.string.tlv.StringTlvBuilder;
  28 import org.opendaylight.yangtools.yang.binding.Notification;
  29
  30 /**
  31  * Created by cgasparini on 13.5.2015.
  32  */
  33 public class InitiationHandler extends AbstractBmpMessageWithTlvParser<TlvsBuilder> {
  34
  35     public InitiationHandler(final BmpTlvRegistry tlvRegistry) {
  36         super(tlvRegistry);
  37     }
  38
  39     private static final int MESSAGE_TYPE = 4;
  40
  41     @Override
  42     public void serializeMessageBody(final Notification message, final ByteBuf buffer) {
  43         Preconditions.checkArgument(message instanceof InitiationMessage, "Incorrect instance of BGP message. The Initiation Message is expected.");
  44         final InitiationMessage initiation = (InitiationMessage) message;
  45         serializeTlvs(initiation.getTlvs(), buffer);
  46     }
  47
  48     @Override
  49     public Notification parseMessageBody(final ByteBuf bytes) throws BmpDeserializationException {
  50         final InitiationMessageBuilder initiationBuilder = new InitiationMessageBuilder();
  51         final TlvsBuilder tlvsBuilder = new TlvsBuilder();
  52         tlvsBuilder.setStringInformation(ImmutableList.of());
  53         parseTlvs(tlvsBuilder, bytes);
  54
  55         if (tlvsBuilder.getDescriptionTlv() == null || tlvsBuilder.getDescriptionTlv().getDescription() == null) {
  56             throw new BmpDeserializationException("Inclusion of sysDescr TLV is mandatory.");
  57         }
  58         if (tlvsBuilder.getNameTlv() == null || tlvsBuilder.getNameTlv().getName() == null) {
  59             throw new BmpDeserializationException("Inclusion of sysName TLV is mandatory.");
  60         }
  61
  62         return initiationBuilder.setTlvs(tlvsBuilder.build()).build();
  63     }
  64
  65     @Override
  66     public int getBmpMessageType() {
  67         return MESSAGE_TYPE;
  68     }
  69
  70     private void serializeTlvs(final Tlvs tlvs, final ByteBuf output) {
  71         serializeTlv(tlvs.getNameTlv(), output);
  72         serializeTlv(tlvs.getDescriptionTlv(), output);
  73         if (tlvs.getStringInformation() != null) {
  74             for (final StringInformation stringInfo : tlvs.getStringInformation()) {
  75                 if (stringInfo.getStringTlv() != null) {
  76                     serializeTlv(stringInfo.getStringTlv(), output);
  77                 }
  78             }
  79         }
  80     }
  81
  82     @Override
  83     protected void addTlv(final TlvsBuilder builder, final Tlv tlv) {
  84         if (tlv instanceof DescriptionTlv) {
  85             builder.setDescriptionTlv((DescriptionTlv) tlv);
  86         } else if (tlv instanceof NameTlv) {
  87             builder.setNameTlv((NameTlv) tlv);
  88         } else if (tlv instanceof StringTlv) {
  89             builder.setStringInformation(ImmutableList.<StringInformation>builder()
  90                     .addAll(builder.getStringInformation())
  91                     .add(new StringInformationBuilder().setStringTlv(new StringTlvBuilder((StringTlv) tlv).build()).build()).build());
  92         }
  93     }
  94 }