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.impl;
10 import com.google.common.collect.Lists;
11 import com.google.common.util.concurrent.Futures;
12 import com.google.common.util.concurrent.ListenableFuture;
13 import com.google.common.util.concurrent.Uninterruptibles;
14 import java.util.List;
15 import java.util.concurrent.CountDownLatch;
16 import java.util.concurrent.TimeUnit;
17 import javax.annotation.concurrent.GuardedBy;
18 import org.junit.Assert;
19 import org.opendaylight.protocol.bgp.rib.spi.BGPSession;
20 import org.opendaylight.protocol.bgp.rib.spi.BGPSessionListener;
21 import org.opendaylight.protocol.bgp.rib.spi.BGPTerminationReason;
22 import org.opendaylight.protocol.bgp.rib.spi.State;
23 import org.opendaylight.protocol.util.CheckUtil.ListenerCheck;
24 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev180329.rib.TablesKey;
25 import org.opendaylight.yangtools.yang.binding.Notification;
26 import org.slf4j.Logger;
27 import org.slf4j.LoggerFactory;
30 * Listener for the client.
32 public final class SimpleSessionListener implements BGPSessionListener, ListenerCheck {
34 private static final Logger LOG = LoggerFactory.getLogger(SimpleSessionListener.class);
36 private final List<Notification> listMsg = Lists.newArrayList();
37 private BGPSession session;
38 private final CountDownLatch sessionLatch = new CountDownLatch(1);
40 public SimpleSessionListener() {
43 synchronized List<Notification> getListMsg() {
48 public void markUptodate(final TablesKey tablesKey) {
49 LOG.debug("Table marked as up-to-date {}", tablesKey);
53 public void onSessionUp(final BGPSession session) {
54 LOG.info("Session Up");
55 this.session = session;
56 this.sessionLatch.countDown();
60 public void onSessionDown(final BGPSession session, final Exception e) {
61 LOG.debug("Session Down", e);
65 public void onSessionTerminated(final BGPSession session, final BGPTerminationReason cause) {
66 LOG.debug("Session terminated. Cause : {}", cause.toString());
70 public synchronized void onMessage(final BGPSession session, final Notification message) {
71 this.listMsg.add(message);
72 LOG.debug("Message received: {}", message);
76 public ListenableFuture<Void> releaseConnection() {
77 LOG.debug("Releasing connection");
78 if (this.session != null) {
81 } catch (final Exception e) {
82 LOG.warn("Error closing session", e);
85 return Futures.immediateFuture(null);
88 public State getState() {
89 return getSession().getState();
92 BGPSessionImpl getSession() {
93 Assert.assertEquals("Session up", true, Uninterruptibles.awaitUninterruptibly(this.sessionLatch, 10, TimeUnit.SECONDS));
94 return (BGPSessionImpl) this.session;
98 public int getListMessageSize() {
99 return this.listMsg.size();