Merge changes Ic434bf4a,I05a3fb18,I47a3783d,I8234bbfd
[controller.git] / opendaylight / netconf / config-netconf-connector / src / main / java / org / opendaylight / controller / netconf / confignetconfconnector / mapping / attributes / toxml / ObjectXmlWriter.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.toxml;
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.w3c.dom.Document;
25
26 public class ObjectXmlWriter extends AttributeIfcSwitchStatement<AttributeWritingStrategy> {
27
28     private Document document;
29     private String key;
30
31     public Map<String, AttributeWritingStrategy> prepareWriting(Map<String, AttributeIfc> yangToAttrConfig,
32             Document document) {
33
34         Map<String, AttributeWritingStrategy> preparedWriting = Maps.newHashMap();
35
36         for (Entry<String, AttributeIfc> mappedAttributeEntry : yangToAttrConfig.entrySet()) {
37             String key = mappedAttributeEntry.getKey();
38             AttributeIfc value = mappedAttributeEntry.getValue();
39             AttributeWritingStrategy strat = prepareWritingStrategy(key, value, document);
40             preparedWriting.put(key, strat);
41         }
42
43         return preparedWriting;
44     }
45
46     public AttributeWritingStrategy prepareWritingStrategy(String key, AttributeIfc expectedAttr, Document document) {
47         Preconditions.checkNotNull(expectedAttr, "Mbean attributes mismatch, unable to find expected attribute for %s",
48                 key);
49         this.document = document;
50         this.key = key;
51         return switchAttribute(expectedAttr);
52     }
53
54     @Override
55     protected AttributeWritingStrategy caseJavaBinaryAttribute(OpenType<?> openType) {
56         return new SimpleBinaryAttributeWritingStrategy(document, key);
57     }
58
59     @Override
60     protected AttributeWritingStrategy caseJavaSimpleAttribute(SimpleType<?> openType) {
61         return new SimpleAttributeWritingStrategy(document, key);
62     }
63
64     @Override
65     protected AttributeWritingStrategy caseJavaArrayAttribute(ArrayType<?> openType) {
66         AttributeWritingStrategy innerStrategy = new SimpleAttributeWritingStrategy(document, key);
67         return new ArrayAttributeWritingStrategy(innerStrategy);
68     }
69
70     @Override
71     protected AttributeWritingStrategy caseJavaIdentityRefAttribute(OpenType<?> openType) {
72         return new SimpleIdentityRefAttributeWritingStrategy(document, key);
73     }
74
75     @Override
76     protected AttributeWritingStrategy caseJavaCompositeAttribute(CompositeType openType) {
77         return new SimpleCompositeAttributeWritingStrategy(document, key);
78     }
79
80     @Override
81     protected AttributeWritingStrategy caseJavaUnionAttribute(OpenType<?> openType) {
82         return new SimpleUnionAttributeWritingStrategy(document, key);
83     }
84
85     @Override
86     protected AttributeWritingStrategy caseDependencyAttribute(SimpleType<?> openType) {
87         return new ObjectNameAttributeWritingStrategy(document, key);
88     }
89
90     @Override
91     protected AttributeWritingStrategy caseTOAttribute(CompositeType openType) {
92         Preconditions.checkState(getLastAttribute() instanceof TOAttribute);
93
94         Map<String, AttributeWritingStrategy> innerStrats = Maps.newHashMap();
95         String currentKey = key;
96         for (Entry<String, AttributeIfc> innerAttrEntry : ((TOAttribute) getLastAttribute()).getYangPropertiesToTypesMap().entrySet()) {
97
98             AttributeWritingStrategy innerStrategy = prepareWritingStrategy(innerAttrEntry.getKey(),
99                     innerAttrEntry.getValue(), document);
100             innerStrats.put(innerAttrEntry.getKey(), innerStrategy);
101         }
102
103         return new CompositeAttributeWritingStrategy(document, currentKey, innerStrats);
104     }
105
106     @Override
107     protected AttributeWritingStrategy caseListAttribute(ArrayType<?> openType) {
108         Preconditions.checkState(getLastAttribute() instanceof ListAttribute);
109         AttributeIfc innerAttribute = ((ListAttribute) getLastAttribute()).getInnerAttribute();
110
111         AttributeWritingStrategy innerStrategy = prepareWritingStrategy(key, innerAttribute, document);
112         return new ArrayAttributeWritingStrategy(innerStrategy);
113     }
114
115     @Override
116     protected AttributeWritingStrategy caseListDependeciesAttribute(ArrayType<?> openType) {
117         Preconditions.checkState(getLastAttribute() instanceof ListDependenciesAttribute);
118         AttributeWritingStrategy innerStrategy = caseDependencyAttribute(SimpleType.OBJECTNAME);
119         return new ArrayAttributeWritingStrategy(innerStrategy);
120     }
121
122 }