Merge "Added support for persistance of service references to netconf."
[controller.git] / opendaylight / netconf / config-netconf-connector / src / main / java / org / opendaylight / controller / netconf / confignetconfconnector / mapping / runtime / Runtime.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.confignetconfconnector.mapping.runtime;
10
11 import com.google.common.collect.HashMultimap;
12 import com.google.common.collect.Maps;
13 import com.google.common.collect.Multimap;
14 import org.opendaylight.controller.config.api.ServiceReferenceReadableRegistry;
15 import org.opendaylight.controller.config.api.jmx.ObjectNameUtil;
16 import org.opendaylight.controller.netconf.confignetconfconnector.mapping.config.Config;
17 import org.opendaylight.controller.netconf.confignetconfconnector.mapping.config.ModuleConfig;
18 import org.opendaylight.controller.netconf.confignetconfconnector.mapping.config.Services;
19 import org.opendaylight.controller.netconf.util.xml.XmlNetconfConstants;
20 import org.opendaylight.controller.netconf.util.xml.XmlUtil;
21 import org.w3c.dom.Document;
22 import org.w3c.dom.Element;
23
24 import javax.management.ObjectName;
25 import java.util.Collection;
26 import java.util.Map;
27 import java.util.Set;
28
29 public class Runtime {
30
31     private final Map<String, Map<String, ModuleRuntime>> moduleRuntimes;
32     private final Map<String, Map<String, ModuleConfig>> moduleConfigs;
33
34     public Runtime(Map<String, Map<String, ModuleRuntime>> moduleRuntimes,
35             Map<String, Map<String, ModuleConfig>> moduleConfigs) {
36         this.moduleRuntimes = moduleRuntimes;
37         this.moduleConfigs = moduleConfigs;
38     }
39
40     private Map<String, Multimap<String, ObjectName>> mapInstancesToModules(Set<ObjectName> instancesToMap) {
41         Map<String, Multimap<String, ObjectName>> retVal = Maps.newHashMap();
42
43         // TODO map to namepsace, prevent module name conflicts
44         // this code does not support same module names from different namespaces
45         // Namespace should be present in ObjectName
46
47         for (ObjectName objectName : instancesToMap) {
48             String moduleName = ObjectNameUtil.getFactoryName(objectName);
49
50             Multimap<String, ObjectName> multimap = retVal.get(moduleName);
51             if (multimap == null) {
52                 multimap = HashMultimap.create();
53                 retVal.put(moduleName, multimap);
54             }
55
56             String instanceName = ObjectNameUtil.getInstanceName(objectName);
57
58             multimap.put(instanceName, objectName);
59         }
60
61         return retVal;
62     }
63
64     public Element toXml(Set<ObjectName> instancesToMap, Set<ObjectName> configBeans, Document document, ServiceReferenceReadableRegistry serviceRegistry) {
65         Services serviceTracker = new Services(serviceRegistry);
66
67         Element root = document.createElement(XmlNetconfConstants.DATA_KEY);
68
69         Element modulesElement = document.createElement(XmlNetconfConstants.MODULES_KEY);
70         XmlUtil.addNamespaceAttr(modulesElement,
71                 XmlNetconfConstants.URN_OPENDAYLIGHT_PARAMS_XML_NS_YANG_CONTROLLER_CONFIG);
72         root.appendChild(modulesElement);
73
74         Map<String, Multimap<String, ObjectName>> moduleToRuntimeInstance = mapInstancesToModules(instancesToMap);
75         Map<String, Map<String, Collection<ObjectName>>> moduleToConfigInstance = Config.getMappedInstances(
76                 configBeans, moduleConfigs);
77
78         for (String localNamespace : moduleConfigs.keySet()) {
79
80             Map<String, Collection<ObjectName>> instanceToMbe = moduleToConfigInstance.get(localNamespace);
81
82             for (String moduleName : moduleConfigs.get(localNamespace).keySet()) {
83                 Multimap<String, ObjectName> instanceToRbe = moduleToRuntimeInstance.get(moduleName);
84
85                 for (ObjectName instanceON : instanceToMbe.get(moduleName)) {
86                     String instanceName = ObjectNameUtil.getInstanceName(instanceON);
87
88                     Element runtimeXml;
89                     ModuleConfig moduleConfig = moduleConfigs.get(localNamespace).get(moduleName);
90                     if(instanceToRbe==null || instanceToRbe.containsKey(instanceName) == false) {
91                         runtimeXml = moduleConfig.toXml(instanceON, serviceTracker, document, localNamespace);
92                     } else {
93                         ModuleRuntime moduleRuntime = moduleRuntimes.get(localNamespace).get(moduleName);
94                         runtimeXml = moduleRuntime.toXml(localNamespace, instanceToRbe.get(instanceName), document,
95                                 moduleConfig, instanceON, serviceTracker);
96                     }
97                     modulesElement.appendChild(runtimeXml);
98                 }
99
100             }
101         }
102
103         return root;
104     }
105
106     private ObjectName findInstance(Collection<ObjectName> objectNames, String instanceName) {
107         for (ObjectName objectName : objectNames) {
108             String name = ObjectNameUtil.getInstanceName(objectName);
109             if(name.equals(instanceName))
110                 return objectName;
111         }
112
113         throw new UnsupportedOperationException("Unable to find config bean instance under name " + instanceName + " among " + objectNames);
114     }
115
116 }