Add support for configuration attributes based on yang leaves with custom types ...
[controller.git] / opendaylight / netconf / config-netconf-connector / src / main / java / org / opendaylight / controller / netconf / confignetconfconnector / mapping / attributes / fromxml / ObjectXmlReader.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.fromxml;
10
11 import com.google.common.base.Preconditions;
12 import com.google.common.collect.Maps;
13 import org.opendaylight.controller.config.yangjmxgenerator.attribute.AttributeIfc;
14 import org.opendaylight.controller.config.yangjmxgenerator.attribute.ListAttribute;
15 import org.opendaylight.controller.config.yangjmxgenerator.attribute.TOAttribute;
16 import org.opendaylight.controller.netconf.confignetconfconnector.mapping.attributes.AttributeIfcSwitchStatement;
17
18 import javax.management.openmbean.ArrayType;
19 import javax.management.openmbean.CompositeType;
20 import javax.management.openmbean.SimpleType;
21 import java.util.Map;
22 import java.util.Map.Entry;
23
24 public class ObjectXmlReader extends AttributeIfcSwitchStatement<AttributeReadingStrategy> {
25
26     private String key;
27
28     public Map<String, AttributeReadingStrategy> prepareReading(Map<String, AttributeIfc> yangToAttrConfig) {
29         Map<String, AttributeReadingStrategy> strategies = Maps.newHashMap();
30
31         for (Entry<String, AttributeIfc> attributeEntry : yangToAttrConfig.entrySet()) {
32             AttributeReadingStrategy strat = prepareReadingStrategy(attributeEntry.getKey(), attributeEntry.getValue());
33             strategies.put(attributeEntry.getKey(), strat);
34         }
35         return strategies;
36     }
37
38     private AttributeReadingStrategy prepareReadingStrategy(String key, AttributeIfc attributeIfc) {
39         this.key = key;
40         return switchAttribute(attributeIfc);
41     }
42
43     @Override
44     public AttributeReadingStrategy caseJavaSimpleAttribute(SimpleType<?> openType) {
45         return new SimpleAttributeReadingStrategy(lastAttribute.getNullableDefault());
46     }
47
48     @Override
49     public AttributeReadingStrategy caseJavaArrayAttribute(ArrayType<?> openType) {
50         SimpleAttributeReadingStrategy innerStrategy = new SimpleAttributeReadingStrategy(lastAttribute.getNullableDefault());
51         return new ArrayAttributeReadingStrategy(lastAttribute.getNullableDefault(), innerStrategy);
52     }
53
54     @Override
55     public AttributeReadingStrategy caseJavaCompositeAttribute(CompositeType openType) {
56         Preconditions.checkState(openType.keySet().size() == 1, "Unexpected number of elements for open type %s, should be 1", openType);
57         String mappingKey = openType.keySet().iterator().next();
58         return new SimpleCompositeAttributeReadingStrategy(lastAttribute.getNullableDefault(), mappingKey);
59     }
60
61     @Override
62     protected AttributeReadingStrategy caseDependencyAttribute(SimpleType<?> openType) {
63         return new ObjectNameAttributeReadingStrategy(lastAttribute.getNullableDefault());
64     }
65
66     @Override
67     protected AttributeReadingStrategy caseTOAttribute(CompositeType openType) {
68         Preconditions.checkState(lastAttribute instanceof TOAttribute);
69         Map<String, AttributeIfc> inner = ((TOAttribute)lastAttribute).getYangPropertiesToTypesMap();
70
71         Map<String, AttributeReadingStrategy> innerStrategies = Maps.newHashMap();
72
73         for (Entry<String, AttributeIfc> innerAttrEntry : inner.entrySet()) {
74             AttributeReadingStrategy innerStrat = prepareReadingStrategy(innerAttrEntry.getKey(),
75                     innerAttrEntry.getValue());
76             innerStrategies.put(innerAttrEntry.getKey(), innerStrat);
77         }
78
79         return new CompositeAttributeReadingStrategy(lastAttribute.getNullableDefault(), innerStrategies);
80     }
81
82     @Override
83     protected AttributeReadingStrategy caseListAttribute(ArrayType<?> openType) {
84         Preconditions.checkState(lastAttribute instanceof ListAttribute);
85         AttributeReadingStrategy innerStrategy = prepareReadingStrategy(key, ((ListAttribute) lastAttribute).getInnerAttribute());
86         return new ArrayAttributeReadingStrategy(lastAttribute.getNullableDefault(), innerStrategy);
87     }
88
89 }