Examples with Keyspace ai.grakn.Keyspace used on opensource projects

Example 21 with Keyspace

use of ai.grakn.Keyspace in project grakn by graknlabs.

the class TxFactoryJanusTest method getGraph.

private static JanusGraph getGraph() {
    Keyspace name = Keyspace.of("hehe" + UUID.randomUUID().toString().replaceAll("-", ""));
    when(session.keyspace()).thenReturn(name);
    janusGraphFactory = new TxFactoryJanus(session);
    Graph graph = janusGraphFactory.open(GraknTxType.WRITE).getTinkerPopGraph();
    assertThat(graph, instanceOf(JanusGraph.class));
    return (JanusGraph) graph;
}

Example 22 with Keyspace

use of ai.grakn.Keyspace in project grakn by graknlabs.

the class GraknSessionProvider method getSession.

@Override
public GraknSession getSession(GraqlShellOptions options, ConsoleReader console) {
    int defaultGrpcPort = config.getProperty(GraknConfigKey.GRPC_PORT);
    SimpleURI defaultGrpcUri = new SimpleURI(Grakn.DEFAULT_URI.getHost(), defaultGrpcPort);
    SimpleURI location = options.getUri();
    SimpleURI uri = location != null ? location : defaultGrpcUri;
    Keyspace keyspace = options.getKeyspace();
    return RemoteGrakn.session(uri, keyspace);
}

Example 23 with Keyspace

use of ai.grakn.Keyspace in project grakn by graknlabs.

the class GraknClientFake method whenBatchExecutorClientCloses_AllTasksHaveCompleted.

@Test
public void whenBatchExecutorClientCloses_AllTasksHaveCompleted() throws GraknClientException {
    Keyspace keyspace = Keyspace.of("yes");
    GraknClientFake graknClient = new GraknClientFake();
    // Make sure there are more queries to execute than are allowed to run at once
    int maxQueries = 10;
    int numQueries = 100;
    BatchExecutorClient.Builder clientBuilder = BatchExecutorClient.newBuilder().taskClient(graknClient).maxQueries(maxQueries);
    Set<Query<?>> queriesToExecute = IntStream.range(0, numQueries).mapToObj(this::createInsertQuery).collect(toImmutableSet());
    try (BatchExecutorClient client = clientBuilder.build()) {
        for (Query<?> query : queriesToExecute) {
            // If we don't subscribe, the query won't execute
            client.add(query, keyspace).subscribe(a -> {
            });
        }
    }
    assertThat(graknClient.queriesExecuted(), containsInAnyOrder(queriesToExecute.toArray()));
}

Example 24 with Keyspace

use of ai.grakn.Keyspace in project grakn by graknlabs.

the class DocTestUtil method getTestGraph.

public static GraknSession getTestGraph(SimpleURI uri, String knowledgeBaseName) {
    Keyspace keyspace = SampleKBLoader.randomKeyspace();
    GraknSession session = Grakn.session(uri, keyspace);
    try (GraknTx tx = session.open(GraknTxType.WRITE)) {
        Consumer<GraknTx> loader = loaders.get(knowledgeBaseName);
        if (loader == null) {
            throw new IllegalArgumentException("Unknown knowledge base '" + knowledgeBaseName + "'");
        }
        loader.accept(tx);
        tx.commit();
    }
    return session;
}

Example 25 with Keyspace

use of ai.grakn.Keyspace in project grakn by graknlabs.

the class BenchmarkIT method loadRandomisedRelationInstances.

private void loadRandomisedRelationInstances(String entityLabel, String fromRoleLabel, String toRoleLabel, String relationLabel, int N, GraknSession session, GraknClient graknClient, Keyspace keyspace) {
    try (BatchExecutorClient loader = BatchExecutorClient.newBuilder().taskClient(graknClient).build()) {
        GraknTx tx = session.open(GraknTxType.READ);
        Var entityVar = var().asUserDefined();
        ConceptId[] instances = tx.graql().match(entityVar.isa(entityLabel)).get().execute().stream().map(ans -> ans.get(entityVar).getId()).toArray(ConceptId[]::new);
        assertEquals(instances.length, N);
        Role fromRole = tx.getRole(fromRoleLabel);
        Role toRole = tx.getRole(toRoleLabel);
        RelationshipType relationType = tx.getRelationshipType(relationLabel);
        Random rand = new Random();
        Multimap<Integer, Integer> assignmentMap = HashMultimap.create();
        for (int i = 0; i < N; i++) {
            int from = rand.nextInt(N - 1);
            int to = rand.nextInt(N - 1);
            while (to == from && assignmentMap.get(from).contains(to)) to = rand.nextInt(N - 1);
            Var fromRolePlayer = Graql.var();
            Var toRolePlayer = Graql.var();
            Pattern relationInsert = Graql.var().rel(Graql.label(fromRole.getLabel()), fromRolePlayer).rel(Graql.label(toRole.getLabel()), toRolePlayer).isa(Graql.label(relationType.getLabel())).and(fromRolePlayer.asUserDefined().id(instances[from])).and(toRolePlayer.asUserDefined().id(instances[to]));
            loader.add(Graql.insert(relationInsert.admin().varPatterns()), keyspace).subscribe();
        }
        tx.close();
    }
}