1 /*
2 * Copyright 2015 Data Archiving and Networked Services (an institute of
3 * Koninklijke Nederlandse Akademie van Wetenschappen), King's College London,
4 * Georg-August-Universitaet Goettingen Stiftung Oeffentlichen Rechts
5 *
6 * Licensed under the EUPL, Version 1.1 or – as soon they will be approved by
7 * the European Commission - subsequent versions of the EUPL (the "Licence");
8 * You may not use this work except in compliance with the Licence.
9 * You may obtain a copy of the Licence at:
10 *
11 * https://joinup.ec.europa.eu/software/page/eupl
12 *
13 * Unless required by applicable law or agreed to in writing, software
14 * distributed under the Licence is distributed on an "AS IS" basis,
15 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16 * See the Licence for the specific language governing
17 * permissions and limitations under the Licence.
18 */
19
20 package eu.ehri.project.core;
21
22 import com.tinkerpop.blueprints.Graph;
23 import com.tinkerpop.frames.FramedGraph;
24 import eu.ehri.project.core.impl.BlueprintsGraphManager;
25 import eu.ehri.project.core.impl.Neo4jGraphManager;
26 import eu.ehri.project.core.impl.neo4j.Neo4j2Graph;
27
28 /**
29 * A factory class for obtaining a {@link GraphManager} instance.
30 */
31 public class GraphManagerFactory {
32
33 /**
34 * Obtain an instance of a graph manager.
35 *
36 * @param graph A transactional Blueprints graph.
37 * @return A graph manager instance.
38 */
39 // NB: Because Java doesn't support multiple wildcard bounds we do some checking
40 // of the bounds manually ourselves, which is ugly but should ensure it's safe
41 // to do an unchecked cast here.
42 @SuppressWarnings("unchecked")
43 public static GraphManager getInstance(FramedGraph<?> graph) {
44 Graph baseGraph = graph.getBaseGraph();
45 if (Neo4j2Graph.class.isAssignableFrom(baseGraph.getClass())) {
46 return new Neo4jGraphManager(graph);
47 } else {
48 return new BlueprintsGraphManager(graph);
49 }
50 }
51 }