2 * Copyright (c) 2013 Cisco Systems, Inc. and others. All rights reserved.
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
8 package org.opendaylight.protocol.bgp.rib.mock;
10 import java.util.List;
13 import org.opendaylight.protocol.bgp.parser.BGPSession;
14 import org.opendaylight.protocol.bgp.parser.BGPSessionListener;
15 import org.opendaylight.protocol.bgp.parser.BGPTableType;
16 import org.opendaylight.protocol.concepts.ListenerRegistration;
17 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.Keepalive;
18 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.Open;
19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.open.BgpParameters;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130918.open.bgp.parameters.CParameters;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130918.open.bgp.parameters.c.parameters.CMultiprotocol;
22 import org.opendaylight.yangtools.yang.binding.Notification;
24 import com.google.common.collect.Sets;
25 import com.google.common.eventbus.EventBus;
26 import com.google.common.eventbus.Subscribe;
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}.
32 class EventBusRegistration extends ListenerRegistration<BGPSessionListener> {
33 private final EventBus eventBus;
35 public static EventBusRegistration createAndRegister(final EventBus eventBus, final BGPSessionListener listener,
36 final List<Notification> allPreviousMessages) {
37 final EventBusRegistration instance = new EventBusRegistration(eventBus, listener, allPreviousMessages);
38 eventBus.register(instance);
42 private EventBusRegistration(final EventBus eventBus, final BGPSessionListener listener, final List<Notification> allPreviousMessages) {
44 this.eventBus = eventBus;
45 for (final Notification message : allPreviousMessages) {
46 sendMessage(listener, message);
51 public void onMessage(final Notification message) {
52 sendMessage(this.listener, message);
56 public synchronized void removeRegistration() {
57 this.eventBus.unregister(this);
60 private static void sendMessage(final BGPSessionListener listener, final Notification message) {
61 if (BGPMock.connectionLostMagicMessage.equals(message)) {
62 listener.onSessionTerminated(null, null);
63 } else if (message instanceof Open) {
64 final Set<BGPTableType> tts = Sets.newHashSet();
65 for (final BgpParameters param : ((Open) message).getBgpParameters()) {
66 if (param instanceof CParameters) {
67 final CParameters p = (CParameters) param;
68 final BGPTableType type = new BGPTableType(((CMultiprotocol) p).getMultiprotocolCapability().getAfi(), ((CMultiprotocol) p).getMultiprotocolCapability().getSafi());
73 listener.onSessionUp(new BGPSession() {
77 // TODO Auto-generated method stub
82 public Set<BGPTableType> getAdvertisedTableTypes() {
86 } else if (message instanceof Keepalive) {
89 listener.onMessage(null, message);