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.mdsal.connector;
11 import com.google.common.base.Optional;
12 import com.google.common.base.Preconditions;
13 import com.google.common.io.CharStreams;
14 import java.io.IOException;
15 import java.io.InputStream;
16 import java.io.InputStreamReader;
17 import java.nio.charset.StandardCharsets;
18 import java.util.HashSet;
20 import org.opendaylight.controller.config.util.capability.Capability;
21 import org.opendaylight.controller.config.util.capability.YangModuleCapability;
22 import org.opendaylight.controller.md.sal.dom.api.DOMDataBroker;
23 import org.opendaylight.controller.md.sal.dom.api.DOMRpcService;
24 import org.opendaylight.controller.sal.core.api.model.SchemaService;
25 import org.opendaylight.netconf.api.monitoring.CapabilityListener;
26 import org.opendaylight.netconf.mapping.api.NetconfOperationServiceFactory;
27 import org.opendaylight.netconf.mapping.api.NetconfOperationServiceFactoryListener;
28 import org.opendaylight.yangtools.yang.common.SimpleDateFormatUtil;
29 import org.opendaylight.yangtools.yang.model.api.Module;
30 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
31 import org.opendaylight.yangtools.yang.model.repo.api.SchemaSourceException;
32 import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier;
33 import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
34 import org.opendaylight.yangtools.yang.model.repo.spi.SchemaSourceProvider;
35 import org.slf4j.Logger;
36 import org.slf4j.LoggerFactory;
38 public class MdsalNetconfOperationServiceFactory implements NetconfOperationServiceFactory, AutoCloseable {
40 private static final Logger LOG = LoggerFactory.getLogger(MdsalNetconfOperationServiceFactory.class);
42 private final DOMDataBroker dataBroker;
43 private final DOMRpcService rpcService;
45 private final CurrentSchemaContext currentSchemaContext;
46 private final SchemaSourceProvider<YangTextSchemaSource> rootSchemaSourceProviderDependency;
47 private final NetconfOperationServiceFactoryListener netconfOperationServiceFactoryListener;
49 public MdsalNetconfOperationServiceFactory(final SchemaService schemaService,
50 final SchemaSourceProvider<YangTextSchemaSource> rootSchemaSourceProviderDependency,
51 final NetconfOperationServiceFactoryListener netconfOperationServiceFactoryListener,
52 final DOMDataBroker dataBroker,
53 final DOMRpcService rpcService) {
55 this.dataBroker = dataBroker;
56 this.rpcService = rpcService;
58 this.rootSchemaSourceProviderDependency = rootSchemaSourceProviderDependency;
59 this.currentSchemaContext = new CurrentSchemaContext(Preconditions.checkNotNull(schemaService), rootSchemaSourceProviderDependency);
60 this.netconfOperationServiceFactoryListener = netconfOperationServiceFactoryListener;
61 this.netconfOperationServiceFactoryListener.onAddNetconfOperationServiceFactory(this);
65 public MdsalNetconfOperationService createService(final String netconfSessionIdForReporting) {
66 Preconditions.checkState(dataBroker != null, "MD-SAL provider not yet initialized");
67 return new MdsalNetconfOperationService(currentSchemaContext, netconfSessionIdForReporting, dataBroker, rpcService);
73 currentSchemaContext.close();
74 if (netconfOperationServiceFactoryListener != null) {
75 netconfOperationServiceFactoryListener.onRemoveNetconfOperationServiceFactory(this);
77 } catch(Exception e) {
78 LOG.error("Failed to close resources correctly - ignore", e);
83 public Set<Capability> getCapabilities() {
84 return transformCapabilities(currentSchemaContext.getCurrentContext(), rootSchemaSourceProviderDependency);
87 static Set<Capability> transformCapabilities(final SchemaContext currentContext, final SchemaSourceProvider<YangTextSchemaSource> rootSchemaSourceProviderDependency) {
88 final Set<Capability> capabilities = new HashSet<>();
90 // Added by netconf-impl by default
91 // capabilities.add(new BasicCapability("urn:ietf:params:netconf:capability:candidate:1.0"));
93 final Set<Module> modules = currentContext.getModules();
94 for (final Module module : modules) {
95 Optional<YangModuleCapability> cap = moduleToCapability(module, rootSchemaSourceProviderDependency);
97 capabilities.add(cap.get());
99 for (final Module submodule : module.getSubmodules()) {
100 cap = moduleToCapability(submodule, rootSchemaSourceProviderDependency);
101 if(cap.isPresent()) {
102 capabilities.add(cap.get());
110 private static Optional<YangModuleCapability> moduleToCapability(
111 final Module module, final SchemaSourceProvider<YangTextSchemaSource> rootSchemaSourceProviderDependency) {
113 final SourceIdentifier moduleSourceIdentifier = SourceIdentifier.create(module.getName(),
114 (SimpleDateFormatUtil.DEFAULT_DATE_REV == module.getRevision() ? Optional.absent() :
115 Optional.of(module.getQNameModule().getFormattedRevision())));
117 InputStream sourceStream = null;
120 sourceStream = rootSchemaSourceProviderDependency.getSource(moduleSourceIdentifier).checkedGet().openStream();
121 source = CharStreams.toString(new InputStreamReader(sourceStream, StandardCharsets.UTF_8));
122 } catch (IOException | SchemaSourceException e) {
123 LOG.warn("Ignoring source for module {}. Unable to read content", moduleSourceIdentifier, e);
128 if (sourceStream != null) {
129 sourceStream.close();
131 } catch (IOException e) {
132 LOG.warn("Error closing yang source stream {}. Ignoring", moduleSourceIdentifier, e);
136 return Optional.of(new YangModuleCapability(module, source));
138 LOG.warn("Missing source for module {}. This module will not be available from netconf server",
139 moduleSourceIdentifier);
141 return Optional.absent();
145 public AutoCloseable registerCapabilityListener(final CapabilityListener listener) {
146 return currentSchemaContext.registerCapabilityListener(listener);