opendaylight/md-sal/sal-dom-broker/src/main/java/org/opendaylight/controller/sal/dom/broker/ConsumerContextImpl.java

   1 /*
   2  * Copyright (c) 2014 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 package org.opendaylight.controller.sal.dom.broker;
   9
  10 import com.google.common.base.Optional;
  11 import com.google.common.base.Preconditions;
  12 import com.google.common.collect.ClassToInstanceMap;
  13 import com.google.common.collect.MutableClassToInstanceMap;
  14 import java.util.Collection;
  15 import java.util.concurrent.atomic.AtomicBoolean;
  16 import org.opendaylight.controller.sal.core.api.Broker.ConsumerSession;
  17 import org.opendaylight.controller.sal.core.api.BrokerService;
  18 import org.opendaylight.controller.sal.core.api.Consumer;
  19 import org.opendaylight.controller.sal.dom.broker.osgi.AbstractBrokerServiceProxy;
  20 import org.opendaylight.controller.sal.dom.broker.osgi.ProxyFactory;
  21
  22 class ConsumerContextImpl implements ConsumerSession {
  23
  24     private final ClassToInstanceMap<BrokerService> instantiatedServices = MutableClassToInstanceMap.create();
  25     private final Consumer consumer;
  26
  27     private final BrokerImpl broker;
  28     private final AtomicBoolean closed = new AtomicBoolean(false);
  29
  30     ConsumerContextImpl(final Consumer provider, final BrokerImpl brokerImpl) {
  31         broker = brokerImpl;
  32         consumer = provider;
  33     }
  34
  35     @Override
  36     public <T extends BrokerService> T getService(final Class<T> service) {
  37         checkNotClosed();
  38         final T localProxy = instantiatedServices.getInstance(service);
  39         if (localProxy != null) {
  40             return localProxy;
  41         }
  42         final Optional<T> serviceImpl = broker.getGlobalService(service);
  43         if (serviceImpl.isPresent()) {
  44             final T ret = ProxyFactory.createProxy(null, serviceImpl.get());
  45             instantiatedServices.putInstance(service, ret);
  46             return ret;
  47         } else {
  48             return null;
  49         }
  50     }
  51
  52     @Override
  53     public void close() {
  54         if (closed.compareAndSet(false, true)) {
  55             Collection<BrokerService> toStop = instantiatedServices.values();
  56             for (BrokerService brokerService : toStop) {
  57                 if (brokerService instanceof AbstractBrokerServiceProxy<?>) {
  58                     ((AbstractBrokerServiceProxy<?>) brokerService).close();
  59                 }
  60             }
  61             broker.consumerSessionClosed(this);
  62         }
  63     }
  64
  65
  66     @Override
  67     public boolean isClosed() {
  68         return closed.get();
  69     }
  70
  71     /**
  72      * Gets broker.
  73      *
  74      * @return the broker
  75      */
  76     protected final BrokerImpl getBrokerChecked() {
  77         checkNotClosed();
  78         return broker;
  79     }
  80
  81     /**
  82      * Gets consumer.
  83      *
  84      * @return the _consumer
  85      */
  86     public Consumer getConsumer() {
  87         return consumer;
  88     }
  89
  90     protected final void checkNotClosed() {
  91         Preconditions.checkState(!closed.get(), "Session is closed.");
  92     }
  93 }