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.commands;
21
22 import com.tinkerpop.blueprints.Graph;
23 import com.tinkerpop.frames.FramedGraph;
24 import eu.ehri.project.core.impl.Neo4jGraphManager;
25 import eu.ehri.project.core.impl.neo4j.Neo4j2Graph;
26 import org.apache.commons.cli.CommandLine;
27
28 /**
29 * Command for generating the (Neo4j) graph schema.
30 */
31 public class GenSchema extends BaseCommand {
32
33 final static String NAME = "gen-schema";
34
35 @Override
36 public String getUsage() {
37 return NAME;
38 }
39
40 @Override
41 public String getHelp() {
42 return "Drop and regenerate the (internal) graph schema and indices.";
43 }
44
45
46 @Override
47 public int execWithOptions(FramedGraph<?> graph, CommandLine cmdLine) throws Exception {
48 Graph baseGraph = graph.getBaseGraph();
49 if (baseGraph instanceof Neo4j2Graph) {
50 Neo4jGraphManager.createIndicesAndConstraints(
51 ((Neo4j2Graph) baseGraph).getRawGraph());
52 } else {
53 logger.warn("Cannot generate schema on a non-Neo4j2 graph");
54 }
55 return 0;
56 }
57 }