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.BGPSession;
  19 import org.opendaylight.protocol.bgp.parser.BGPSessionListener;
  20 import org.opendaylight.protocol.bgp.parser.message.BGPKeepAliveMessage;
  21 import org.opendaylight.protocol.bgp.parser.message.BGPOpenMessage;
  22 import org.opendaylight.protocol.bgp.parser.parameter.MultiprotocolCapability;
  23 import org.opendaylight.protocol.concepts.ListenerRegistration;
  24
  25 import com.google.common.collect.Sets;
  26 import com.google.common.eventbus.EventBus;
  27 import com.google.common.eventbus.Subscribe;
  28
  29 /**
  30  * This class has @Subscribe annotated methods which receive events from {@link EventBus} . Events are produced by
  31  * {@link BGPMock}, and each instance notifies exactly one {@link BGPSessionListener}.
  32  */
  33 class EventBusRegistration implements ListenerRegistration<BGPSessionListener> {
  34         private final EventBus eventBus;
  35         private final BGPSessionListener listener;
  36         @GuardedBy("this")
  37         private boolean closed = false;
  38
  39         public static EventBusRegistration createAndRegister(final EventBus eventBus, final BGPSessionListener listener,
  40                         final List<BGPMessage> allPreviousMessages) {
  41                 final EventBusRegistration instance = new EventBusRegistration(eventBus, listener, allPreviousMessages);
  42                 eventBus.register(instance);
  43                 return instance;
  44         }
  45
  46         private EventBusRegistration(final EventBus eventBus, final BGPSessionListener listener, final List<BGPMessage> allPreviousMessages) {
  47                 this.eventBus = eventBus;
  48                 this.listener = listener;
  49                 for (final BGPMessage message : allPreviousMessages) {
  50                         sendMessage(listener, message);
  51                 }
  52         }
  53
  54         @Subscribe
  55         public void onMessage(final BGPMessage message) {
  56                 sendMessage(this.listener, message);
  57         }
  58
  59         @Override
  60         public synchronized void close() {
  61                 if (this.closed) {
  62                         return;
  63                 }
  64                 this.eventBus.unregister(this);
  65                 this.closed = true;
  66         }
  67
  68         private static void sendMessage(final BGPSessionListener listener, final BGPMessage message) {
  69                 if (BGPMock.connectionLostMagicMessage.equals(message)) {
  70                         listener.onSessionTerminated(null, null);
  71                 } else if (message instanceof BGPOpenMessage) {
  72                         final Set<BGPTableType> tts = Sets.newHashSet();
  73                         for (final BGPParameter param : ((BGPOpenMessage) message).getOptParams()) {
  74                                 if (param instanceof MultiprotocolCapability) {
  75                                         tts.add(((MultiprotocolCapability) param).getTableType());
  76                                 }
  77                         }
  78
  79                         listener.onSessionUp(new BGPSession() {
  80
  81                                 @Override
  82                                 public void close() {
  83                                         // TODO Auto-generated method stub
  84
  85                                 }
  86
  87                                 @Override
  88                                 public Set<BGPTableType> getAdvertisedTableTypes() {
  89                                         return tts;
  90                                 }
  91                         });
  92                 } else if (message instanceof BGPKeepAliveMessage) {
  93                         // do nothing
  94                 } else {
  95                         listener.onMessage(null, message);
  96                 }
  97         }
  98
  99         @Override
 100         public BGPSessionListener getListener() {
 101                 return this.listener;
 102         }
 103 }