/*
* Copyright 2010-2014 Orient Technologies LTD (info--at--orientechnologies.com)
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.orientechnologies.orient.graph.console;
import com.orientechnologies.common.console.TTYConsoleReader;
import com.orientechnologies.common.console.annotation.ConsoleCommand;
import com.orientechnologies.common.console.annotation.ConsoleParameter;
import com.orientechnologies.orient.console.OConsoleDatabaseApp;
import com.orientechnologies.orient.core.command.OCommandExecutorNotFoundException;
import com.orientechnologies.orient.core.exception.OStorageException;
import com.orientechnologies.orient.graph.gremlin.OCommandGremlin;
import com.orientechnologies.orient.graph.gremlin.OGremlinHelper;
import com.orientechnologies.orient.graph.migration.OGraphMigration;
/**
* Gremlin specialized console.
*
* @author Luca Garulli (l.garulli--at--orientechnologies.com)
*
*/
public class OGremlinConsole extends OConsoleDatabaseApp {
public OGremlinConsole(final String[] args) {
super(args);
}
public static void main(final String[] args) {
int result;
try {
boolean tty = false;
try {
if (setTerminalToCBreak())
tty = true;
} catch (Exception e) {
}
final OConsoleDatabaseApp console = new OGremlinConsole(args);
if (tty)
console.setReader(new TTYConsoleReader());
result = console.run();
} finally {
try {
stty("echo");
} catch (Exception e) {
}
}
System.exit(result);
}
@Override
protected void onBefore() {
super.onBefore();
out.println("\nInstalling extensions for GREMLIN language v." + OGremlinHelper.getEngineVersion());
OGremlinHelper.global().create();
}
@Override
protected boolean isCollectingCommands(final String iLine) {
return super.isCollectingCommands(iLine) || iLine.startsWith("gremlin");
}
@ConsoleCommand(splitInWords = false, description = "Execute a GREMLIN script")
public void gremlin(@ConsoleParameter(name = "script-text", description = "The script text to execute") final String iScriptText) {
checkForDatabase();
if (iScriptText == null || iScriptText.length() == 0)
return;
currentResultSet.clear();
long start = System.currentTimeMillis();
try {
final Object result = currentDatabase.command(new OCommandGremlin(iScriptText)).execute();
float elapsedSeconds = (System.currentTimeMillis() - start) / 1000;
out.println("\n" + result);
out.printf("\nScript executed in %f sec(s).", elapsedSeconds);
} catch (OStorageException e) {
final Throwable cause = e.getCause();
if (cause instanceof OCommandExecutorNotFoundException)
out.printf("\nError: the GREMLIN command executor is not installed, check your configuration");
}
}
@ConsoleCommand(description = "Migrates graph from OMVRBTree to ORidBag")
public void upgradeGraph() {
OGraphMigration migration = new OGraphMigration(getCurrentDatabase(), this);
migration.execute();
message("Graph has been upgraded.");
}
}