bgp/rib-mock/src/main/java/org/opendaylight/protocol/bgp/rib/mock/EventBusRegistration.java

   1 /*
   2  * Copyright (c) 2013 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 package org.opendaylight.protocol.bgp.rib.mock;
   9
  10 import java.util.List;
  11 import java.util.Set;
  12
  13 import javax.annotation.concurrent.GuardedBy;
  14
  15 import org.opendaylight.protocol.bgp.concepts.BGPTableType;
  16 import org.opendaylight.protocol.bgp.parser.BGPMessage;
  17 import org.opendaylight.protocol.bgp.parser.BGPParameter;
  18 import org.opendaylight.protocol.bgp.parser.BGPSessionListener;
  19 import org.opendaylight.protocol.bgp.parser.message.BGPKeepAliveMessage;
  20 import org.opendaylight.protocol.bgp.parser.message.BGPOpenMessage;
  21 import org.opendaylight.protocol.bgp.parser.parameter.MultiprotocolCapability;
  22
  23 import org.opendaylight.protocol.concepts.ListenerRegistration;
  24 import com.google.common.collect.Sets;
  25 import com.google.common.eventbus.EventBus;
  26 import com.google.common.eventbus.Subscribe;
  27
  28 /**
  29  * This class has @Subscribe annotated methods which receive events from {@link EventBus} . Events are produced by
  30  * {@link BGPMock}, and each instance notifies exactly one {@link BGPSessionListener}.
  31  */
  32 class EventBusRegistration implements ListenerRegistration<BGPSessionListener> {
  33         private final EventBus eventBus;
  34         private final BGPSessionListener listener;
  35         @GuardedBy("this")
  36         private boolean closed = false;
  37
  38         public static EventBusRegistration createAndRegister(final EventBus eventBus, final BGPSessionListener listener,
  39                         final List<BGPMessage> allPreviousMessages) {
  40                 final EventBusRegistration instance = new EventBusRegistration(eventBus, listener, allPreviousMessages);
  41                 eventBus.register(instance);
  42                 return instance;
  43         }
  44
  45         private EventBusRegistration(final EventBus eventBus, final BGPSessionListener listener, final List<BGPMessage> allPreviousMessages) {
  46                 this.eventBus = eventBus;
  47                 this.listener = listener;
  48                 for (final BGPMessage message : allPreviousMessages)
  49                         sendMessage(listener, message);
  50         }
  51
  52         @Subscribe
  53         public void onMessage(final BGPMessage message) {
  54                 sendMessage(this.listener, message);
  55         }
  56
  57         @Override
  58         public synchronized void close() {
  59                 if (this.closed)
  60                         return;
  61                 this.eventBus.unregister(this);
  62                 this.closed = true;
  63         }
  64
  65         private static void sendMessage(final BGPSessionListener listener, final BGPMessage message) {
  66                 if (BGPMock.connectionLostMagicMessage.equals(message)) {
  67                         listener.onSessionTerminated(null);
  68                 } else if (message instanceof BGPOpenMessage) {
  69                         final Set<BGPTableType> tts = Sets.newHashSet();
  70                         for (final BGPParameter param : ((BGPOpenMessage) message).getOptParams()) {
  71                                 if (param instanceof MultiprotocolCapability) {
  72                                         tts.add(((MultiprotocolCapability) param).getTableType());
  73                                 }
  74                         }
  75                         listener.onSessionUp(tts);
  76                 } else if (message instanceof BGPKeepAliveMessage) {
  77                         // do nothing
  78                 } else {
  79                         listener.onMessage(message);
  80                 }
  81         }
  82
  83         @Override
  84         public BGPSessionListener getListener() {
  85                 return this.listener;
  86         }
  87 }