bgp/bmp-impl/src/main/java/org/opendaylight/protocol/bmp/impl/app/RouterSessionManager.java

   1 /*
   2  * Copyright (c) 2015 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
   9 package org.opendaylight.protocol.bmp.impl.app;
  10
  11 import com.google.common.base.Preconditions;
  12 import java.util.Map;
  13 import java.util.concurrent.ConcurrentHashMap;
  14 import org.opendaylight.controller.md.sal.dom.api.DOMDataBroker;
  15 import org.opendaylight.protocol.bgp.rib.spi.RIBExtensionConsumerContext;
  16 import org.opendaylight.protocol.bmp.api.BmpSessionListener;
  17 import org.opendaylight.protocol.bmp.api.BmpSessionListenerFactory;
  18 import org.opendaylight.protocol.bmp.impl.spi.BmpRouter;
  19 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.monitor.rev150512.RouterId;
  20 import org.opendaylight.yang.gen.v1.urn.opendaylight.params.xml.ns.yang.bmp.monitor.rev150512.routers.Router;
  21 import org.opendaylight.yangtools.binding.data.codec.api.BindingCodecTree;
  22 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  23 import org.slf4j.Logger;
  24 import org.slf4j.LoggerFactory;
  25
  26 final class RouterSessionManager implements BmpSessionListenerFactory, AutoCloseable {
  27
  28     private static final Logger LOG = LoggerFactory.getLogger(RouterSessionManager.class);
  29
  30     private final Map<RouterId, BmpRouter> sessionListeners = new ConcurrentHashMap<>();
  31
  32     private final YangInstanceIdentifier yangRoutersId;
  33     private final DOMDataBroker domDataBroker;
  34     private final RIBExtensionConsumerContext extensions;
  35     private final BindingCodecTree tree;
  36
  37     public RouterSessionManager(final YangInstanceIdentifier yangMonitorId, final DOMDataBroker domDataBroker,
  38             final RIBExtensionConsumerContext extensions, final BindingCodecTree tree) {
  39         this.domDataBroker = domDataBroker;
  40         this.yangRoutersId = YangInstanceIdentifier.builder(yangMonitorId).node(Router.QNAME).build();
  41         this.extensions = extensions;
  42         this.tree = tree;
  43     }
  44
  45     @Override
  46     public BmpSessionListener getSessionListener() {
  47         return new BmpRouterImpl(this);
  48     }
  49
  50     private synchronized boolean isSessionExist(final BmpRouter sessionListener) {
  51         Preconditions.checkNotNull(sessionListener);
  52         return sessionListeners.containsKey(Preconditions.checkNotNull(sessionListener.getRouterId()));
  53     }
  54
  55     synchronized boolean addSessionListener(final BmpRouter sessionListener) {
  56         if (isSessionExist(sessionListener)) {
  57             LOG.warn("Session listener for router {} was already added.", sessionListener.getRouterId());
  58             return false;
  59         }
  60         this.sessionListeners.put(sessionListener.getRouterId(), sessionListener);
  61         return true;
  62     }
  63
  64     synchronized void removeSessionListener(final BmpRouter sessionListener) {
  65         if (!isSessionExist(sessionListener)) {
  66             LOG.warn("Session listener for router {} was already removed.", sessionListener.getRouterId());
  67             return;
  68         }
  69         this.sessionListeners.remove(sessionListener.getRouterId());
  70     }
  71
  72     @Override
  73     public void close() throws Exception {
  74         for (final BmpRouter sessionListener : this.sessionListeners.values()) {
  75             sessionListener.close();
  76         }
  77     }
  78
  79     YangInstanceIdentifier getRoutersYangIId() {
  80         return this.yangRoutersId;
  81     }
  82
  83     DOMDataBroker getDomDataBroker() {
  84         return this.domDataBroker;
  85     }
  86
  87     RIBExtensionConsumerContext getExtensions() {
  88         return extensions;
  89     }
  90
  91     BindingCodecTree getCodecTree() {
  92         return this.tree;
  93     }
  94
  95 }