2 * Copyright (c) 2014 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.controller.sal.dom.broker;
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;
23 class ConsumerContextImpl implements ConsumerSession {
25 private final ClassToInstanceMap<BrokerService> instantiatedServices = MutableClassToInstanceMap.create();
26 private final Consumer consumer;
28 private final BrokerImpl broker;
29 private final AtomicBoolean closed = new AtomicBoolean(false);
31 ConsumerContextImpl(final Consumer provider, final BrokerImpl brokerImpl) {
37 public <T extends BrokerService> T getService(final Class<T> service) {
39 final T localProxy = instantiatedServices.getInstance(service);
40 if (localProxy != null) {
43 final Optional<T> serviceImpl = broker.getGlobalService(service);
44 if (serviceImpl.isPresent()) {
45 final T ret = ProxyFactory.createProxy(null, serviceImpl.get());
46 instantiatedServices.putInstance(service, ret);
55 if (closed.compareAndSet(false, true)) {
56 Collection<BrokerService> toStop = instantiatedServices.values();
57 for (BrokerService brokerService : toStop) {
58 if (brokerService instanceof AbstractBrokerServiceProxy<?>) {
59 ((AbstractBrokerServiceProxy<?>) brokerService).close();
62 broker.consumerSessionClosed(this);
68 public boolean isClosed() {
77 protected final BrokerImpl getBrokerChecked() {
85 * @return the _consumer
87 public Consumer getConsumer() {
91 protected final void checkNotClosed() {
92 Preconditions.checkState(!closed.get(), "Session is closed.");