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 com.google.common.collect.Sets;
11 import com.google.common.eventbus.EventBus;
12 import com.google.common.eventbus.Subscribe;
13 import java.util.List;
15 import org.opendaylight.protocol.bgp.parser.BgpTableTypeImpl;
16 import org.opendaylight.protocol.bgp.rib.spi.BGPSession;
17 import org.opendaylight.protocol.bgp.rib.spi.BGPSessionListener;
18 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.AsNumber;
19 import org.opendaylight.yang.gen.v1.urn.ietf.params.xml.ns.yang.ietf.inet.types.rev100924.Ipv4Address;
20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.Keepalive;
21 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.Open;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.message.rev130919.open.BgpParameters;
23 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.BgpTableType;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.multiprotocol.rev130919.open.bgp.parameters.c.parameters.MultiprotocolCase;
25 import org.opendaylight.yangtools.concepts.AbstractListenerRegistration;
26 import org.opendaylight.yangtools.yang.binding.Notification;
27 import org.slf4j.Logger;
28 import org.slf4j.LoggerFactory;
31 * This class has @Subscribe annotated methods which receive events from {@link EventBus} . Events are produced by
32 * {@link BGPMock}, and each instance notifies exactly one {@link BGPSessionListener}.
34 final class EventBusRegistration extends AbstractListenerRegistration<BGPSessionListener> {
36 private static final Logger LOG = LoggerFactory.getLogger(EventBusRegistration.class);
38 private final EventBus eventBus;
40 public static EventBusRegistration createAndRegister(final EventBus eventBus, final BGPSessionListener listener,
41 final List<Notification> allPreviousMessages) {
42 final EventBusRegistration instance = new EventBusRegistration(eventBus, listener, allPreviousMessages);
43 eventBus.register(instance);
47 private EventBusRegistration(final EventBus eventBus, final BGPSessionListener listener, final List<Notification> allPreviousMessages) {
49 this.eventBus = eventBus;
50 for (final Notification message : allPreviousMessages) {
51 sendMessage(listener, message);
56 public void onMessage(final Notification message) {
57 sendMessage(this.getInstance(), message);
61 public synchronized void removeRegistration() {
62 this.eventBus.unregister(this);
65 private static void sendMessage(final BGPSessionListener listener, final Notification message) {
66 if (BGPMock.CONNECTION_LOST_MAGIC_MSG.equals(message)) {
67 listener.onSessionTerminated(null, null);
68 } else if (message instanceof Open) {
69 final Set<BgpTableType> tts = Sets.newHashSet();
70 for (final BgpParameters param : ((Open) message).getBgpParameters()) {
71 if (param.getCParameters() instanceof MultiprotocolCase) {
72 final MultiprotocolCase p = (MultiprotocolCase) param.getCParameters();
73 LOG.debug("Adding open parameter {}", p);
74 final BgpTableType type = new BgpTableTypeImpl(p.getMultiprotocolCapability().getAfi(), p.getMultiprotocolCapability().getSafi());
79 listener.onSessionUp(new BGPSession() {
81 private static final long AS = 30L;
85 LOG.debug("Session {} closed", this);
89 public Set<BgpTableType> getAdvertisedTableTypes() {
94 public Ipv4Address getBgpId() {
95 return new Ipv4Address("127.0.0.1");
99 public AsNumber getAsNumber() {
100 return new AsNumber(AS);
103 } else if (!(message instanceof Keepalive)) {
104 listener.onMessage(null, message);