Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
org.umlg.runtime.adaptor.GroovyExecutor Maven / Gradle / Ivy
package org.umlg.runtime.adaptor;
import groovy.lang.Writable;
import groovy.text.SimpleTemplateEngine;
import org.apache.commons.lang3.time.StopWatch;
import org.apache.tinkerpop.gremlin.groovy.engine.GremlinExecutor;
import org.apache.tinkerpop.gremlin.jsr223.ImportGremlinPlugin;
import org.umlg.runtime.domain.UmlgNode;
import java.lang.reflect.Field;
import java.util.*;
/**
* Date: 2015/06/24
* Time: 1:06 PM
*/
public class GroovyExecutor {
public static GroovyExecutor INSTANCE = new GroovyExecutor();
private GremlinExecutor gremlinExecutor;
private GroovyExecutor() {
start();
}
public Object executeGroovy(Object context, String groovy) {
// groovy = groovy.replaceAll("::", "____");
if (context != null) {
if (context instanceof UmlgNode) {
Object id = ((UmlgNode) context).getId();
groovy = groovy.replaceAll("self(?=([^\"']*[\"'][^\"']*[\"'])*[^\"']*$)", "g.V(\"" + id.toString() + "\")");
} else if (!(context instanceof Long)) {
groovy = groovy.replaceAll("self(?=([^\"']*[\"'][^\"']*[\"'])*[^\"']*$)", "g.V(\"" + context.toString() + "\")");
} else if (context instanceof Long) {
groovy = groovy.replaceAll("self(?=([^\"']*[\"'][^\"']*[\"'])*[^\"']*$)", "g.V(\"" + context.toString() + "L\")");
} else {
groovy = groovy.replaceAll("self(?=([^\"']*[\"'][^\"']*[\"'])*[^\"']*$)", "g.V(" + context.toString() + ")");
}
}
try {
return eval(groovy);
} catch (Exception e) {
throw UmlgExceptionUtilFactory.getTumlExceptionUtil().handle(e);
}
}
/**
* Executes a groovy query. If the contextId is null then it is ignored.
* If it is not null then all instances of the keywork "this" will be replaced with "g.v(contextId)"
*
* @param contextId
* @param groovy
* @return
*/
public String executeGroovyAsString(Object contextId, String groovy) {
StopWatch stopWatch = new StopWatch();
stopWatch.start();
Object result = executeGroovy(contextId, groovy);
Iterator tempIterator = Collections.emptyIterator();
StringBuilder sb = new StringBuilder();
while (true) {
if (tempIterator.hasNext()) {
while (tempIterator.hasNext()) {
final Object object = tempIterator.next();
sb.append(((null == object) ? null : object.toString()));
sb.append("\n");
}
break;
} else {
try {
if (result instanceof Iterator) {
tempIterator = (Iterator) result;
if (!tempIterator.hasNext()) break;
} else if (result instanceof Iterable) {
tempIterator = ((Iterable) result).iterator();
if (!tempIterator.hasNext()) break;
} else if (result instanceof Object[]) {
tempIterator = new ArrayIterator((Object[]) result);
if (!tempIterator.hasNext()) break;
} else if (result instanceof Map) {
tempIterator = ((Map) result).entrySet().iterator();
if (!tempIterator.hasNext()) break;
} else {
sb.append(((null == result) ? null : result.toString()));
break;
}
} catch (final Exception e) {
throw e;
}
}
}
stopWatch.stop();
sb.append("Time to execute query = ");
sb.append(stopWatch.toString());
return sb.toString();
}
public String evalAsTemplate(String script, Map bindings) throws Exception {
bindings.put("g", UMLG.get().getUnderlyingGraph().traversal());
SimpleTemplateEngine engine = new SimpleTemplateEngine();
Writable w = engine.createTemplate(script).make(bindings);
return w.toString();
}
public Object eval(String script) throws Exception {
Map bindings = new HashMap<>();
bindings.put("g", UMLG.get().getUnderlyingGraph().traversal());
return this.gremlinExecutor.eval(script, bindings).get();
}
public Object eval(String script, Map bindings) throws Exception {
bindings.put("g", UMLG.get().getUnderlyingGraph().traversal());
return this.gremlinExecutor.eval(script, bindings).get();
}
public void close() throws Exception {
this.gremlinExecutor.close();
}
public void restart() throws Exception {
this.gremlinExecutor.close();
INSTANCE = new GroovyExecutor();
}
private void start() {
List imports = new ArrayList<>();
List staticImports = new ArrayList<>();
Class umlgGroovyImporter;
try {
umlgGroovyImporter = Class.forName("org.umlg.runtime.adaptor.UmlgGroovyImporter");
//TODO this is to slow for every unit tests
Field importsField = umlgGroovyImporter.getField("imports");
imports.addAll((Set) importsField.get(null));
imports.addAll(umlgAdaptorClasses());
// Field importsStaticField = umlgGroovyImporter.getField("importStatic");
// staticImports.addAll((Set) importsStaticField.get(null));
final Map> config = new HashMap<>();
// final Map scriptPluginConfig = new HashMap<>();
// scriptPluginConfig.put("files", Collections.singletonList("/usr/share/rorotika/cm/groovy/GremlinExecutorInit.groovy"));
// config.put(ScriptFileGremlinPlugin.class.getName(), scriptPluginConfig);
final Map importGremlinPluginConfig = new HashMap<>();
importGremlinPluginConfig.put("classImports", imports);
config.put(ImportGremlinPlugin.class.getName(), importGremlinPluginConfig);
this.gremlinExecutor = GremlinExecutor.build()
.addPlugins("gremlin-groovy", config)
// .scriptEvaluationTimeout(1000 * 180)
.afterSuccess(
t -> UMLG.get().rollback()
)
.afterFailure(
(t, e) -> UMLG.get().rollback()
)
.create();
} catch (Exception e) {
throw new RuntimeException("Failed to initialize GremlinExecutor", e);
}
}
private List umlgAdaptorClasses() {
return Arrays.asList(
ArrayIterator.class.getName(),
DefaultDataCreator.class.getName(),
EdgeSchemaCreator.class.getName(),
GroovyExecutor.class.getName(),
StringArrayContains.class.getName(),
TransactionIdentifier.class.getName(),
TransactionThreadEntityVar.class.getName(),
TransactionThreadMetaNodeVar.class.getName(),
TransactionThreadNotificationVar.class.getName(),
UMLG.class.getName(),
UmlgAdaptorImplementation.class.getName(),
UmlgAdminApp.class.getName(),
UmlgAdminAppFactory.class.getName(),
UmlgAssociationClassManager.class.getName(),
UmlgDefaultDbExceptionUtilImpl.class.getName(),
UmlgDefaultLabelConverter.class.getName(),
UmlgDefaultQualifierId.class.getName(),
UmlgExceptionUtil.class.getName(),
UmlgExceptionUtilFactory.class.getName(),
UmlgGraph.class.getName(),
UmlgGraphManager.class.getName(),
UmlGIndexFactory.class.getName(),
UmlgIndexManager.class.getName(),
UmlgLabelConverter.class.getName(),
UmlgLabelConverterFactory.class.getName(),
UmlgMetaNodeManager.class.getName(),
UmlgParameter.class.getName(),
UmlgQualifierId.class.getName(),
UmlgQualifierIdFactory.class.getName(),
UmlgQueryEnum.class.getName(),
UmlgSchemaCreator.class.getName(),
UmlgSchemaCreatorFactory.class.getName(),
UmlgSchemaFactory.class.getName(),
UmlgSchemaMap.class.getName(),
UmlgTmpIdManager.class.getName(),
UmlgTransactionEventHandler.class.getName(),
UmlgTransactionEventHandlerImpl.class.getName(),
VertexSchemaCreator.class.getName()
);
}
}
