IntenseFinance/nosqlbench
3
0
Fork 0
mirror of https://github.com/nosqlbench/nosqlbench.git synced 2024-12-24 08:00:00 -06:00
Code Issues Packages Projects Releases Wiki Activity

Fix conversion bug for fluent imports

Browse Source
This commit is contained in:
Jonathan Shook 2022-01-20 16:15:48 -06:00
parent 4faf74052a
commit ce4d61928a
2 changed files with 37 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

43
adapter-cqld4/src/main/java/io/nosqlbench/adapter/cqld4/opmappers/Cqld4FluentGraphOpMapper.java
View File

@ -16,18 +16,20 @@ import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversalSo
import org.codehaus.groovy.control.CompilerConfiguration;
import org.codehaus.groovy.control.customizers.ImportCustomizer;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.function.LongFunction;
import java.util.function.Supplier;
public class Cqld4FluentGraphOpMapper implements OpMapper<Op> {
public class Cqld4FluentGraphOpMapper implements OpMapper<Op> {
private final CqlSession session;
private GraphTraversalSource gtsPlaceHolder;
public Cqld4FluentGraphOpMapper(CqlSession session) {
this.session = session;
}
}
@Override
public OpDispenser<? extends Op> apply(ParsedOp cmd) {
@ -38,14 +40,20 @@ public class Cqld4FluentGraphOpMapper implements OpMapper<Op> {
CompilerConfiguration compilerConfiguration = new CompilerConfiguration();
List<String> imports = cmd.getStaticConfigOr("imports", List.of());
ImportCustomizer importer = new ImportCustomizer();
importer.addImports(imports.toArray(new String[0]));
compilerConfiguration.addCompilationCustomizers(importer);
cmd.getOptionalStaticValue("imports", List.class).ifPresent(
l -> {
ArrayList<String> stringList = new ArrayList<>();
l.forEach(o -> stringList.add(o.toString()));
String[] verifiedClasses = expandClassNames(l);
ImportCustomizer importer = new ImportCustomizer();
importer.addImports(verifiedClasses);
compilerConfiguration.addCompilationCustomizers(importer);
}
);
Supplier<Script> supplier = () -> {
groovy.lang.Binding groovyBindings = new Binding(new LinkedHashMap<String,Object>(Map.of("g",g)));
GroovyShell gshell = new GroovyShell(groovyBindings,compilerConfiguration);
groovy.lang.Binding groovyBindings = new Binding(new LinkedHashMap<String, Object>(Map.of("g", g)));
GroovyShell gshell = new GroovyShell(groovyBindings, compilerConfiguration);
return gshell.parse(scriptBodyWithRawVarRefs);
};
@ -54,4 +62,23 @@ public class Cqld4FluentGraphOpMapper implements OpMapper<Op> {
return new Cqld4FluentGraphOpDispenser(cmd, graphnameFunc, session, virtdataBindings, supplier);
}
private String[] expandClassNames(List l) {
ClassLoader loader = Cqld4FluentGraphOpMapper.class.getClassLoader();
List<String> classNames = new ArrayList<>();
for (Object name : l) {
String candidateName = name.toString();
if (candidateName.endsWith(".*")) {
throw new RuntimeException("You can not use wildcard package imports like '" + candidateName + "'");
}
try {
loader.loadClass(candidateName);
classNames.add(candidateName);
} catch (Exception e) {
throw new RuntimeException("Class '" + candidateName + "' was not found for fluent imports.");
}
}
return classNames.toArray(new String[0]);
}
}

2
adapter-cqld4/src/main/resources/activities/baselines/graph-wheels.yaml
View File

@ -123,6 +123,8 @@ blocks:
read:
type: fluent
graphname: <<graphname:graph_wheels>>
imports:
- "org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.__"
fluent: >-
g.V().hasLabel("device").has("deviceid", UUID.fromString({deviceid}))