aai-core/src/main/java/org/openecomp/aai/dbgen/SchemaGenerator.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.dbgen;
  22
  23
  24 import com.att.eelf.configuration.EELFLogger;
  25 import com.att.eelf.configuration.EELFManager;
  26 import com.google.common.collect.Multimap;
  27 import com.thinkaurelius.titan.core.*;
  28 import com.thinkaurelius.titan.core.schema.TitanManagement;
  29 import org.apache.tinkerpop.gremlin.structure.Vertex;
  30 import org.openecomp.aai.db.props.AAIProperties;
  31 import org.openecomp.aai.exceptions.AAIException;
  32 import org.openecomp.aai.introspection.Introspector;
  33 import org.openecomp.aai.introspection.Loader;
  34 import org.openecomp.aai.introspection.LoaderFactory;
  35 import org.openecomp.aai.introspection.ModelType;
  36 import org.openecomp.aai.schema.enums.PropertyMetadata;
  37 import org.openecomp.aai.serialization.db.EdgeRule;
  38 import org.openecomp.aai.serialization.db.EdgeRules;
  39 import org.openecomp.aai.util.AAIConfig;
  40
  41 import java.util.*;
  42
  43
  44 public class SchemaGenerator {
  45
  46         private static final EELFLogger LOGGER = EELFManager.getInstance().getLogger(SchemaGenerator.class);
  47         private static boolean addDefaultCR = true;
  48
  49
  50          /**
  51          * Load schema into titan.
  52          *
  53          * @param graph the graph
  54          * @param graphMgmt the graph mgmt
  55          * @param addDefaultCloudRegion the add default cloud region
  56          */
  57         public static void loadSchemaIntoTitan(final TitanGraph graph, final TitanManagement graphMgmt, boolean addDefaultCloudRegion) {
  58                  addDefaultCR = addDefaultCloudRegion;
  59                  loadSchemaIntoTitan(graph, graphMgmt);
  60          }
  61
  62     /**
  63      * Load schema into titan.
  64      *
  65      * @param graph the graph
  66      * @param graphMgmt the graph mgmt
  67      */
  68     public static void loadSchemaIntoTitan(final TitanGraph graph, final TitanManagement graphMgmt) {
  69
  70         try {
  71                 AAIConfig.init();
  72         }
  73         catch (Exception ex){
  74                         LOGGER.error(" ERROR - Could not run AAIConfig.init(). ", ex);
  75                         System.exit(1);
  76                 }
  77
  78         // NOTE - Titan 0.5.3 doesn't keep a list of legal node Labels.
  79         //   They are only used when a vertex is actually being created.  Titan 1.1 will keep track (we think).
  80
  81
  82                 // Use EdgeRules to make sure edgeLabels are defined in the db.  NOTE: the multiplicty used here is
  83         // always "MULTI".  This is not the same as our internal "Many2Many", "One2One", "One2Many" or "Many2One"
  84         // We use the same edge-label for edges between many different types of nodes and our internal
  85         // multiplicty definitions depends on which two types of nodes are being connected.
  86
  87                 Multimap<String, EdgeRule> edges = null;
  88                 Set<String> labels = new HashSet<>();
  89                 try {
  90                         edges = EdgeRules.getInstance().getAllRules();
  91                         for (EdgeRule rule : edges.values()) {
  92                                 labels.add(rule.getLabel());
  93                         }
  94                 } catch (AAIException e) {
  95                         LOGGER.error("could not get edge rules", e);
  96                         System.exit(1);
  97                 }
  98                 for( String label: labels){
  99                         if( graphMgmt.containsRelationType(label) ) {
 100                 LOGGER.debug(" EdgeLabel  [" + label + "] already existed. ");
 101             } else {
 102                 LOGGER.debug("Making EdgeLabel: [" + label + "]");
 103                 graphMgmt.makeEdgeLabel(label).multiplicity(Multiplicity.valueOf("MULTI")).make();
 104             }
 105         }
 106
 107                 Loader loader = LoaderFactory.createLoaderForVersion(ModelType.MOXY, AAIProperties.LATEST);
 108                 Map<String, Introspector> objs = loader.getAllObjects();
 109                 Map<String, PropertyKey> seenProps = new HashMap<>();
 110
 111                 for (Introspector obj : objs.values()) {
 112                         for (String propName : obj.getProperties()) {
 113                                 String dbPropName = propName;
 114                                 Optional<String> alias = obj.getPropertyMetadata(propName, PropertyMetadata.DB_ALIAS);
 115                                 if (alias.isPresent()) {
 116                                         dbPropName = alias.get();
 117                                 }
 118                                 if( graphMgmt.containsRelationType(propName) ){
 119                         LOGGER.debug(" PropertyKey  [" + propName + "] already existed in the DB. ");
 120                     } else {
 121                         Class<?> type = obj.getClass(propName);
 122                         Cardinality cardinality = Cardinality.SINGLE;
 123                         boolean process = false;
 124                         if (obj.isListType(propName) && obj.isSimpleGenericType(propName)) {
 125                                 cardinality = Cardinality.SET;
 126                                 type = obj.getGenericTypeClass(propName);
 127                                 process = true;
 128                         } else if (obj.isSimpleType(propName)) {
 129                                 process = true;
 130                         }
 131
 132                         if (process) {
 133
 134                                 LOGGER.info("Creating PropertyKey: [" + dbPropName + "], ["+ type.getSimpleName() + "], [" + cardinality + "]");
 135                                 PropertyKey propK;
 136                                 if (!seenProps.containsKey(dbPropName)) {
 137                                         propK = graphMgmt.makePropertyKey(dbPropName).dataType(type).cardinality(cardinality).make();
 138                                         seenProps.put(dbPropName, propK);
 139                                 } else {
 140                                         propK = seenProps.get(dbPropName);
 141                                 }
 142                                 if (graphMgmt.containsGraphIndex(dbPropName)) {
 143                                         LOGGER.debug(" Index  [" + dbPropName + "] already existed in the DB. ");
 144                                 } else {
 145                                         if( obj.getIndexedProperties().contains(propName) ){
 146                                                 if( obj.getUniqueProperties().contains(propName) ){
 147                                                                         LOGGER.info("Add Unique index for PropertyKey: [" + dbPropName + "]");
 148                                                  graphMgmt.buildIndex(dbPropName,Vertex.class).addKey(propK).unique().buildCompositeIndex();
 149                                              } else {
 150                                                 LOGGER.info("Add index for PropertyKey: [" + dbPropName + "]");
 151                                                  graphMgmt.buildIndex(dbPropName,Vertex.class).addKey(propK).buildCompositeIndex();
 152                                              }
 153                                          } else {
 154                                                 LOGGER.info("No index added for PropertyKey: [" + dbPropName + "]");
 155                                          }
 156                                 }
 157                         }
 158                     }
 159                         }
 160                 }
 161
 162         LOGGER.info("-- About to call graphMgmt commit");
 163         graphMgmt.commit();
 164     }// End of loadSchemaIntoTitan()
 165
 166 }
 167
 168