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.connect.netconf;
10 import com.google.common.base.Function;
11 import com.google.common.base.Optional;
12 import com.google.common.base.Preconditions;
13 import com.google.common.collect.Collections2;
14 import com.google.common.collect.Lists;
15 import com.google.common.collect.Sets;
16 import com.google.common.util.concurrent.CheckedFuture;
17 import com.google.common.util.concurrent.FutureCallback;
18 import com.google.common.util.concurrent.Futures;
19 import com.google.common.util.concurrent.ListenableFuture;
20 import com.google.common.util.concurrent.ListeningExecutorService;
21 import com.google.common.util.concurrent.MoreExecutors;
22 import java.util.Collection;
23 import java.util.LinkedList;
24 import java.util.List;
26 import java.util.concurrent.Callable;
27 import java.util.concurrent.ExecutorService;
28 import org.opendaylight.controller.netconf.api.NetconfMessage;
29 import org.opendaylight.controller.sal.connect.api.MessageTransformer;
30 import org.opendaylight.controller.sal.connect.api.RemoteDevice;
31 import org.opendaylight.controller.sal.connect.api.RemoteDeviceCommunicator;
32 import org.opendaylight.controller.sal.connect.api.RemoteDeviceHandler;
33 import org.opendaylight.controller.sal.connect.netconf.listener.NetconfSessionCapabilities;
34 import org.opendaylight.controller.sal.connect.netconf.sal.NetconfDeviceRpc;
35 import org.opendaylight.controller.sal.connect.netconf.schema.NetconfRemoteSchemaYangSourceProvider;
36 import org.opendaylight.controller.sal.connect.util.RemoteDeviceId;
37 import org.opendaylight.yangtools.yang.common.QName;
38 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
39 import org.opendaylight.yangtools.yang.model.repo.api.MissingSchemaSourceException;
40 import org.opendaylight.yangtools.yang.model.repo.api.SchemaContextFactory;
41 import org.opendaylight.yangtools.yang.model.repo.api.SchemaResolutionException;
42 import org.opendaylight.yangtools.yang.model.repo.api.SchemaSourceRepresentation;
43 import org.opendaylight.yangtools.yang.model.repo.api.SourceIdentifier;
44 import org.opendaylight.yangtools.yang.model.repo.api.YangTextSchemaSource;
45 import org.opendaylight.yangtools.yang.model.repo.spi.PotentialSchemaSource;
46 import org.opendaylight.yangtools.yang.model.repo.spi.SchemaSourceRegistration;
47 import org.opendaylight.yangtools.yang.model.repo.spi.SchemaSourceRegistry;
48 import org.slf4j.Logger;
49 import org.slf4j.LoggerFactory;
52 * This is a mediator between NetconfDeviceCommunicator and NetconfDeviceSalFacade
54 public final class NetconfDevice implements RemoteDevice<NetconfSessionCapabilities, NetconfMessage> {
56 private static final Logger logger = LoggerFactory.getLogger(NetconfDevice.class);
58 public static final Function<QName, SourceIdentifier> QNAME_TO_SOURCE_ID_FUNCTION = new Function<QName, SourceIdentifier>() {
60 public SourceIdentifier apply(final QName input) {
61 return new SourceIdentifier(input.getLocalName(), Optional.fromNullable(input.getFormattedRevision()));
65 private final RemoteDeviceId id;
67 private final SchemaContextFactory schemaContextFactory;
68 private final RemoteDeviceHandler<NetconfSessionCapabilities> salFacade;
69 private final ListeningExecutorService processingExecutor;
70 private final SchemaSourceRegistry schemaRegistry;
71 private final MessageTransformer<NetconfMessage> messageTransformer;
72 private final NetconfStateSchemas.NetconfStateSchemasResolver stateSchemasResolver;
73 private final NotificationHandler notificationHandler;
74 private final List<SchemaSourceRegistration<? extends SchemaSourceRepresentation>> sourceRegistrations = Lists.newArrayList();
76 public NetconfDevice(final SchemaResourcesDTO schemaResourcesDTO, final RemoteDeviceId id, final RemoteDeviceHandler<NetconfSessionCapabilities> salFacade,
77 final ExecutorService globalProcessingExecutor, final MessageTransformer<NetconfMessage> messageTransformer) {
79 this.schemaRegistry = schemaResourcesDTO.getSchemaRegistry();
80 this.messageTransformer = messageTransformer;
81 this.schemaContextFactory = schemaResourcesDTO.getSchemaContextFactory();
82 this.salFacade = salFacade;
83 this.stateSchemasResolver = schemaResourcesDTO.getStateSchemasResolver();
84 this.processingExecutor = MoreExecutors.listeningDecorator(globalProcessingExecutor);
85 this.notificationHandler = new NotificationHandler(salFacade, messageTransformer, id);
89 public void onRemoteSessionUp(final NetconfSessionCapabilities remoteSessionCapabilities,
90 final RemoteDeviceCommunicator<NetconfMessage> listener) {
91 // SchemaContext setup has to be performed in a dedicated thread since
92 // we are in a netty thread in this method
93 // Yang models are being downloaded in this method and it would cause a
94 // deadlock if we used the netty thread
95 // http://netty.io/wiki/thread-model.html
96 logger.debug("{}: Session to remote device established with {}", id, remoteSessionCapabilities);
98 final NetconfDeviceRpc deviceRpc = setUpDeviceRpc(listener);
100 final DeviceSourcesResolver task = new DeviceSourcesResolver(deviceRpc, remoteSessionCapabilities, id, stateSchemasResolver);
101 final ListenableFuture<DeviceSources> sourceResolverFuture = processingExecutor.submit(task);
103 final FutureCallback<DeviceSources> resolvedSourceCallback = new FutureCallback<DeviceSources>() {
105 public void onSuccess(final DeviceSources result) {
106 addProvidedSourcesToSchemaRegistry(deviceRpc, result);
110 private void setUpSchema(final DeviceSources result) {
111 processingExecutor.submit(new RecursiveSchemaSetup(result, remoteSessionCapabilities, deviceRpc, listener));
115 public void onFailure(final Throwable t) {
116 logger.warn("{}: Unexpected error resolving device sources: {}", id, t);
117 handleSalInitializationFailure(t, listener);
121 Futures.addCallback(sourceResolverFuture, resolvedSourceCallback);
124 private void handleSalInitializationSuccess(final SchemaContext result, final NetconfSessionCapabilities remoteSessionCapabilities, final NetconfDeviceRpc deviceRpc) {
125 updateMessageTransformer(result);
126 salFacade.onDeviceConnected(result, remoteSessionCapabilities, deviceRpc);
127 notificationHandler.onRemoteSchemaUp();
129 logger.debug("{}: Initialization in sal successful", id);
130 logger.info("{}: Netconf connector initialized successfully", id);
133 private void handleSalInitializationFailure(final Throwable t, final RemoteDeviceCommunicator<NetconfMessage> listener) {
134 logger.error("{}: Initialization in sal failed, disconnecting from device", id, t);
136 onRemoteSessionDown();
140 * Update initial message transformer to use retrieved schema
141 * @param currentSchemaContext
143 private void updateMessageTransformer(final SchemaContext currentSchemaContext) {
144 messageTransformer.onGlobalContextUpdated(currentSchemaContext);
147 private void addProvidedSourcesToSchemaRegistry(final NetconfDeviceRpc deviceRpc, final DeviceSources deviceSources) {
148 final NetconfRemoteSchemaYangSourceProvider yangProvider = new NetconfRemoteSchemaYangSourceProvider(id, deviceRpc);
149 for (final SourceIdentifier sourceId : deviceSources.getProvidedSources()) {
150 sourceRegistrations.add(schemaRegistry.registerSchemaSource(yangProvider,
151 PotentialSchemaSource.create(sourceId, YangTextSchemaSource.class, PotentialSchemaSource.Costs.REMOTE_IO.getValue())));
155 private NetconfDeviceRpc setUpDeviceRpc(final RemoteDeviceCommunicator<NetconfMessage> listener) {
156 return new NetconfDeviceRpc(listener, messageTransformer);
160 public void onRemoteSessionDown() {
161 salFacade.onDeviceDisconnected();
162 for (final SchemaSourceRegistration<? extends SchemaSourceRepresentation> sourceRegistration : sourceRegistrations) {
163 sourceRegistration.close();
168 public void onNotification(final NetconfMessage notification) {
169 notificationHandler.handleNotification(notification);
173 * Just a transfer object containing schema related dependencies. Injected in constructor.
175 public static class SchemaResourcesDTO {
176 private final SchemaSourceRegistry schemaRegistry;
177 private final SchemaContextFactory schemaContextFactory;
178 private final NetconfStateSchemas.NetconfStateSchemasResolver stateSchemasResolver;
180 public SchemaResourcesDTO(final SchemaSourceRegistry schemaRegistry, final SchemaContextFactory schemaContextFactory, final NetconfStateSchemas.NetconfStateSchemasResolver stateSchemasResolver) {
181 this.schemaRegistry = Preconditions.checkNotNull(schemaRegistry);
182 this.schemaContextFactory = Preconditions.checkNotNull(schemaContextFactory);
183 this.stateSchemasResolver = Preconditions.checkNotNull(stateSchemasResolver);
186 public SchemaSourceRegistry getSchemaRegistry() {
187 return schemaRegistry;
190 public SchemaContextFactory getSchemaContextFactory() {
191 return schemaContextFactory;
194 public NetconfStateSchemas.NetconfStateSchemasResolver getStateSchemasResolver() {
195 return stateSchemasResolver;
200 * Schema building callable.
202 private static class DeviceSourcesResolver implements Callable<DeviceSources> {
203 private final NetconfDeviceRpc deviceRpc;
204 private final NetconfSessionCapabilities remoteSessionCapabilities;
205 private final RemoteDeviceId id;
206 private final NetconfStateSchemas.NetconfStateSchemasResolver stateSchemasResolver;
208 public DeviceSourcesResolver(final NetconfDeviceRpc deviceRpc, final NetconfSessionCapabilities remoteSessionCapabilities, final RemoteDeviceId id, final NetconfStateSchemas.NetconfStateSchemasResolver stateSchemasResolver) {
209 this.deviceRpc = deviceRpc;
210 this.remoteSessionCapabilities = remoteSessionCapabilities;
212 this.stateSchemasResolver = stateSchemasResolver;
216 public DeviceSources call() throws Exception {
218 final Set<SourceIdentifier> requiredSources = Sets.newHashSet(Collections2.transform(
219 remoteSessionCapabilities.getModuleBasedCaps(), QNAME_TO_SOURCE_ID_FUNCTION));
221 // If monitoring is not supported, we will still attempt to create schema, sources might be already provided
222 final NetconfStateSchemas availableSchemas = stateSchemasResolver.resolve(deviceRpc, remoteSessionCapabilities, id);
223 logger.debug("{}: Schemas exposed by ietf-netconf-monitoring: {}", id, availableSchemas.getAvailableYangSchemasQNames());
225 final Set<SourceIdentifier> providedSources = Sets.newHashSet(Collections2.transform(
226 availableSchemas.getAvailableYangSchemasQNames(), QNAME_TO_SOURCE_ID_FUNCTION));
228 final Set<SourceIdentifier> requiredSourcesNotProvided = Sets.difference(requiredSources, providedSources);
230 if (!requiredSourcesNotProvided.isEmpty()) {
231 logger.warn("{}: Netconf device does not provide all yang models reported in hello message capabilities, required but not provided: {}",
232 id, requiredSourcesNotProvided);
233 logger.warn("{}: Attempting to build schema context from required sources", id);
237 // TODO should we perform this ? We have a mechanism to fix initialization of devices not reporting or required modules in hello
238 // That is overriding capabilities in configuration using attribute yang-module-capabilities
239 // This is more user friendly even though it clashes with attribute yang-module-capabilities
240 // Some devices do not report all required models in hello message, but provide them
241 final Set<SourceIdentifier> providedSourcesNotRequired = Sets.difference(providedSources, requiredSources);
242 if (!providedSourcesNotRequired.isEmpty()) {
243 logger.warn("{}: Netconf device provides additional yang models not reported in hello message capabilities: {}",
244 id, providedSourcesNotRequired);
245 logger.warn("{}: Adding provided but not required sources as required to prevent failures", id);
246 requiredSources.addAll(providedSourcesNotRequired);
249 return new DeviceSources(requiredSources, providedSources);
254 * Contains RequiredSources - sources from capabilities.
257 private static final class DeviceSources {
258 private final Collection<SourceIdentifier> requiredSources;
259 private final Collection<SourceIdentifier> providedSources;
261 public DeviceSources(final Collection<SourceIdentifier> requiredSources, final Collection<SourceIdentifier> providedSources) {
262 this.requiredSources = requiredSources;
263 this.providedSources = providedSources;
266 public Collection<SourceIdentifier> getRequiredSources() {
267 return requiredSources;
270 public Collection<SourceIdentifier> getProvidedSources() {
271 return providedSources;
277 * Schema builder that tries to build schema context from provided sources or biggest subset of it.
279 private final class RecursiveSchemaSetup implements Runnable {
280 private final DeviceSources deviceSources;
281 private final NetconfSessionCapabilities remoteSessionCapabilities;
282 private final NetconfDeviceRpc deviceRpc;
283 private final RemoteDeviceCommunicator<NetconfMessage> listener;
285 public RecursiveSchemaSetup(final DeviceSources deviceSources, final NetconfSessionCapabilities remoteSessionCapabilities, final NetconfDeviceRpc deviceRpc, final RemoteDeviceCommunicator<NetconfMessage> listener) {
286 this.deviceSources = deviceSources;
287 this.remoteSessionCapabilities = remoteSessionCapabilities;
288 this.deviceRpc = deviceRpc;
289 this.listener = listener;
294 setUpSchema(deviceSources.getRequiredSources());
298 * Recursively build schema context, in case of success or final failure notify device
300 private void setUpSchema(final Collection<SourceIdentifier> requiredSources) {
301 logger.trace("{}: Trying to build schema context from {}", id, requiredSources);
303 // If no more sources, fail
304 if(requiredSources.isEmpty()) {
305 handleSalInitializationFailure(new IllegalStateException(id + ": No more sources for schema context"), listener);
309 final CheckedFuture<SchemaContext, SchemaResolutionException> schemaBuilderFuture = schemaContextFactory.createSchemaContext(requiredSources);
311 final FutureCallback<SchemaContext> RecursiveSchemaBuilderCallback = new FutureCallback<SchemaContext>() {
314 public void onSuccess(final SchemaContext result) {
315 logger.debug("{}: Schema context built successfully from {}", id, requiredSources);
316 handleSalInitializationSuccess(result, remoteSessionCapabilities, deviceRpc);
320 public void onFailure(final Throwable t) {
321 // In case source missing, try without it
322 if (t instanceof MissingSchemaSourceException) {
323 final SourceIdentifier missingSource = ((MissingSchemaSourceException) t).getSourceId();
324 logger.warn("{}: Unable to build schema context, missing source {}, will reattempt without it", id, missingSource);
325 setUpSchema(stripMissingSource(requiredSources, missingSource));
327 // In case resolution error, try only with resolved sources
328 } else if (t instanceof SchemaResolutionException) {
329 // TODO check for infinite loop
330 final SchemaResolutionException resolutionException = (SchemaResolutionException) t;
331 logger.warn("{}: Unable to build schema context, unsatisfied imports {}, will reattempt with resolved only", id, resolutionException.getUnsatisfiedImports());
332 setUpSchema(resolutionException.getResolvedSources());
333 // unknown error, fail
335 handleSalInitializationFailure(t, listener);
340 Futures.addCallback(schemaBuilderFuture, RecursiveSchemaBuilderCallback);
343 private Collection<SourceIdentifier> stripMissingSource(final Collection<SourceIdentifier> requiredSources, final SourceIdentifier sIdToRemove) {
344 final LinkedList<SourceIdentifier> sourceIdentifiers = Lists.newLinkedList(requiredSources);
345 final boolean removed = sourceIdentifiers.remove(sIdToRemove);
346 Preconditions.checkState(removed, "{}: Trying to remove {} from {} failed", id, sIdToRemove, requiredSources);
347 return sourceIdentifiers;