aai-core/src/main/java/org/openecomp/aai/introspection/ModelInjestor.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * org.openecomp.aai
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.openecomp.aai.introspection;
  22
  23 import java.io.File;
  24 import java.io.FileInputStream;
  25 import java.io.FileNotFoundException;
  26 import java.io.InputStream;
  27 import java.util.HashMap;
  28 import java.util.Map;
  29 import java.util.regex.Matcher;
  30 import java.util.regex.Pattern;
  31
  32 import javax.xml.bind.JAXBException;
  33
  34 import org.eclipse.persistence.dynamic.DynamicType;
  35 import org.eclipse.persistence.jaxb.JAXBContextProperties;
  36 import org.eclipse.persistence.jaxb.dynamic.DynamicJAXBContext;
  37 import org.eclipse.persistence.jaxb.dynamic.DynamicJAXBContextFactory;
  38
  39 import org.openecomp.aai.util.AAIConstants;
  40
  41
  42 public class ModelInjestor {
  43
  44         private Map<Version, DynamicJAXBContext> versionContextMap = new HashMap<>();
  45         private static final Pattern classNamePattern = Pattern.compile("\\.(v\\d+)\\.");
  46         private static final Pattern uriPattern =       Pattern.compile("(v\\d+)\\/");
  47
  48
  49         /**
  50          * Instantiates a new model injestor.
  51          */
  52         private ModelInjestor() {
  53                 try {
  54                         injestModels();
  55                 } catch (FileNotFoundException | JAXBException e) {
  56                         throw new RuntimeException(e);
  57                 }
  58         }
  59
  60         private static class Helper {
  61                 private static final ModelInjestor INSTANCE = new ModelInjestor();
  62         }
  63
  64         /**
  65          * Gets the single instance of ModelInjestor.
  66          *
  67          * @return single instance of ModelInjestor
  68          */
  69         public synchronized static ModelInjestor getInstance() {
  70                 return Helper.INSTANCE;
  71         }
  72
  73         /**
  74          * Injest models.
  75          *
  76          * @throws FileNotFoundException the file not found exception
  77          * @throws JAXBException the JAXB exception
  78          */
  79         private void injestModels() throws FileNotFoundException, JAXBException {
  80
  81                 for (Version version : Version.values()) {
  82                         this.injestModel(version);
  83                 }
  84         }
  85
  86         /**
  87          * Injest model.
  88          *
  89          * @param version the version
  90          * @throws JAXBException the JAXB exception
  91          * @throws FileNotFoundException the file not found exception
  92          */
  93         private void injestModel (Version version) throws JAXBException, FileNotFoundException {
  94                 String fileName = this.getOXMFileName(version);
  95                 InputStream iStream = new FileInputStream(new File(fileName));
  96                 Map<String, Object> properties = new HashMap<String, Object>();
  97                 properties.put(JAXBContextProperties.OXM_METADATA_SOURCE, iStream);
  98                 final DynamicJAXBContext jaxbContext = DynamicJAXBContextFactory.createContextFromOXM(this.getClass().getClassLoader(), properties);
  99                 versionContextMap.put(version, jaxbContext);
 100         }
 101
 102         /**
 103          * Gets the version from class name.
 104          *
 105          * @param classname the classname
 106          * @return the version from class name
 107          */
 108         public Version getVersionFromClassName (String classname) {
 109                 Matcher m = classNamePattern.matcher(classname);
 110                 String version = "v2"; //for the OXM, only the v2 ones don't include a model name, hence this default
 111                 if (m.find()) {
 112                         version = m.group(1);
 113                 }
 114
 115                 return Version.valueOf(version);
 116         }
 117
 118         /**
 119          * Gets the context for URI.
 120          *
 121          * @param uri the uri
 122          * @return the context for URI
 123          */
 124         public DynamicJAXBContext getContextForURI(String uri) {
 125                 DynamicJAXBContext result = null;
 126                 Matcher m = uriPattern.matcher(uri);
 127                 Version version = null;
 128                 if (m.find()) {
 129                         version = Version.valueOf(m.group(1));
 130                         result = versionContextMap.get(version);
 131                 }
 132
 133                 return result;
 134         }
 135
 136         /**
 137          * Gets the context for version.
 138          *
 139          * @param version the version
 140          * @return the context for version
 141          */
 142         public DynamicJAXBContext getContextForVersion(Version version) {
 143                 DynamicJAXBContext result = null;
 144
 145                 result = versionContextMap.get(version);
 146
 147
 148                 return result;
 149         }
 150
 151         /**
 152          * Gets the dynamic type for class name.
 153          *
 154          * @param classname the classname
 155          * @return the dynamic type for class name
 156          */
 157         public DynamicType getDynamicTypeForClassName(String classname) {
 158                 DynamicType result = null;
 159                 DynamicJAXBContext context = null;
 160
 161                 Version version = this.getVersionFromClassName(classname);
 162
 163                 context = versionContextMap.get(version);
 164
 165                 if (context != null) {
 166                         result = context.getDynamicType(classname);
 167                 }
 168
 169                 return result;
 170         }
 171
 172         public String getOXMFileName(Version v) {
 173                 return  AAIConstants.AAI_HOME_ETC_OXM + "aai_oxm_" + v.toString() + ".xml";
 174         }
 175
 176 }