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 org.slf4j.Logger;
  26 import org.slf4j.LoggerFactory;
  27 import org.apache.commons.configuration.PropertiesConfiguration;
  28 import org.apache.commons.lang.exception.ExceptionUtils;
  29 import org.apache.tinkerpop.gremlin.structure.Graph;
  30 import org.apache.tinkerpop.gremlin.structure.io.IoCore;
  31 import org.janusgraph.core.JanusGraph;
  32 import org.janusgraph.core.JanusGraphFactory;
  33 import org.janusgraph.core.schema.JanusGraphManagement;
  34 import org.onap.aai.config.SpringContextAware;
  35 import org.onap.aai.dbgen.SchemaGenerator;
  36 import org.onap.aai.dbgen.SchemaGenerator4Hist;
  37 import org.onap.aai.exceptions.AAIException;
  38 import org.onap.aai.util.AAIConstants;
  39
  40 import java.io.FileNotFoundException;
  41 import java.util.Properties;
  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
  63     /**
  64      * Instantiates a new AAI graph.
  65      */
  66     private AAIGraph() {
  67         try {
  68             String serviceName = System.getProperty("aai.service.name", "NA");
  69             String rtConfig = System.getProperty("realtime.db.config");
  70             if (rtConfig == null) {
  71                 rtConfig = AAIConstants.REALTIME_DB_CONFIG;
  72             }
  73             this.loadGraph(rtConfig, serviceName);
  74         } catch (Exception e) {
  75             throw new RuntimeException("Failed to instantiate graphs", e);
  76         }
  77     }
  78
  79     private static class Helper {
  80         private static final AAIGraph INSTANCE = new AAIGraph();
  81
  82         private Helper() {
  83
  84         }
  85     }
  86
  87     /**
  88      * Gets the single instance of AAIGraph.
  89      *
  90      * @return single instance of AAIGraph
  91      */
  92     public static AAIGraph getInstance() {
  93         isInit = true;
  94         return Helper.INSTANCE;
  95     }
  96
  97     public static boolean isInit() {
  98         return isInit;
  99     }
 100
 101     private void loadGraph(String configPath, String serviceName) throws Exception {
 102         // Graph being opened by JanusGraphFactory is being placed in hashmap to be used later
 103         // These graphs shouldn't be closed until the application shutdown
 104         try {
 105             PropertiesConfiguration propertiesConfiguration = new AAIGraphConfig.Builder(configPath)
 106                     .forService(serviceName).withGraphType("realtime").buildConfiguration();
 107             graph = JanusGraphFactory.open(propertiesConfiguration);
 108
 109             Properties graphProps = new Properties();
 110             propertiesConfiguration.getKeys()
 111                     .forEachRemaining(k -> graphProps.setProperty(k, propertiesConfiguration.getString(k)));
 112
 113             if (IN_MEMORY.equals(graphProps.get("storage.backend"))) {
 114                 // Load the propertyKeys, indexes and edge-Labels into the DB
 115                 loadSchema(graph);
 116                 loadSnapShotToInMemoryGraph(graph, graphProps);
 117             }
 118
 119             if (graph == null) {
 120                 throw new AAIException("AAI_5102");
 121             }
 122
 123         } catch (FileNotFoundException e) {
 124             throw new AAIException("AAI_4001", e);
 125         }
 126     }
 127
 128     private void loadSnapShotToInMemoryGraph(JanusGraph graph, Properties graphProps) {
 129         if (logger.isDebugEnabled()) {
 130             logger.debug("Load Snapshot to InMemory Graph");
 131         }
 132         if (graphProps.containsKey("load.snapshot.file")) {
 133             String value = graphProps.getProperty("load.snapshot.file");
 134             if ("true".equals(value)) {
 135                 try (Graph transaction = graph.newTransaction()) {
 136                     String location = System.getProperty("snapshot.location");
 137                     logger.info("Loading snapshot to inmemory graph.");
 138                     transaction.io(IoCore.graphson()).readGraph(location);
 139                     transaction.tx().commit();
 140                     logger.info("Snapshot loaded to inmemory graph.");
 141                 } catch (Exception e) {
 142                     logger.info(String.format("ERROR: Could not load datasnapshot to in memory graph. %n%s", ExceptionUtils.getFullStackTrace(e)));
 143                     throw new RuntimeException(e);
 144                 }
 145             }
 146         }
 147     }
 148
 149     private void loadSchema(JanusGraph graph) {
 150         // Load the propertyKeys, indexes and edge-Labels into the DB
 151         JanusGraphManagement graphMgt = graph.openManagement();
 152
 153         logger.info("-- loading schema into JanusGraph");
 154         if ("true".equals(SpringContextAware.getApplicationContext().getEnvironment().getProperty("history.enabled", "false"))) {
 155             SchemaGenerator4Hist.loadSchemaIntoJanusGraph(graphMgt, IN_MEMORY);
 156         } else {
 157             SchemaGenerator.loadSchemaIntoJanusGraph(graphMgt, IN_MEMORY);
 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 }