Merge changes Ic434bf4a,I05a3fb18,I47a3783d,I8234bbfd
[controller.git] / opendaylight / netconf / config-netconf-connector / src / main / java / org / opendaylight / controller / netconf / confignetconfconnector / mapping / attributes / resolving / ObjectResolver.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.attributes.resolving;
10
11 import com.google.common.base.Preconditions;
12 import com.google.common.collect.Maps;
13 import java.util.Map;
14 import java.util.Map.Entry;
15 import javax.management.openmbean.ArrayType;
16 import javax.management.openmbean.CompositeType;
17 import javax.management.openmbean.OpenType;
18 import javax.management.openmbean.SimpleType;
19 import org.opendaylight.controller.config.yangjmxgenerator.attribute.AttributeIfc;
20 import org.opendaylight.controller.config.yangjmxgenerator.attribute.ListAttribute;
21 import org.opendaylight.controller.config.yangjmxgenerator.attribute.ListDependenciesAttribute;
22 import org.opendaylight.controller.config.yangjmxgenerator.attribute.TOAttribute;
23 import org.opendaylight.controller.netconf.confignetconfconnector.mapping.attributes.AttributeIfcSwitchStatement;
24 import org.opendaylight.controller.netconf.confignetconfconnector.mapping.config.ServiceRegistryWrapper;
25
26 public class ObjectResolver extends AttributeIfcSwitchStatement<AttributeResolvingStrategy<?, ? extends OpenType<?>>> {
27
28     private final ServiceRegistryWrapper serviceTracker;
29
30     public ObjectResolver(ServiceRegistryWrapper serviceTracker) {
31         this.serviceTracker = serviceTracker;
32     }
33
34     public Map<String, AttributeResolvingStrategy<?, ? extends OpenType<?>>> prepareResolving(
35             Map<String, AttributeIfc> configDefinition) {
36         Map<String, AttributeResolvingStrategy<?, ? extends OpenType<?>>> strategies = Maps.newHashMap();
37
38         for (Entry<String, AttributeIfc> attrEntry : configDefinition.entrySet()) {
39             strategies.put(attrEntry.getKey(),
40                     prepareStrategy(attrEntry.getValue()));
41         }
42
43         return strategies;
44     }
45
46     private AttributeResolvingStrategy<?, ? extends OpenType<?>> prepareStrategy(AttributeIfc attributeIfc) {
47
48         return switchAttribute(attributeIfc);
49     }
50
51     private Map<String, String> createYangToJmxMapping(TOAttribute attributeIfc) {
52         Map<String, String> retVal = Maps.newHashMap();
53         for (Entry<String, AttributeIfc> entry : attributeIfc.getYangPropertiesToTypesMap().entrySet()) {
54             retVal.put(entry.getKey(), (entry.getValue()).getLowerCaseCammelCase());
55         }
56         return retVal;
57     }
58
59     @Override
60     protected AttributeResolvingStrategy<?, ? extends OpenType<?>>  caseJavaSimpleAttribute(SimpleType<?> openType) {
61         return new SimpleAttributeResolvingStrategy(openType);
62     }
63
64     @Override
65     protected AttributeResolvingStrategy<?, ? extends OpenType<?>>  caseJavaArrayAttribute(ArrayType<?> openType) {
66         SimpleType<?> innerType = (SimpleType<?>) openType.getElementOpenType();
67         AttributeResolvingStrategy<?, ? extends OpenType<?>> strat = new SimpleAttributeResolvingStrategy(innerType);
68         return new ArrayAttributeResolvingStrategy(strat, openType);
69     }
70
71     @Override
72     protected AttributeResolvingStrategy<?, ? extends OpenType<?>>  caseJavaCompositeAttribute(CompositeType openType) {
73         Map<String, AttributeResolvingStrategy<?, ? extends OpenType<?>>> innerMap = Maps.newHashMap();
74         Map<String, String> yangToJmxMapping = Maps.newHashMap();
75
76         fillMappingForComposite(openType, innerMap, yangToJmxMapping);
77         return new CompositeAttributeResolvingStrategy(innerMap, openType, yangToJmxMapping);
78     }
79
80     private void fillMappingForComposite(CompositeType openType, Map<String, AttributeResolvingStrategy<?, ? extends OpenType<?>>> innerMap, Map<String, String> yangToJmxMapping) {
81         for (String innerAttributeKey : openType.keySet()) {
82             innerMap.put(innerAttributeKey, caseJavaAttribute(openType.getType(innerAttributeKey)));
83             yangToJmxMapping.put(innerAttributeKey, innerAttributeKey);
84         }
85     }
86
87     @Override
88     protected AttributeResolvingStrategy<?, ? extends OpenType<?>> caseJavaUnionAttribute(OpenType<?> openType) {
89
90         Preconditions.checkState(openType instanceof CompositeType, "Unexpected open type, expected %s but was %s");
91         CompositeType compositeType = (CompositeType) openType;
92
93         Map<String, AttributeResolvingStrategy<?, ? extends OpenType<?>>> innerMap = Maps.newHashMap();
94         Map<String, String> yangToJmxMapping = Maps.newHashMap();
95         fillMappingForComposite(compositeType, innerMap, yangToJmxMapping);
96
97         return new UnionCompositeAttributeResolvingStrategy(innerMap, compositeType, yangToJmxMapping);
98     }
99
100     @Override
101     protected AttributeResolvingStrategy<?, ? extends OpenType<?>> caseDependencyAttribute(
102             SimpleType<?> openType) {
103         return new ObjectNameAttributeResolvingStrategy(serviceTracker);
104     }
105
106     @Override
107     protected AttributeResolvingStrategy<?, ? extends OpenType<?>> caseTOAttribute(CompositeType openType) {
108         Preconditions.checkState(getLastAttribute() instanceof TOAttribute);
109         TOAttribute toAttribute = (TOAttribute) getLastAttribute();
110
111         Map<String, AttributeResolvingStrategy<?, ? extends OpenType<?>>> innerMap = Maps.newHashMap();
112
113         for (String innerName : openType.keySet()) {
114
115             AttributeIfc innerAttributeIfc = toAttribute.getJmxPropertiesToTypesMap().get(innerName);
116             innerMap.put(innerAttributeIfc.getAttributeYangName(),
117                     prepareStrategy(innerAttributeIfc));
118         }
119         return new CompositeAttributeResolvingStrategy(innerMap, openType, createYangToJmxMapping(toAttribute));
120     }
121
122     @Override
123     protected AttributeResolvingStrategy<?, ? extends OpenType<?>> caseListAttribute(ArrayType<?> openType) {
124         Preconditions.checkState(getLastAttribute() instanceof ListAttribute);
125         AttributeIfc innerAttribute = ((ListAttribute) getLastAttribute()).getInnerAttribute();
126         return new ArrayAttributeResolvingStrategy(prepareStrategy(innerAttribute), openType);
127     }
128
129     @Override
130     protected AttributeResolvingStrategy<?, ? extends OpenType<?>> caseListDependeciesAttribute(ArrayType<?> openType) {
131         Preconditions.checkState(getLastAttribute() instanceof ListDependenciesAttribute);
132         return new ArrayAttributeResolvingStrategy(caseDependencyAttribute(SimpleType.OBJECTNAME), openType);
133     }
134
135 }