bmp/bmp-parser-impl/src/main/java/org/opendaylight/protocol/bmp/parser/message/TerminationHandler.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.rev171207.TerminationMessage;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev171207.TerminationMessageBuilder;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev171207.Tlv;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev171207.reason.tlv.ReasonTlv;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev171207.string.informations.StringInformation;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev171207.string.informations.StringInformationBuilder;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev171207.string.tlv.StringTlv;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev171207.string.tlv.StringTlvBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev171207.termination.Tlvs;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev171207.termination.TlvsBuilder;
  27 import org.opendaylight.yangtools.yang.binding.Notification;
  28
  29 public class TerminationHandler extends AbstractBmpMessageWithTlvParser<TlvsBuilder> {
  30
  31     public TerminationHandler(final BmpTlvRegistry tlvRegistry) {
  32         super(tlvRegistry);
  33     }
  34
  35     private static final int MESSAGE_TYPE = 5;
  36
  37     @Override
  38     public Notification parseMessageBody(final ByteBuf bytes) throws BmpDeserializationException {
  39         final TerminationMessageBuilder terminationMessage = new TerminationMessageBuilder();
  40         final TlvsBuilder tlvsBuilder = new TlvsBuilder();
  41         tlvsBuilder.setStringInformation(ImmutableList.of());
  42         parseTlvs(tlvsBuilder, bytes);
  43         if (tlvsBuilder.getReasonTlv() == null || tlvsBuilder.getReasonTlv().getReason() == null) {
  44             throw new BmpDeserializationException("Inclusion of Reason TLV is mandatory.");
  45         }
  46         return terminationMessage.setTlvs(tlvsBuilder.build()).build();
  47     }
  48
  49
  50     @Override
  51     public void serializeMessageBody(final Notification message, final ByteBuf buffer) {
  52         Preconditions.checkArgument(message instanceof TerminationMessage,
  53                 "An instance of Termination message is required");
  54         final TerminationMessage terminationMsg = (TerminationMessage) message;
  55         serializeTlvs(terminationMsg.getTlvs(), buffer);
  56     }
  57
  58     @Override
  59     public int getBmpMessageType() {
  60         return MESSAGE_TYPE;
  61     }
  62
  63     protected void serializeTlvs(final Tlvs tlvs, final ByteBuf output) {
  64         serializeTlv(tlvs.getReasonTlv(), output);
  65         if (tlvs.getStringInformation() != null) {
  66             for (final StringInformation stringInfo : tlvs.getStringInformation()) {
  67                 if (stringInfo.getStringTlv() != null) {
  68                     serializeTlv(stringInfo.getStringTlv(), output);
  69                 }
  70             }
  71         }
  72     }
  73
  74     @Override
  75     protected void addTlv(final TlvsBuilder builder, final Tlv tlv) {
  76         if (tlv instanceof ReasonTlv) {
  77             builder.setReasonTlv((ReasonTlv) tlv);
  78         } else if (tlv instanceof StringTlv) {
  79             builder.setStringInformation(ImmutableList.<StringInformation>builder()
  80                     .addAll(builder.getStringInformation())
  81                     .add(new StringInformationBuilder()
  82                             .setStringTlv(new StringTlvBuilder((StringTlv) tlv).build()).build()).build());
  83         }
  84     }
  85 }