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