diff --git a/adapter-cqld4/src/main/java/io/nosqlbench/adapter/cqld4/opmappers/Cqld4FluentGraphOpMapper.java b/adapter-cqld4/src/main/java/io/nosqlbench/adapter/cqld4/opmappers/Cqld4FluentGraphOpMapper.java index 9a1c44396..e9a1b93e2 100644 --- a/adapter-cqld4/src/main/java/io/nosqlbench/adapter/cqld4/opmappers/Cqld4FluentGraphOpMapper.java +++ b/adapter-cqld4/src/main/java/io/nosqlbench/adapter/cqld4/opmappers/Cqld4FluentGraphOpMapper.java @@ -13,8 +13,11 @@ import io.nosqlbench.virtdata.core.bindings.Bindings; import io.nosqlbench.virtdata.core.bindings.BindingsTemplate; import io.nosqlbench.virtdata.core.templates.ParsedTemplate; import org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversalSource; +import org.codehaus.groovy.control.CompilerConfiguration; +import org.codehaus.groovy.control.customizers.ImportCustomizer; import java.util.LinkedHashMap; +import java.util.List; import java.util.Map; import java.util.function.LongFunction; import java.util.function.Supplier; @@ -32,18 +35,22 @@ public class Cqld4FluentGraphOpMapper implements OpMapper { ParsedTemplate fluent = cmd.getAsTemplate("fluent").orElseThrow(); String scriptBodyWithRawVarRefs = fluent.getPositionalStatement(); + + CompilerConfiguration compilerConfiguration = new CompilerConfiguration(); + + List imports = cmd.getStaticConfigOr("imports", List.of()); + ImportCustomizer importer = new ImportCustomizer(); + importer.addImports(imports.toArray(new String[0])); + compilerConfiguration.addCompilationCustomizers(importer); + Supplier