Merge "Refactor LeaderElectionScenarioTests"
[controller.git] / opendaylight / netconf / mdsal-netconf-connector / src / main / java / org / opendaylight / controller / netconf / mdsal / connector / MdsalNetconfOperationServiceFactory.java
1 /*
2  * Copyright (c) 2013 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
9 package org.opendaylight.controller.netconf.mdsal.connector;
10
11 import com.google.common.base.Preconditions;
12 import java.util.HashSet;
13 import java.util.Set;
14 import org.opendaylight.controller.md.sal.dom.api.DOMDataBroker;
15 import org.opendaylight.controller.netconf.api.Capability;
16 import org.opendaylight.controller.netconf.api.monitoring.CapabilityListener;
17 import org.opendaylight.controller.netconf.mapping.api.NetconfOperationServiceFactory;
18 import org.opendaylight.controller.netconf.util.capability.BasicCapability;
19 import org.opendaylight.controller.netconf.util.capability.YangModuleCapability;
20 import org.opendaylight.controller.sal.core.api.model.SchemaService;
21 import org.opendaylight.yangtools.yang.model.api.Module;
22 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
23 import org.slf4j.Logger;
24 import org.slf4j.LoggerFactory;
25
26 public class MdsalNetconfOperationServiceFactory implements NetconfOperationServiceFactory, AutoCloseable {
27
28     private static final Logger LOG = LoggerFactory.getLogger(MdsalNetconfOperationServiceFactory.class);
29
30     private final DOMDataBroker dataBroker;
31     private final CurrentSchemaContext currentSchemaContext;
32
33     public MdsalNetconfOperationServiceFactory(final SchemaService schemaService, final DOMDataBroker domDataBroker) {
34         this.currentSchemaContext = new CurrentSchemaContext(Preconditions.checkNotNull(schemaService));
35         this.dataBroker = Preconditions.checkNotNull(domDataBroker);
36     }
37
38     @Override
39     public MdsalNetconfOperationService createService(final String netconfSessionIdForReporting) {
40         return new MdsalNetconfOperationService(currentSchemaContext, netconfSessionIdForReporting, dataBroker);
41     }
42
43     @Override
44     public void close() throws Exception {
45         currentSchemaContext.close();
46     }
47
48     @Override
49     public Set<Capability> getCapabilities() {
50         return transformCapabilities(currentSchemaContext.getCurrentContext());
51     }
52
53     static Set<Capability> transformCapabilities(final SchemaContext currentContext1) {
54         final Set<Capability> capabilities = new HashSet<>();
55         // [RFC6241] 8.3.  Candidate Configuration Capability
56         capabilities.add(new BasicCapability("urn:ietf:params:netconf:capability:candidate:1.0"));
57
58         final SchemaContext currentContext = currentContext1;
59         final Set<Module> modules = currentContext.getModules();
60         for (final Module module : modules) {
61             if(currentContext.getModuleSource(module).isPresent()) {
62                 capabilities.add(new YangModuleCapability(module, currentContext.getModuleSource(module).get()));
63             } else {
64                 LOG.warn("Missing source for module {}. This module will not be available from netconf server",
65                         module);
66             }
67         }
68
69         return capabilities;
70     }
71
72     @Override
73     public AutoCloseable registerCapabilityListener(final CapabilityListener listener) {
74         return currentSchemaContext.registerCapabilityListener(listener);
75     }
76
77 }