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