opendaylight/blueprint/src/main/java/org/opendaylight/controller/blueprint/ext/DataStoreAppConfigDefaultXMLReader.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 com.google.common.base.Optional;
  11 import com.google.common.base.Strings;
  12 import com.google.common.io.Resources;
  13 import java.io.IOException;
  14 import java.io.InputStream;
  15 import java.net.URISyntaxException;
  16 import java.net.URL;
  17 import javax.xml.parsers.ParserConfigurationException;
  18 import javax.xml.stream.XMLStreamException;
  19 import org.opendaylight.controller.sal.core.api.model.SchemaService;
  20 import org.opendaylight.mdsal.binding.dom.codec.api.BindingNormalizedNodeSerializer;
  21 import org.opendaylight.yangtools.util.xml.UntrustedXML;
  22 import org.opendaylight.yangtools.yang.binding.DataObject;
  23 import org.opendaylight.yangtools.yang.common.QName;
  24 import org.opendaylight.yangtools.yang.data.api.YangInstanceIdentifier;
  25 import org.opendaylight.yangtools.yang.data.api.schema.NormalizedNode;
  26 import org.opendaylight.yangtools.yang.model.api.DataSchemaNode;
  27 import org.opendaylight.yangtools.yang.model.api.Module;
  28 import org.opendaylight.yangtools.yang.model.api.SchemaContext;
  29 import org.slf4j.Logger;
  30 import org.slf4j.LoggerFactory;
  31 import org.w3c.dom.Document;
  32 import org.xml.sax.SAXException;
  33
  34 /**
  35  * DataObject XML file reader used by {@link DataStoreAppConfigMetadata}.
  36  * Available as a standalone class to make it easy to write unit tests which can
  37  * catch malformed default "clustered-app-conf" config data XML files in
  38  * downstream projects.
  39  *
  40  * @author Thomas Pantelis (originally; re-factored by Michael Vorburger.ch)
  41  */
  42 public class DataStoreAppConfigDefaultXMLReader<T extends DataObject> {
  43
  44     private static final Logger LOG = LoggerFactory.getLogger(DataStoreAppConfigDefaultXMLReader.class);
  45
  46     private final String logName;
  47     private final String defaultAppConfigFileName;
  48     private final SchemaService schemaService;
  49     private final BindingNormalizedNodeSerializer bindingSerializer;
  50     private final BindingContext bindingContext;
  51     private final ConfigURLProvider inputStreamProvider;
  52
  53     @FunctionalInterface
  54     public interface FallbackConfigProvider {
  55         NormalizedNode<?,?> get(SchemaContext schemaContext, DataSchemaNode dataSchema) throws IOException,
  56                 XMLStreamException, ParserConfigurationException, SAXException, URISyntaxException;
  57     }
  58
  59     @FunctionalInterface
  60     public interface ConfigURLProvider {
  61         Optional<URL> getURL(String appConfigFileName) throws IOException;
  62     }
  63
  64     public DataStoreAppConfigDefaultXMLReader(
  65             final String logName,
  66             final String defaultAppConfigFileName,
  67             final SchemaService schemaService,
  68             final BindingNormalizedNodeSerializer bindingSerializer,
  69             final BindingContext bindingContext,
  70             final ConfigURLProvider inputStreamProvider) {
  71
  72         this.logName = logName;
  73         this.defaultAppConfigFileName = defaultAppConfigFileName;
  74         this.schemaService = schemaService;
  75         this.bindingSerializer = bindingSerializer;
  76         this.bindingContext = bindingContext;
  77         this.inputStreamProvider = inputStreamProvider;
  78     }
  79
  80     public DataStoreAppConfigDefaultXMLReader(
  81             final Class<?> testClass,
  82             final String defaultAppConfigFileName,
  83             final SchemaService schemaService,
  84             final BindingNormalizedNodeSerializer bindingSerializer,
  85             final Class<T> klass) {
  86         this(testClass.getName(), defaultAppConfigFileName, schemaService, bindingSerializer,
  87             BindingContext.create(testClass.getName(), klass, null),
  88             appConfigFileName -> Optional.of(getURL(testClass, defaultAppConfigFileName)));
  89     }
  90
  91     private static URL getURL(final Class<?> testClass, final String defaultAppConfigFileName) {
  92         return Resources.getResource(testClass, defaultAppConfigFileName);
  93     }
  94
  95     public T createDefaultInstance() throws ConfigXMLReaderException, ParserConfigurationException, XMLStreamException,
  96             IOException, SAXException, URISyntaxException {
  97         return createDefaultInstance((schemaContext, dataSchema) -> {
  98             throw new IllegalArgumentException("Failed to read XML "
  99                     + "(not creating model from defaults as runtime would, for better clarity in tests)");
 100         });
 101     }
 102
 103     @SuppressWarnings("unchecked")
 104     public T createDefaultInstance(final FallbackConfigProvider fallback) throws ConfigXMLReaderException,
 105             URISyntaxException, ParserConfigurationException, XMLStreamException, SAXException, IOException {
 106         YangInstanceIdentifier yangPath = bindingSerializer.toYangInstanceIdentifier(bindingContext.appConfigPath);
 107
 108         LOG.debug("{}: Creating app config instance from path {}, Qname: {}", logName, yangPath,
 109                 bindingContext.bindingQName);
 110
 111         if (schemaService == null) {
 112             throw new ConfigXMLReaderException(
 113                     String.format("%s: Could not obtain the SchemaService OSGi service", logName));
 114         }
 115
 116         SchemaContext schemaContext = schemaService.getGlobalContext();
 117
 118         Module module = schemaContext.findModuleByNamespaceAndRevision(bindingContext.bindingQName.getNamespace(),
 119                 bindingContext.bindingQName.getRevision());
 120         if (module == null) {
 121             throw new ConfigXMLReaderException(
 122                     String.format("%s: Could not obtain the module schema for namespace %s, revision %s",
 123                     logName, bindingContext.bindingQName.getNamespace(), bindingContext.bindingQName.getRevision()));
 124         }
 125
 126         DataSchemaNode dataSchema = module.getDataChildByName(bindingContext.bindingQName);
 127         if (dataSchema == null) {
 128             throw new ConfigXMLReaderException(
 129                     String.format("%s: Could not obtain the schema for %s", logName,
 130                     bindingContext.bindingQName));
 131         }
 132
 133         if (!bindingContext.schemaType.isAssignableFrom(dataSchema.getClass())) {
 134             throw new ConfigXMLReaderException(
 135                     String.format("%s: Expected schema type %s for %s but actual type is %s", logName,
 136                     bindingContext.schemaType, bindingContext.bindingQName, dataSchema.getClass()));
 137         }
 138
 139         NormalizedNode<?, ?> dataNode = parsePossibleDefaultAppConfigXMLFile(schemaContext, dataSchema);
 140         if (dataNode == null) {
 141             dataNode = fallback.get(schemaService.getGlobalContext(), dataSchema);
 142         }
 143
 144         DataObject appConfig = bindingSerializer.fromNormalizedNode(yangPath, dataNode).getValue();
 145         if (appConfig == null) {
 146             // This shouldn't happen but need to handle it in case...
 147             throw new ConfigXMLReaderException(
 148                     String.format("%s: Could not create instance for app config binding %s",
 149                     logName, bindingContext.appConfigBindingClass));
 150         } else {
 151             return (T) appConfig;
 152         }
 153     }
 154
 155     private NormalizedNode<?, ?> parsePossibleDefaultAppConfigXMLFile(final SchemaContext schemaContext,
 156             final DataSchemaNode dataSchema) throws ConfigXMLReaderException {
 157
 158         String appConfigFileName = defaultAppConfigFileName;
 159         if (Strings.isNullOrEmpty(appConfigFileName)) {
 160             String moduleName = findYangModuleName(bindingContext.bindingQName, schemaContext);
 161             appConfigFileName = moduleName + "_" + bindingContext.bindingQName.getLocalName() + ".xml";
 162         }
 163
 164         Optional<URL> optionalURL;
 165         try {
 166             optionalURL = inputStreamProvider.getURL(appConfigFileName);
 167         } catch (final IOException e) {
 168             String msg = String.format("%s: Could not getURL()", logName);
 169             LOG.error(msg, e);
 170             throw new ConfigXMLReaderException(msg, e);
 171         }
 172         if (!optionalURL.isPresent()) {
 173             return null;
 174         }
 175         URL url = optionalURL.get();
 176         try (InputStream is = url.openStream()) {
 177             Document root = UntrustedXML.newDocumentBuilder().parse(is);
 178             NormalizedNode<?, ?> dataNode = bindingContext.parseDataElement(root.getDocumentElement(), dataSchema,
 179                     schemaContext);
 180
 181             LOG.debug("{}: Parsed data node: {}", logName, dataNode);
 182
 183             return dataNode;
 184         } catch (final IOException | SAXException | XMLStreamException | ParserConfigurationException
 185                 | URISyntaxException e) {
 186             String msg = String.format("%s: Could not read/parse app config %s", logName, url);
 187             LOG.error(msg, e);
 188             throw new ConfigXMLReaderException(msg, e);
 189         }
 190     }
 191
 192     private String findYangModuleName(final QName qname, final SchemaContext schemaContext)
 193             throws ConfigXMLReaderException {
 194         for (Module m : schemaContext.getModules()) {
 195             if (qname.getModule().equals(m.getQNameModule())) {
 196                 return m.getName();
 197             }
 198         }
 199         throw new ConfigXMLReaderException(
 200                 String.format("%s: Could not find yang module for QName %s", logName, qname));
 201     }
 202
 203 }