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.Uninterruptibles;
12 import java.util.List;
13 import java.util.concurrent.CountDownLatch;
14 import java.util.concurrent.TimeUnit;
15 import javax.annotation.concurrent.GuardedBy;
16 import org.junit.Assert;
17 import org.opendaylight.protocol.bgp.rib.spi.BGPSession;
18 import org.opendaylight.protocol.bgp.rib.spi.BGPSessionListener;
19 import org.opendaylight.protocol.bgp.rib.spi.BGPTerminationReason;
20 import org.opendaylight.protocol.bgp.rib.spi.State;
21 import org.opendaylight.protocol.util.CheckUtil.ListenerCheck;
22 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bgp.rib.rev130925.rib.TablesKey;
23 import org.opendaylight.yangtools.yang.binding.Notification;
24 import org.slf4j.Logger;
25 import org.slf4j.LoggerFactory;
28 * Listener for the client.
30 public final class SimpleSessionListener implements BGPSessionListener, ListenerCheck {
32 private static final Logger LOG = LoggerFactory.getLogger(SimpleSessionListener.class);
34 private final List<Notification> listMsg = Lists.newArrayList();
35 private BGPSession session;
36 private final CountDownLatch sessionLatch = new CountDownLatch(1);
38 public SimpleSessionListener() {
41 synchronized List<Notification> getListMsg() {
46 public void markUptodate(final TablesKey tablesKey) {
47 LOG.debug("Table marked as up-to-date {}", tablesKey);
51 public void onSessionUp(final BGPSession session) {
52 LOG.info("Session Up");
53 this.session = session;
54 this.sessionLatch.countDown();
58 public void onSessionDown(final BGPSession session, final Exception e) {
59 LOG.debug("Session Down", e);
63 public void onSessionTerminated(final BGPSession session, final BGPTerminationReason cause) {
64 LOG.debug("Session terminated. Cause : {}", cause.toString());
68 public synchronized void onMessage(final BGPSession session, final Notification message) {
69 this.listMsg.add(message);
70 LOG.debug("Message received: {}", message);
74 public void releaseConnection() {
75 LOG.debug("Releasing connection");
76 if (this.session != null) {
79 } catch (final Exception e) {
80 LOG.warn("Error closing session", e);
85 public State getState() {
86 return getSession().getState();
89 BGPSessionImpl getSession() {
90 Assert.assertEquals("Session up", true, Uninterruptibles.awaitUninterruptibly(this.sessionLatch, 10, TimeUnit.SECONDS));
91 return (BGPSessionImpl) this.session;
95 public int getListMessageSize() {
96 return this.listMsg.size();