bmp/bmp-parser-impl/src/main/java/org/opendaylight/protocol/bmp/parser/message/RouteMirroringMessageHandler.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 io.netty.buffer.ByteBuf;
  13 import io.netty.buffer.Unpooled;
  14 import org.opendaylight.protocol.bgp.parser.spi.MessageRegistry;
  15 import org.opendaylight.protocol.bmp.spi.parser.AbstractBmpPerPeerMessageParser;
  16 import org.opendaylight.protocol.bmp.spi.parser.BmpDeserializationException;
  17 import org.opendaylight.protocol.bmp.spi.parser.BmpTlvRegistry;
  18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.OpenBuilder;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev180329.UpdateBuilder;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.RouteMirroringMessage;
  21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.RouteMirroringMessageBuilder;
  22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.Tlv;
  23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.mirror.Tlvs;
  24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.mirror.TlvsBuilder;
  25 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.mirror.information.tlv.MirrorInformationTlv;
  26 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.mirror.pdu.tlvs.PduOpenTlv;
  27 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.message.rev180329.mirror.pdu.tlvs.PduUpdateTlv;
  28 import org.opendaylight.yangtools.yang.binding.Notification;
  29
  30 public class RouteMirroringMessageHandler extends AbstractBmpPerPeerMessageParser<TlvsBuilder> {
  31
  32     private static final int MESSAGE_TYPE = 6;
  33
  34     public RouteMirroringMessageHandler(final MessageRegistry bgpMssageRegistry, final BmpTlvRegistry tlvRegistry) {
  35         super(bgpMssageRegistry, tlvRegistry);
  36     }
  37
  38     @Override
  39     public void serializeMessageBody(final Notification message, final ByteBuf buffer) {
  40         super.serializeMessageBody(message, buffer);
  41         Preconditions.checkArgument(message instanceof RouteMirroringMessage,
  42                 "An instance of RouteMirroringMessage is required");
  43         final RouteMirroringMessage routeMirror = (RouteMirroringMessage) message;
  44         serializeTlvs(routeMirror.getTlvs(), buffer);
  45     }
  46
  47     @Override
  48     public Notification parseMessageBody(final ByteBuf bytes) throws BmpDeserializationException {
  49         final RouteMirroringMessageBuilder routeMirror = new RouteMirroringMessageBuilder()
  50                 .setPeerHeader(parsePerPeerHeader(bytes));
  51         final TlvsBuilder tlvsBuilder = new TlvsBuilder();
  52         parseTlvs(tlvsBuilder, bytes);
  53         return routeMirror.setTlvs(tlvsBuilder.build()).build();
  54     }
  55
  56     protected void serializeTlvs(final Tlvs tlvs, final ByteBuf output) {
  57         final ByteBuf tlvsBuffer = Unpooled.buffer();
  58         if (tlvs.getMirrorInformationTlv() != null) {
  59             serializeTlv(tlvs.getMirrorInformationTlv(), tlvsBuffer);
  60         }
  61         if (tlvs.getPduUpdateTlv() != null) {
  62             getBgpMessageRegistry().serializeMessage(new UpdateBuilder(tlvs.getPduUpdateTlv()).build(), tlvsBuffer);
  63         }
  64         if (tlvs.getPduOpenTlv() != null) {
  65             getBgpMessageRegistry().serializeMessage(new OpenBuilder(tlvs.getPduOpenTlv()).build(), tlvsBuffer);
  66         }
  67
  68         output.writeBytes(tlvsBuffer);
  69     }
  70
  71
  72     @Override
  73     public int getBmpMessageType() {
  74         return MESSAGE_TYPE;
  75     }
  76
  77     @Override
  78     protected void addTlv(final TlvsBuilder builder, final Tlv tlv) {
  79         if (tlv instanceof MirrorInformationTlv) {
  80             builder.setMirrorInformationTlv((MirrorInformationTlv) tlv);
  81         } else if (tlv instanceof PduUpdateTlv) {
  82             builder.setPduUpdateTlv((PduUpdateTlv) tlv);
  83         } else if (tlv instanceof PduOpenTlv) {
  84             builder.setPduOpenTlv((PduOpenTlv) tlv);
  85         }
  86     }
  87 }