Initial commit with all the necessary files

[aai/aai-common.git] / aai-core / src / main / java / org / openecomp / aai / dbgen / GenTester.java

diff --git a/aai-core/src/main/java/org/openecomp/aai/dbgen/GenTester.java b/aai-core/src/main/java/org/openecomp/aai/dbgen/GenTester.java
new file mode 100644 (file)
index 0000000..43f8f76
--- /dev/null
+++ b/aai-core/src/main/java/org/openecomp/aai/dbgen/GenTester.java
@@ -0,0 +1,123 @@
+/*-
+ * ============LICENSE_START=======================================================
+ * org.openecomp.aai
+ * ================================================================================
+ * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
+ * ================================================================================
+ * Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+     http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+ * ============LICENSE_END=========================================================
+ */
+
+package org.openecomp.aai.dbgen;
+
+import com.att.eelf.configuration.Configuration;
+import com.att.eelf.configuration.EELFLogger;
+import com.att.eelf.configuration.EELFManager;
+import com.thinkaurelius.titan.core.TitanGraph;
+import com.thinkaurelius.titan.core.schema.TitanManagement;
+import org.openecomp.aai.dbmap.AAIGraph;
+import org.openecomp.aai.logging.ErrorLogHelper;
+import org.openecomp.aai.util.AAIConfig;
+import org.openecomp.aai.util.AAIConstants;
+
+import java.util.Properties;
+
+
+public class GenTester {
+
+       private static final EELFLogger LOGGER = EELFManager.getInstance().getLogger(GenTester.class);
+       
+       /**
+        * The main method.
+        *
+        * @param args the arguments
+        */
+       public static void main(String[] args) {
+          
+               TitanGraph graph = null;
+               
+               // Set the logging file properties to be used by EELFManager
+               Properties props = System.getProperties();
+               props.setProperty(Configuration.PROPERTY_LOGGING_FILE_NAME, AAIConstants.AAI_CREATE_DB_SCHEMA_LOGBACK_PROPS);
+               props.setProperty(Configuration.PROPERTY_LOGGING_FILE_PATH, AAIConstants.AAI_HOME_ETC_APP_PROPERTIES);
+               boolean addDefaultCR = true;
+               
+               try {   
+                       AAIConfig.init();
+               if (args != null && args.length > 0 ){
+                       if( "genDbRulesOnly".equals(args[0]) ){
+                               ErrorLogHelper.logError("AAI_3100",
+                                               " This option is no longer supported. What was in DbRules is now derived from the OXM files. ");
+                               return;
+                       }
+                       else if ( "GEN_DB_WITH_NO_SCHEMA".equals(args[0]) ){
+                               // Note this is done to create an empty DB with no Schema so that
+                                       // an HBase copyTable can be used to set up a copy of the db.
+                                       LOGGER.info("    ---- NOTE --- about to load a graph without doing any schema processing (takes a little while) --------   ");
+                                       graph = AAIGraph.getInstance().getGraph();
+                               
+                              if( graph == null ){
+                                          ErrorLogHelper.logError("AAI_5102", "Error creating Titan graph.");
+                                  return;
+                              }
+                              else {
+                                  LOGGER.auditEvent("Successfully loaded a Titan graph without doing any schema work.  ");
+                                  return;
+                              }
+                       } else if ("GEN_DB_WITH_NO_DEFAULT_CR".equals(args[0])) {
+                               addDefaultCR = false;
+                       }
+                       else {
+                               ErrorLogHelper.logError("AAI_3000", "Unrecognized argument passed to GenTester.java: [" + args[0] + "]. ");
+                               LOGGER.error("Unrecognized argument passed to GenTester.java: [" + args[0] + "]. ");
+                               LOGGER.error("Either pass no argument for normal processing, or use 'GEN_DB_WITH_NO_SCHEMA'.");
+                               return;
+                       }
+               }
+               
+                       //AAIConfig.init();
+                       ErrorLogHelper.loadProperties();
+                       LOGGER.info("    ---- NOTE --- about to open graph (takes a little while)--------;");
+                       graph = AAIGraph.getInstance().getGraph();
+               
+                       if( graph == null ){
+                               ErrorLogHelper.logError("AAI_5102", "Error creating Titan graph. ");
+                               return;
+                       }
+
+              // Load the propertyKeys, indexes and edge-Labels into the DB
+              TitanManagement graphMgt = graph.openManagement();
+
+              LOGGER.info("-- Loading new schema elements into Titan --");
+              SchemaGenerator.loadSchemaIntoTitan( graph, graphMgt, addDefaultCR );
+
+           } catch(Exception ex) {
+               ErrorLogHelper.logError("AAI_4000", ex.getMessage());
+           }
+           
+
+           if( graph != null ){
+                   LOGGER.info("-- graph commit");
+               graph.tx().commit();
+
+                       LOGGER.info("-- graph shutdown ");
+               graph.close();
+           }
+           
+           LOGGER.auditEvent("-- all done, if program does not exit, please kill.");
+           System.exit(0);
+    }
+
+}
+
+