opendaylight/blueprint/src/main/java/org/opendaylight/controller/blueprint/ext/BindingContext.java

   1 /*
   2  * Copyright (c) 2016 Brocade Communications 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 package org.opendaylight.controller.blueprint.ext;
   9
  10 import static com.google.common.base.Preconditions.checkArgument;
  11
  12 import com.google.common.base.Strings;
  13 import com.google.common.collect.Iterables;
  14 import java.io.IOException;
  15 import java.lang.reflect.InvocationTargetException;
  16 import java.net.URISyntaxException;
  17 import java.util.Set;
  18 import javax.xml.parsers.ParserConfigurationException;
  19 import javax.xml.stream.XMLStreamException;
  20 import javax.xml.transform.dom.DOMSource;
  21 import org.opendaylight.mdsal.binding.spec.reflect.BindingReflections;
  22 import org.opendaylight.yangtools.yang.binding.DataObject;
  23 import org.opendaylight.yangtools.yang.binding.Identifiable;
  24 import org.opendaylight.yangtools.yang.binding.Identifier;
  25 import org.opendaylight.yangtools.yang.binding.InstanceIdentifier;
  26 import org.opendaylight.yangtools.yang.common.QName;
  27 import org.opendaylight.yangtools.yang.data.api.schema.MapEntryNode;
  28 import org.opendaylight.yangtools.yang.data.api.schema.MapNode;
  29 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  30 import org.opendaylight.yangtools.yang.data.api.schema.stream.NormalizedNodeStreamWriter;
  31 import org.opendaylight.yangtools.yang.data.codec.xml.XmlParserStream;
  32 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNodes;
  33 import org.opendaylight.yangtools.yang.data.impl.schema.ImmutableNormalizedNodeStreamWriter;
  34 import org.opendaylight.yangtools.yang.data.impl.schema.NormalizedNodeResult;
  35 import org.opendaylight.yangtools.yang.model.api.ContainerSchemaNode;
  36 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
  37 import org.opendaylight.yangtools.yang.model.api.ListSchemaNode;
  38 import org.opendaylight.yangtools.yang.model.api.SchemaTreeInference;
  39 import org.opendaylight.yangtools.yang.model.api.stmt.KeyEffectiveStatement;
  40 import org.opendaylight.yangtools.yang.model.api.stmt.SchemaTreeEffectiveStatement;
  41 import org.osgi.service.blueprint.container.ComponentDefinitionException;
  42 import org.w3c.dom.Element;
  43 import org.xml.sax.SAXException;
  44
  45 /**
  46  * Base class to abstract binding type-specific behavior.
  47  *
  48  * @author Thomas Pantelis (originally; re-factored by Michael Vorburger.ch)
  49  */
  50 public abstract class BindingContext {
  51     private static String GET_KEY_METHOD = "key";
  52
  53     public static BindingContext create(final String logName, final Class<? extends DataObject> klass,
  54             final String appConfigListKeyValue) {
  55         if (Identifiable.class.isAssignableFrom(klass)) {
  56             // The binding class corresponds to a yang list.
  57             if (Strings.isNullOrEmpty(appConfigListKeyValue)) {
  58                 throw new ComponentDefinitionException(String.format(
  59                         "%s: App config binding class %s represents a yang list therefore \"%s\" must be specified",
  60                         logName, klass.getName(), DataStoreAppConfigMetadata.LIST_KEY_VALUE));
  61             }
  62
  63             try {
  64                 return ListBindingContext.newInstance(klass, appConfigListKeyValue);
  65             } catch (InstantiationException | IllegalAccessException | IllegalArgumentException
  66                     | InvocationTargetException | NoSuchMethodException | SecurityException e) {
  67                 throw new ComponentDefinitionException(String.format(
  68                         "%s: Error initializing for app config list binding class %s",
  69                         logName, klass.getName()), e);
  70             }
  71
  72         } else {
  73             return new ContainerBindingContext(klass);
  74         }
  75     }
  76
  77     public final InstanceIdentifier<DataObject> appConfigPath;
  78     public final Class<DataObject> appConfigBindingClass;
  79     public final Class<? extends DataSchemaNode> schemaType;
  80     public final QName bindingQName;
  81
  82     private BindingContext(final Class<DataObject> appConfigBindingClass,
  83             final InstanceIdentifier<DataObject> appConfigPath, final Class<? extends DataSchemaNode> schemaType) {
  84         this.appConfigBindingClass = appConfigBindingClass;
  85         this.appConfigPath = appConfigPath;
  86         this.schemaType = schemaType;
  87
  88         bindingQName = BindingReflections.findQName(appConfigBindingClass);
  89     }
  90
  91     public NormalizedNode parseDataElement(final Element element, final SchemaTreeInference dataSchema)
  92             throws XMLStreamException, IOException, ParserConfigurationException, SAXException, URISyntaxException {
  93         final NormalizedNodeResult resultHolder = new NormalizedNodeResult();
  94         final NormalizedNodeStreamWriter writer = ImmutableNormalizedNodeStreamWriter.from(resultHolder);
  95         final XmlParserStream xmlParser = XmlParserStream.create(writer, dataSchema);
  96         xmlParser.traverse(new DOMSource(element));
  97
  98         final NormalizedNode result = resultHolder.getResult();
  99         if (result instanceof MapNode) {
 100             final MapNode mapNode = (MapNode) result;
 101             final MapEntryNode mapEntryNode = mapNode.body().iterator().next();
 102             return mapEntryNode;
 103         }
 104
 105         return result;
 106     }
 107
 108     public abstract NormalizedNode newDefaultNode(SchemaTreeInference dataSchema);
 109
 110     /**
 111      * BindingContext implementation for a container binding.
 112      */
 113     private static class ContainerBindingContext extends BindingContext {
 114         @SuppressWarnings("unchecked")
 115         ContainerBindingContext(final Class<? extends DataObject> appConfigBindingClass) {
 116             super((Class<DataObject>) appConfigBindingClass,
 117                     InstanceIdentifier.create((Class<DataObject>) appConfigBindingClass), ContainerSchemaNode.class);
 118         }
 119
 120         @Override
 121         public NormalizedNode newDefaultNode(final SchemaTreeInference dataSchema) {
 122             return ImmutableNodes.containerNode(bindingQName);
 123         }
 124     }
 125
 126     /**
 127      * BindingContext implementation for a list binding.
 128      */
 129     private static class ListBindingContext extends BindingContext {
 130         final String appConfigListKeyValue;
 131
 132         @SuppressWarnings("unchecked")
 133         ListBindingContext(final Class<? extends DataObject> appConfigBindingClass,
 134                 final InstanceIdentifier<? extends DataObject> appConfigPath, final String appConfigListKeyValue) {
 135             super((Class<DataObject>) appConfigBindingClass, (InstanceIdentifier<DataObject>) appConfigPath,
 136                     ListSchemaNode.class);
 137             this.appConfigListKeyValue = appConfigListKeyValue;
 138         }
 139
 140         @SuppressWarnings({ "rawtypes", "unchecked" })
 141         static ListBindingContext newInstance(final Class<? extends DataObject> bindingClass,
 142                 final String listKeyValue) throws InstantiationException, IllegalAccessException,
 143                 IllegalArgumentException, InvocationTargetException, NoSuchMethodException, SecurityException {
 144             // We assume the yang list key type is string.
 145             Identifier keyInstance = (Identifier) bindingClass.getMethod(GET_KEY_METHOD).getReturnType()
 146                     .getConstructor(String.class).newInstance(listKeyValue);
 147             InstanceIdentifier appConfigPath = InstanceIdentifier.builder((Class)bindingClass, keyInstance).build();
 148             return new ListBindingContext(bindingClass, appConfigPath, listKeyValue);
 149         }
 150
 151         @Override
 152         public NormalizedNode newDefaultNode(final SchemaTreeInference dataSchema) {
 153             final SchemaTreeEffectiveStatement<?> stmt = Iterables.getLast(dataSchema.statementPath());
 154
 155             // We assume there's only one key for the list.
 156             final Set<QName> keys = stmt.findFirstEffectiveSubstatementArgument(KeyEffectiveStatement.class)
 157                 .orElseThrow();
 158
 159             checkArgument(keys.size() == 1, "Expected only 1 key for list %s", appConfigBindingClass);
 160             QName listKeyQName = keys.iterator().next();
 161             return ImmutableNodes.mapEntryBuilder(bindingQName, listKeyQName, appConfigListKeyValue).build();
 162         }
 163     }
 164 }