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
9 package org.opendaylight.netconf.impl.osgi;
11 import com.google.common.collect.HashMultimap;
12 import com.google.common.collect.ImmutableSet;
13 import com.google.common.collect.ImmutableSet.Builder;
14 import com.google.common.collect.Maps;
15 import com.google.common.collect.Multimap;
16 import com.google.common.collect.Multimaps;
17 import com.google.common.collect.Sets;
18 import io.netty.util.internal.ConcurrentSet;
19 import java.util.HashSet;
20 import java.util.List;
23 import org.opendaylight.controller.config.util.capability.Capability;
24 import org.opendaylight.netconf.mapping.api.NetconfOperation;
25 import org.opendaylight.netconf.mapping.api.NetconfOperationService;
26 import org.opendaylight.netconf.mapping.api.NetconfOperationServiceFactory;
27 import org.opendaylight.netconf.mapping.api.NetconfOperationServiceFactoryListener;
28 import org.opendaylight.netconf.util.CloseableUtil;
29 import org.opendaylight.netconf.api.monitoring.CapabilityListener;
30 import org.slf4j.Logger;
31 import org.slf4j.LoggerFactory;
34 * NetconfOperationService aggregator. Makes a collection of operation services accessible as one.
36 public class AggregatedNetconfOperationServiceFactory implements NetconfOperationServiceFactory, NetconfOperationServiceFactoryListener, AutoCloseable {
38 private static final Logger LOG = LoggerFactory.getLogger(AggregatedNetconfOperationServiceFactory.class);
40 private final Set<NetconfOperationServiceFactory> factories = new ConcurrentSet<>();
41 private final Multimap<NetconfOperationServiceFactory, AutoCloseable> registrations = Multimaps.synchronizedMultimap(HashMultimap.create());
42 private final Set<CapabilityListener> listeners = new ConcurrentSet<>();
44 public AggregatedNetconfOperationServiceFactory() {
47 public AggregatedNetconfOperationServiceFactory(final List<NetconfOperationServiceFactory> mappers) {
48 mappers.forEach(this::onAddNetconfOperationServiceFactory);
52 public synchronized void onAddNetconfOperationServiceFactory(NetconfOperationServiceFactory service) {
53 factories.add(service);
55 for (final CapabilityListener listener : listeners) {
56 AutoCloseable reg = service.registerCapabilityListener(listener);
57 registrations.put(service, reg);
62 public synchronized void onRemoveNetconfOperationServiceFactory(NetconfOperationServiceFactory service) {
63 factories.remove(service);
65 for (final AutoCloseable autoCloseable : registrations.get(service)) {
67 autoCloseable.close();
68 } catch (Exception e) {
69 LOG.warn("Unable to close listener registration", e);
73 registrations.removeAll(service);
77 public Set<Capability> getCapabilities() {
78 final HashSet<Capability> capabilities = Sets.newHashSet();
79 for (final NetconfOperationServiceFactory factory : factories) {
80 capabilities.addAll(factory.getCapabilities());
86 public synchronized AutoCloseable registerCapabilityListener(final CapabilityListener listener) {
87 final Map<NetconfOperationServiceFactory, AutoCloseable> regs = Maps.newHashMap();
89 for (final NetconfOperationServiceFactory factory : factories) {
90 final AutoCloseable reg = factory.registerCapabilityListener(listener);
91 regs.put(factory, reg);
93 listeners.add(listener);
95 return new AutoCloseable() {
97 public void close() throws Exception {
98 synchronized (AggregatedNetconfOperationServiceFactory.this) {
99 listeners.remove(listener);
100 CloseableUtil.closeAll(regs.values());
101 for (final Map.Entry<NetconfOperationServiceFactory, AutoCloseable> reg : regs.entrySet()) {
102 registrations.remove(reg.getKey(), reg.getValue());
110 public synchronized NetconfOperationService createService(final String netconfSessionIdForReporting) {
111 return new AggregatedNetconfOperation(factories, netconfSessionIdForReporting);
115 public synchronized void close() throws Exception {
117 for (AutoCloseable reg : registrations.values()) {
120 registrations.clear();
124 private static final class AggregatedNetconfOperation implements NetconfOperationService {
126 private final Set<NetconfOperationService> services;
128 public AggregatedNetconfOperation(final Set<NetconfOperationServiceFactory> factories, final String netconfSessionIdForReporting) {
129 final Builder<NetconfOperationService> b = ImmutableSet.builder();
130 for (final NetconfOperationServiceFactory factory : factories) {
131 b.add(factory.createService(netconfSessionIdForReporting));
133 this.services = b.build();
137 public Set<NetconfOperation> getNetconfOperations() {
138 final HashSet<NetconfOperation> operations = Sets.newHashSet();
139 for (final NetconfOperationService service : services) {
140 operations.addAll(service.getNetconfOperations());
146 public void close() {
148 CloseableUtil.closeAll(services);
149 } catch (final Exception e) {
150 throw new IllegalStateException("Unable to properly close all aggregated services", e);