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