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