aai-core/src/main/java/org/onap/aai/dbmap/AAIGraph.java

   1 /**
   2  * ============LICENSE_START=======================================================
   3  * org.onap.aai
   4  * ================================================================================
   5  * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  *  Modifications Copyright © 2018 IBM.
   8  * ================================================================================
   9  * Licensed under the Apache License, Version 2.0 (the "License");
  10  * you may not use this file except in compliance with the License.
  11  * You may obtain a copy of the License at
  12  *
  13  *    http://www.apache.org/licenses/LICENSE-2.0
  14  *
  15  * Unless required by applicable law or agreed to in writing, software
  16  * distributed under the License is distributed on an "AS IS" BASIS,
  17  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  18  * See the License for the specific language governing permissions and
  19  * limitations under the License.
  20  * ============LICENSE_END=========================================================
  21  */
  22
  23 package org.onap.aai.dbmap;
  24
  25 import java.io.FileNotFoundException;
  26 import java.util.Properties;
  27
  28 import org.apache.commons.configuration.PropertiesConfiguration;
  29 import org.apache.commons.lang3.exception.ExceptionUtils;
  30 import org.apache.tinkerpop.gremlin.structure.Graph;
  31 import org.apache.tinkerpop.gremlin.structure.io.IoCore;
  32 import org.janusgraph.core.JanusGraph;
  33 import org.janusgraph.core.JanusGraphFactory;
  34 import org.janusgraph.core.schema.JanusGraphManagement;
  35 import org.onap.aai.config.SpringContextAware;
  36 import org.onap.aai.dbgen.SchemaGenerator;
  37 import org.onap.aai.dbgen.SchemaGenerator4Hist;
  38 import org.onap.aai.exceptions.AAIException;
  39 import org.onap.aai.util.AAIConstants;
  40 import org.slf4j.Logger;
  41 import org.slf4j.LoggerFactory;
  42
  43 /**
  44  * Database Mapping class which acts as the middle man between the REST
  45  * interface objects and JanusGraph DB objects. This class provides methods to commit
  46  * the objects received on the REST interface into the JanusGraph graph database as
  47  * vertices and edges. Transactions are also managed here by using a JanusGraph
  48  * object to load, commit/rollback and shutdown for each request. The data model
  49  * rules such as keys/required properties are handled by calling DBMeth methods
  50  * which are driven by a specification file in json.
  51  *
  52  *
  53  */
  54 public class AAIGraph {
  55
  56     private static final Logger logger = LoggerFactory.getLogger(AAIGraph.class);
  57     private static final String IN_MEMORY = "inmemory";
  58     protected JanusGraph graph;
  59     private static boolean isInit = false;
  60
  61     /**
  62      * Instantiates a new AAI graph.
  63      */
  64     private AAIGraph() {
  65         try {
  66             String serviceName = System.getProperty("aai.service.name", "NA");
  67             String rtConfig = System.getProperty("realtime.db.config");
  68             if (rtConfig == null) {
  69                 rtConfig = AAIConstants.REALTIME_DB_CONFIG;
  70             }
  71             this.loadGraph(rtConfig, serviceName);
  72         } catch (Exception e) {
  73             throw new RuntimeException("Failed to instantiate graphs", e);
  74         }
  75     }
  76
  77     private static class Helper {
  78         private static final AAIGraph INSTANCE = new AAIGraph();
  79
  80         private Helper() {
  81
  82         }
  83     }
  84
  85     /**
  86      * Gets the single instance of AAIGraph.
  87      *
  88      * @return single instance of AAIGraph
  89      */
  90     public static AAIGraph getInstance() {
  91         isInit = true;
  92         return Helper.INSTANCE;
  93     }
  94
  95     public static boolean isInit() {
  96         return isInit;
  97     }
  98
  99     private void loadGraph(String configPath, String serviceName) throws Exception {
 100         // Graph being opened by JanusGraphFactory is being placed in hashmap to be used later
 101         // These graphs shouldn't be closed until the application shutdown
 102         try {
 103             PropertiesConfiguration propertiesConfiguration = new AAIGraphConfig.Builder(configPath)
 104                     .forService(serviceName).withGraphType("realtime").buildConfiguration();
 105             graph = JanusGraphFactory.open(propertiesConfiguration);
 106
 107             Properties graphProps = new Properties();
 108             propertiesConfiguration.getKeys()
 109                     .forEachRemaining(k -> graphProps.setProperty(k, propertiesConfiguration.getString(k)));
 110
 111             if (IN_MEMORY.equals(graphProps.get("storage.backend"))) {
 112                 // Load the propertyKeys, indexes and edge-Labels into the DB
 113                 loadSchema(graph);
 114                 loadSnapShotToInMemoryGraph(graph, graphProps);
 115             }
 116
 117             if (graph == null) {
 118                 throw new AAIException("AAI_5102");
 119             }
 120
 121         } catch (FileNotFoundException e) {
 122             throw new AAIException("AAI_4001", e);
 123         }
 124     }
 125
 126     private void loadSnapShotToInMemoryGraph(JanusGraph graph, Properties graphProps) {
 127         if (logger.isDebugEnabled()) {
 128             logger.debug("Load Snapshot to InMemory Graph");
 129         }
 130         if (graphProps.containsKey("load.snapshot.file")) {
 131             String value = graphProps.getProperty("load.snapshot.file");
 132             if ("true".equals(value)) {
 133                 try (Graph transaction = graph.newTransaction()) {
 134                     String location = System.getProperty("snapshot.location");
 135                     logger.info("Loading snapshot to inmemory graph.");
 136                     transaction.io(IoCore.graphson()).readGraph(location);
 137                     transaction.tx().commit();
 138                     logger.info("Snapshot loaded to inmemory graph.");
 139                 } catch (Exception e) {
 140                     logger.info(String.format("ERROR: Could not load datasnapshot to in memory graph. %n%s",
 141                             ExceptionUtils.getStackTrace(e)));
 142                     throw new RuntimeException(e);
 143                 }
 144             }
 145         }
 146     }
 147
 148     private void loadSchema(JanusGraph graph) {
 149         // Load the propertyKeys, indexes and edge-Labels into the DB
 150         boolean dbNotEmpty = graph.traversal().V().limit(1).hasNext();
 151         logger.info("-- loading schema into JanusGraph");
 152         if ("true".equals(
 153             SpringContextAware.getApplicationContext().getEnvironment().getProperty("history.enabled", "false"))) {
 154             JanusGraphManagement graphMgt = graph.openManagement();
 155             SchemaGenerator4Hist.loadSchemaIntoJanusGraph(graphMgt, IN_MEMORY);
 156         } else {
 157             SchemaGenerator.loadSchemaIntoJanusGraph(graph, IN_MEMORY, dbNotEmpty);
 158         }
 159     }
 160
 161     /**
 162      * Close all of the graph connections made in the instance.
 163      */
 164     public void graphShutdown() {
 165         if (graph != null && graph.isOpen()) {
 166             graph.close();
 167         }
 168     }
 169
 170     /**
 171      * Gets the graph.
 172      *
 173      * @return the graph
 174      */
 175     public JanusGraph getGraph() {
 176         return graph;
 177     }
 178
 179 }