2 * Copyright (c) 2013 Cisco Systems, Inc. and others. All rights reserved.
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
9 package org.opendaylight.controller.netconf.confignetconfconnector.mapping.attributes.mapping;
11 import com.google.common.base.Preconditions;
12 import com.google.common.collect.Maps;
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.DependencyAttribute;
21 import org.opendaylight.controller.config.yangjmxgenerator.attribute.ListAttribute;
22 import org.opendaylight.controller.config.yangjmxgenerator.attribute.ListDependenciesAttribute;
23 import org.opendaylight.controller.config.yangjmxgenerator.attribute.TOAttribute;
24 import org.opendaylight.controller.netconf.confignetconfconnector.mapping.attributes.AttributeIfcSwitchStatement;
25 import org.opendaylight.controller.netconf.confignetconfconnector.osgi.EnumResolver;
27 public class ObjectMapper extends AttributeIfcSwitchStatement<AttributeMappingStrategy<?, ? extends OpenType<?>>> {
29 private EnumResolver enumResolver;
31 public Map<String, AttributeMappingStrategy<?, ? extends OpenType<?>>> prepareMapping(
32 Map<String, AttributeIfc> configDefinition, EnumResolver enumResolver) {
33 this.enumResolver = Preconditions.checkNotNull(enumResolver);
34 Map<String, AttributeMappingStrategy<?, ? extends OpenType<?>>> strategies = Maps.newHashMap();
36 for (Entry<String, AttributeIfc> attrEntry : configDefinition.entrySet()) {
37 strategies.put(attrEntry.getKey(), prepareStrategy(attrEntry.getValue()));
43 public AttributeMappingStrategy<?, ? extends OpenType<?>> prepareStrategy(AttributeIfc attributeIfc) {
45 if(attributeIfc instanceof DependencyAttribute) {
46 namespaceOfDepAttr = ((DependencyAttribute)attributeIfc).getDependency().getSie().getQName().getNamespace().toString();
47 } else if (attributeIfc instanceof ListDependenciesAttribute) {
48 namespaceOfDepAttr = ((ListDependenciesAttribute)attributeIfc).getDependency().getSie().getQName().getNamespace().toString();
51 return switchAttribute(attributeIfc);
54 private Map<String, String> createJmxToYangMapping(TOAttribute attributeIfc) {
55 Map<String, String> retVal = Maps.newHashMap();
56 for (Entry<String, AttributeIfc> entry : attributeIfc.getJmxPropertiesToTypesMap().entrySet()) {
57 retVal.put(entry.getKey(), (entry.getValue()).getAttributeYangName());
63 protected AttributeMappingStrategy<?, ? extends OpenType<?>> caseJavaSimpleAttribute(SimpleType<?> openType) {
64 return new SimpleAttributeMappingStrategy(openType);
68 protected AttributeMappingStrategy<?, ? extends OpenType<?>> caseJavaEnumAttribute(final OpenType<?> openType) {
69 return new EnumAttributeMappingStrategy(((CompositeType) openType), enumResolver);
73 protected AttributeMappingStrategy<?, ? extends OpenType<?>> caseJavaArrayAttribute(ArrayType<?> openType) {
75 AttributeMappingStrategy<?, ? extends OpenType<?>> innerStrategy = new SimpleAttributeMappingStrategy(
76 (SimpleType<?>) openType.getElementOpenType());
77 return new ArrayAttributeMappingStrategy(openType, innerStrategy);
81 protected AttributeMappingStrategy<?, ? extends OpenType<?>> caseJavaCompositeAttribute(CompositeType openType) {
82 Map<String, AttributeMappingStrategy<?, ? extends OpenType<?>>> innerStrategies = Maps.newHashMap();
84 Map<String, String> attributeMapping = Maps.newHashMap();
86 for (String innerAttributeKey : openType.keySet()) {
88 innerStrategies.put(innerAttributeKey, caseJavaAttribute(openType.getType(innerAttributeKey)));
89 attributeMapping.put(innerAttributeKey, innerAttributeKey);
92 return new CompositeAttributeMappingStrategy(openType, innerStrategies, attributeMapping);
96 protected AttributeMappingStrategy<?, ? extends OpenType<?>> caseJavaUnionAttribute(OpenType<?> openType) {
97 Map<String, AttributeMappingStrategy<?, ? extends OpenType<?>>> innerStrategies = Maps.newHashMap();
99 Map<String, String> attributeMapping = Maps.newHashMap();
101 CompositeType compositeType = (CompositeType) openType;
102 for (String innerAttributeKey : compositeType.keySet()) {
104 innerStrategies.put(innerAttributeKey, caseJavaAttribute(compositeType.getType(innerAttributeKey)));
105 attributeMapping.put(innerAttributeKey, innerAttributeKey);
108 return new UnionCompositeAttributeMappingStrategy(compositeType, innerStrategies, attributeMapping);
111 private String namespaceOfDepAttr;
114 protected AttributeMappingStrategy<?, ? extends OpenType<?>> caseDependencyAttribute(
115 SimpleType<?> openType) {
116 return new ObjectNameAttributeMappingStrategy(openType, namespaceOfDepAttr);
120 protected AttributeMappingStrategy<?, ? extends OpenType<?>> caseTOAttribute(CompositeType openType) {
121 Map<String, AttributeMappingStrategy<?, ? extends OpenType<?>>> innerStrategies = Maps.newHashMap();
123 Preconditions.checkState(getLastAttribute() instanceof TOAttribute);
124 TOAttribute lastTO = (TOAttribute) getLastAttribute();
126 for (Entry<String, AttributeIfc> innerAttrEntry : ((TOAttribute)getLastAttribute()).getJmxPropertiesToTypesMap().entrySet()) {
127 innerStrategies.put(innerAttrEntry.getKey(), prepareStrategy(innerAttrEntry.getValue()));
130 return new CompositeAttributeMappingStrategy(openType, innerStrategies,
131 createJmxToYangMapping(lastTO));
135 protected AttributeMappingStrategy<?, ? extends OpenType<?>> caseListAttribute(ArrayType<?> openType) {
136 Preconditions.checkState(getLastAttribute() instanceof ListAttribute);
137 return new ArrayAttributeMappingStrategy(openType,
138 prepareStrategy(((ListAttribute) getLastAttribute()).getInnerAttribute()));
142 protected AttributeMappingStrategy<?, ? extends OpenType<?>> caseListDependeciesAttribute(ArrayType<?> openType) {
143 Preconditions.checkState(getLastAttribute() instanceof ListDependenciesAttribute);
144 return new ArrayAttributeMappingStrategy(openType, caseDependencyAttribute(SimpleType.OBJECTNAME));