2 * ============LICENSE_START=======================================================
4 * ================================================================================
5 * Copyright © 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
11 * http://www.apache.org/licenses/LICENSE-2.0
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=========================================================
20 * ECOMP is a trademark and service mark of AT&T Intellectual Property.
22 package org.onap.aai.dbgen;
24 import com.att.eelf.configuration.Configuration;
25 import com.att.eelf.configuration.EELFLogger;
26 import com.att.eelf.configuration.EELFManager;
27 import com.thinkaurelius.titan.core.TitanGraph;
28 import com.thinkaurelius.titan.core.schema.TitanManagement;
29 import org.onap.aai.dbmap.AAIGraph;
30 import org.onap.aai.logging.ErrorLogHelper;
31 import org.onap.aai.logging.LoggingContext;
32 import org.onap.aai.logging.LoggingContext.StatusCode;
33 import org.onap.aai.util.AAIConfig;
34 import org.onap.aai.util.AAIConstants;
36 import java.util.Properties;
37 import java.util.UUID;
40 public class GenTester {
42 private static EELFLogger LOGGER;
47 * @param args the arguments
49 public static void main(String[] args) {
51 TitanGraph graph = null;
52 System.setProperty("aai.service.name", GenTester.class.getSimpleName());
53 // Set the logging file properties to be used by EELFManager
54 Properties props = System.getProperties();
55 props.setProperty(Configuration.PROPERTY_LOGGING_FILE_NAME, AAIConstants.AAI_CREATE_DB_SCHEMA_LOGBACK_PROPS);
56 props.setProperty(Configuration.PROPERTY_LOGGING_FILE_PATH, AAIConstants.AAI_HOME_ETC_APP_PROPERTIES);
57 LOGGER = EELFManager.getInstance().getLogger(GenTester.class);
58 boolean addDefaultCR = true;
60 LoggingContext.init();
61 LoggingContext.component("DBGenTester");
62 LoggingContext.partnerName("AAI-TOOLS");
63 LoggingContext.targetEntity("AAI");
64 LoggingContext.requestId(UUID.randomUUID().toString());
65 LoggingContext.serviceName("AAI");
66 LoggingContext.targetServiceName("main");
67 LoggingContext.statusCode(StatusCode.COMPLETE);
68 LoggingContext.responseCode(LoggingContext.SUCCESS);
71 if (args != null && args.length > 0 ){
72 if( "genDbRulesOnly".equals(args[0]) ){
73 ErrorLogHelper.logError("AAI_3100",
74 " This option is no longer supported. What was in DbRules is now derived from the OXM files. ");
77 else if ( "GEN_DB_WITH_NO_SCHEMA".equals(args[0]) ){
78 // Note this is done to create an empty DB with no Schema so that
79 // an HBase copyTable can be used to set up a copy of the db.
80 String imsg = " ---- NOTE --- about to load a graph without doing any schema processing (takes a little while) -------- ";
81 System.out.println(imsg);
83 graph = AAIGraph.getInstance().getGraph();
86 ErrorLogHelper.logError("AAI_5102", "Error creating Titan graph.");
90 String amsg = "Successfully loaded a Titan graph without doing any schema work. ";
91 System.out.println(amsg);
92 LOGGER.auditEvent(amsg);
95 } else if ("GEN_DB_WITH_NO_DEFAULT_CR".equals(args[0])) {
99 ErrorLogHelper.logError("AAI_3000", "Unrecognized argument passed to GenTester.java: [" + args[0] + "]. ");
101 String emsg = "Unrecognized argument passed to GenTester.java: [" + args[0] + "]. ";
102 System.out.println(emsg);
103 LoggingContext.statusCode(StatusCode.ERROR);
104 LoggingContext.responseCode(LoggingContext.BUSINESS_PROCESS_ERROR);
107 emsg = "Either pass no argument for normal processing, or use 'GEN_DB_WITH_NO_SCHEMA'.";
108 System.out.println(emsg);
116 ErrorLogHelper.loadProperties();
117 String imsg = " ---- NOTE --- about to open graph (takes a little while)--------;";
118 System.out.println(imsg);
120 graph = AAIGraph.getInstance().getGraph();
123 ErrorLogHelper.logError("AAI_5102", "Error creating Titan graph. ");
127 // Load the propertyKeys, indexes and edge-Labels into the DB
128 TitanManagement graphMgt = graph.openManagement();
130 imsg = "-- Loading new schema elements into Titan --";
131 System.out.println(imsg);
133 SchemaGenerator.loadSchemaIntoTitan( graph, graphMgt, addDefaultCR );
135 } catch(Exception ex) {
136 ErrorLogHelper.logError("AAI_4000", ex.getMessage());
141 String imsg = "-- graph commit";
142 System.out.println(imsg);
146 imsg = "-- graph shutdown ";
147 System.out.println(imsg);
152 LOGGER.auditEvent("-- all done, if program does not exit, please kill.");