public static SampleKBLoader empty(){ return new SampleKBLoader(null); }
@Override public Consumer<GraknTx> build(){ return (GraknTx tx) -> { SampleKBLoader.loadFromFile(tx, schemaFile); SampleKBLoader.loadFromFile(tx, dataFile); SampleKBLoader.loadFromFile(tx, rulesFile); }; } }
public GraknSession newSession() { GraknSession session = (GraknTestUtil.usingJanus()) ? EmbeddedGraknSession.createEngineSession(randomKeyspace()) : EmbeddedGraknSession.inMemory(randomKeyspace()); openedSessions.add(session); return session; }
public EmbeddedGraknTx<?> tx() { checkInContext(); return loader.tx(); }
private static SampleKBContext getContext(@Nullable Consumer<GraknTx> preLoad){ return new SampleKBContext(SampleKBLoader.preLoad(preLoad)); }
public void rollback() { checkInContext(); loader.rollback(); }
public EmbeddedGraknTx<?> tx(){ if(tx == null || tx.isClosed()){ //Load the graph if we need to if(!graphLoaded) { try(GraknTx tx = factory.open(GraknTxType.WRITE)){ load(tx); tx.commit(); graphLoaded = true; } } tx = factory.open(GraknTxType.WRITE); } return tx; }
private SampleKBLoader(@Nullable Consumer<GraknTx> preLoad){ EmbeddedGraknSession session = GraknTestUtil.usingTinker() ? EmbeddedGraknSession.inMemory(randomKeyspace()) : EmbeddedGraknSession.createEngineSession(randomKeyspace()); factory = GraknTxFactoryBuilder.getInstance().getFactory(session, false); this.preLoad = preLoad; }
public void rollback() { if (tx instanceof GraknTxTinker) { tx.close(); tx.clearGraph(); graphLoaded = false; } else if (!tx.isClosed()) { tx.close(); } tx = tx(); }
@Override protected void buildSchema(GraknTx tx) { SampleKBLoader.loadFromFile(tx, "ldbc-snb-schema.gql"); SampleKBLoader.loadFromFile(tx, "ldbc-snb-product-schema.gql"); }
public static SampleKBLoader preLoad(@Nullable Consumer<GraknTx> build){ return new SampleKBLoader(build); }
@Override protected void buildRules(GraknTx tx) { SampleKBLoader.loadFromFile(tx, "admission-rules.gql"); } }
@Override protected void buildInstances(GraknTx tx) { SampleKBLoader.loadFromFile(tx, "ldbc-snb-data.gql"); } }
@Override protected void buildRules(GraknTx tx) { SampleKBLoader.loadFromFile(tx, "ldbc-snb-rules.gql"); }
@Override public Consumer<GraknTx> build(){ return (GraknTx graph) -> { SampleKBLoader.loadFromFile(graph, gqlFile); buildExtensionalDB(graph, n, m); }; }
public static SampleKBContext load(String ... files){ return getContext((graknGraph) -> { for (String file : files) { SampleKBLoader.loadFromFile(graknGraph, file); } }); }
@Override public Consumer<GraknTx> build(){ return (GraknTx graph) -> { SampleKBLoader.loadFromFile(graph, gqlFile); buildExtensionalDB(graph, n, m); }; }
@Override public Consumer<GraknTx> build(){ return (GraknTx graph) -> { SampleKBLoader.loadFromFile(graph, gqlFile); buildExtensionalDB(graph, n, m); }; }
@Override public Consumer<GraknTx> build(){ return (GraknTx tx) -> { SampleKBLoader.loadFromFile(tx, gqlFile); buildExtensionalDB(tx, n, m); }; }
@Override public Consumer<GraknTx> build(){ return (GraknTx graph) -> { SampleKBLoader.loadFromFile(graph, gqlFile); buildExtensionalDB(graph, n); }; }