import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
public List<Person> findPersonsByPrefix(String firstNamePrefix) {

        String query = String.format(
                "g.V().hasLabel('%s').has('%s', Text.textContainsPrefix('%s')).valueMap(true)",
                PERSON.LABEL,
                PERSON.FIRST_NAME,
                firstNamePrefix);

        try {
            List<Result> results = client.submit(query).all().get();

            List<Person> people = new ArrayList<>(results.size());
            for(Result result : results) {
                people.add(toPerson(result.get(Map.class)));
            }

            return people;
        }
        catch (Exception ex) {
            throw new RuntimeException(ex);
        }
    }
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Test
public void shouldProcessRequestsOutOfOrder() throws Exception {
    final Cluster cluster = Cluster.open();
    final Client client = cluster.connect();

    final ResultSet rsFive = client.submit("Thread.sleep(5000);'five'");
    final ResultSet rsZero = client.submit("'zero'");

    final CompletableFuture<List<Result>> futureFive = rsFive.all();
    final CompletableFuture<List<Result>> futureZero = rsZero.all();

    final long start = System.nanoTime();
    assertFalse(futureFive.isDone());
    assertEquals("zero", futureZero.get().get(0).getString());

    logger.info("Eval of 'zero' complete: " + TimeUtil.millisSince(start));

    assertFalse(futureFive.isDone());
    assertEquals("five", futureFive.get(10, TimeUnit.SECONDS).get(0).getString());

    logger.info("Eval of 'five' complete: " + TimeUtil.millisSince(start));
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Test
public void shouldProcessSessionRequestsInOrder() throws Exception {
    final Cluster cluster = Cluster.open();
    final Client client = cluster.connect(name.getMethodName());

    final ResultSet rsFive = client.submit("Thread.sleep(5000);'five'");
    final ResultSet rsZero = client.submit("'zero'");

    final CompletableFuture<List<Result>> futureFive = rsFive.all();
    final CompletableFuture<List<Result>> futureZero = rsZero.all();

    final AtomicBoolean hit = new AtomicBoolean(false);
    while (!futureFive.isDone()) {
        // futureZero can't finish before futureFive - racy business here?
        assertThat(futureZero.isDone(), is(false));
        hit.set(true);
    }

    // should have entered the loop at least once and thus proven that futureZero didn't return ahead of
    // futureFive
    assertThat(hit.get(), is(true));

    assertEquals("zero", futureZero.get().get(0).getString());
    assertEquals("five", futureFive.get(10, TimeUnit.SECONDS).get(0).getString());
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Test
public void shouldIterate() throws Exception {
    final Cluster cluster = Cluster.open();
    final Client client = cluster.connect();

    final ResultSet results = client.submit("[1,2,3,4,5,6,7,8,9]");
    final Iterator<Result> itty = results.iterator();
    final AtomicInteger counter = new AtomicInteger(0);
    while (itty.hasNext()) {
        counter.incrementAndGet();
        assertEquals(counter.get(), itty.next().getInt());
    }

    assertEquals(9, counter.get());
    assertThat(results.allItemsAvailable(), is(true));

    // can't stream it again
    assertThat(results.iterator().hasNext(), is(false));

    cluster.close();
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Test
public void shouldSerializeToStringWhenRequested() throws Exception {
    final Map<String, Object> m = new HashMap<>();
    m.put("serializeResultToString", true);
    final GryoMessageSerializerV1d0 serializer = new GryoMessageSerializerV1d0();
    serializer.configure(m, null);

    final Cluster cluster = Cluster.build().serializer(serializer).create();
    final Client client = cluster.connect();

    final ResultSet resultSet = client.submit("TinkerFactory.createClassic()");
    final List<Result> results = resultSet.all().join();
    assertEquals(1, results.size());
    assertEquals("tinkergraph[vertices:6 edges:6]", results.get(0).getString());

    cluster.close();
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Test
public void shouldProcessRequestsOutOfOrder() throws Exception {
    final Cluster cluster = TestClientFactory.open();
    final Client client = cluster.connect();

    final ResultSet rsFive = client.submit("Thread.sleep(5000);'five'");
    final ResultSet rsZero = client.submit("'zero'");

    final CompletableFuture<List<Result>> futureFive = rsFive.all();
    final CompletableFuture<List<Result>> futureZero = rsZero.all();

    final long start = System.nanoTime();
    assertFalse(futureFive.isDone());
    assertEquals("zero", futureZero.get().get(0).getString());

    logger.info("Eval of 'zero' complete: " + TimeUtil.millisSince(start));

    assertFalse(futureFive.isDone());
    assertEquals("five", futureFive.get(10, TimeUnit.SECONDS).get(0).getString());

    logger.info("Eval of 'five' complete: " + TimeUtil.millisSince(start));
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Test
public void shouldIterate() throws Exception {
    final Cluster cluster = TestClientFactory.open();
    final Client client = cluster.connect();

    final ResultSet results = client.submit("[1,2,3,4,5,6,7,8,9]");
    final Iterator<Result> itty = results.iterator();
    final AtomicInteger counter = new AtomicInteger(0);
    while (itty.hasNext()) {
        counter.incrementAndGet();
        assertEquals(counter.get(), itty.next().getInt());
    }

    assertEquals(9, counter.get());
    assertThat(results.allItemsAvailable(), is(true));

    // can't stream it again
    assertThat(results.iterator().hasNext(), is(false));

    cluster.close();
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Test
public void shouldSerializeToStringWhenRequestedGryoV1() throws Exception {
    final Map<String, Object> m = new HashMap<>();
    m.put("serializeResultToString", true);
    final GryoMessageSerializerV1d0 serializer = new GryoMessageSerializerV1d0();
    serializer.configure(m, null);

    final Cluster cluster = TestClientFactory.build().serializer(serializer).create();
    final Client client = cluster.connect();

    final ResultSet resultSet = client.submit("TinkerFactory.createClassic()");
    final List<Result> results = resultSet.all().join();
    assertEquals(1, results.size());
    assertEquals("tinkergraph[vertices:6 edges:6]", results.get(0).getString());

    cluster.close();
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Test
public void shouldSerializeToStringWhenRequestedGryoV3() throws Exception {
    final Map<String, Object> m = new HashMap<>();
    m.put("serializeResultToString", true);
    final GryoMessageSerializerV3d0 serializer = new GryoMessageSerializerV3d0();
    serializer.configure(m, null);

    final Cluster cluster = TestClientFactory.build().serializer(serializer).create();
    final Client client = cluster.connect();

    final ResultSet resultSet = client.submit("TinkerFactory.createClassic()");
    final List<Result> results = resultSet.all().join();
    assertEquals(1, results.size());
    assertEquals("tinkergraph[vertices:6 edges:6]", results.get(0).getString());

    cluster.close();
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Test
public void shouldWorkWithGraphSONV2Serialization() throws Exception {
    final Cluster cluster = TestClientFactory.build().serializer(Serializers.GRAPHSON_V2D0).create();
    final Client client = cluster.connect();

    final List<Result> r = client.submit("TinkerFactory.createModern().traversal().V(1)").all().join();
    assertEquals(1, r.size());

    final Vertex v = r.get(0).get(DetachedVertex.class);
    assertEquals(1, v.id());
    assertEquals("person", v.label());

    assertEquals(2, IteratorUtils.count(v.properties()));
    assertEquals("marko", v.value("name"));
    assertEquals(29, Integer.parseInt(v.value("age").toString()));

    cluster.close();
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Test
public void shouldWorkWithGraphSONV3Serialization() throws Exception {
    final Cluster cluster = TestClientFactory.build().serializer(Serializers.GRAPHSON_V3D0).create();
    final Client client = cluster.connect();

    final List<Result> r = client.submit("TinkerFactory.createModern().traversal().V(1)").all().join();
    assertEquals(1, r.size());

    final Vertex v = r.get(0).get(DetachedVertex.class);
    assertEquals(1, v.id());
    assertEquals("person", v.label());

    assertEquals(2, IteratorUtils.count(v.properties()));
    assertEquals("marko", v.value("name"));
    assertEquals(29, Integer.parseInt(v.value("age").toString()));

    cluster.close();
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Test
public void shouldSubmitScript() throws InterruptedException, ExecutionException {
    List<Result> results = client.submit("g.V().valueMap()").all().get();
    assertThat(results, notNullValue());
    assertThat(results, not(empty()));
    results.stream().forEach(System.out::println);
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Override
public Object submit(final List<String> args) throws RemoteException {
    final String line = RemoteAcceptor.getScript(String.join(" ", args), this.shell);

    try {
        final List<Result> resultSet = send(line);
        this.shell.getInterp().getContext().setProperty(RESULT, resultSet);
        return resultSet.stream().map(result -> result.getObject()).iterator();
    } catch (SaslException sasl) {
        throw new RemoteException("Security error - check username/password and related settings", sasl);
    } catch (Exception ex) {
        final Optional<ResponseException> inner = findResponseException(ex);
        if (inner.isPresent()) {
            final ResponseException responseException = inner.get();
            if (responseException.getResponseStatusCode() == ResponseStatusCode.SERVER_ERROR_SERIALIZATION)
                throw new RemoteException(String.format("Server could not serialize the result requested. Server error - %s. Note that the class must be serializable by the client and server for proper operation.", responseException.getMessage()));
            else
                throw new RemoteException(responseException.getMessage());
        } else if (ex.getCause() != null) {
            final Throwable rootCause = ExceptionUtils.getRootCause(ex);
            if (rootCause instanceof TimeoutException)
                throw new RemoteException("Host did not respond in a timely fashion - check the server status and submit again.");
            else
                throw new RemoteException(rootCause.getMessage());
        } else
            throw new RemoteException(ex.getMessage());
    }
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
private List<Result> send(final String gremlin) throws SaslException {
    try {
        final ResultSet rs = this.currentClient.submitAsync(gremlin, aliases, Collections.emptyMap()).get();
        return timeout > NO_TIMEOUT ? rs.all().get(timeout, TimeUnit.MILLISECONDS) : rs.all().get();
    } catch(TimeoutException ignored) {
        throw new IllegalStateException("Request timed out while processing - increase the timeout with the :remote command");
    } catch (Exception e) {
        // handle security error as-is and unwrapped
        final Optional<Throwable> throwable  = Stream.of(ExceptionUtils.getThrowables(e)).filter(t -> t instanceof SaslException).findFirst();
        if (throwable.isPresent())
            throw (SaslException) throwable.get();

        throw new IllegalStateException(e.getMessage(), e);
    }
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Test
public void shouldConnectAndSubmitSession() throws Exception {
    assertThat(acceptor.connect(Arrays.asList(TestHelper.generateTempFileFromResource(this.getClass(), "remote.yaml", ".tmp").getAbsolutePath(), "session")).toString(), startsWith("Configured "));
    assertEquals("1", ((Iterator) acceptor.submit(Collections.singletonList("x = 1"))).next());
    assertEquals("0", ((Iterator) acceptor.submit(Collections.singletonList("x - 1"))).next());
    assertEquals("0", ((List<Result>) groovysh.getInterp().getContext().getProperty(DriverRemoteAcceptor.RESULT)).iterator().next().getString());
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Test
public void shouldConnectAndSubmitManagedSession() throws Exception {
    assertThat(acceptor.connect(Arrays.asList(TestHelper.generateTempFileFromResource(this.getClass(), "remote.yaml", ".tmp").getAbsolutePath(), "session-managed")).toString(), startsWith("Configured "));
    assertEquals("1", ((Iterator) acceptor.submit(Collections.singletonList("x = 1"))).next());
    assertEquals("0", ((Iterator) acceptor.submit(Collections.singletonList("x - 1"))).next());
    assertEquals("0", ((List<Result>) groovysh.getInterp().getContext().getProperty(DriverRemoteAcceptor.RESULT)).iterator().next().getString());
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Test
public void shouldConnectAndReturnVerticesWithAnAlias() throws Exception {
    assertThat(acceptor.connect(Collections.singletonList(TestHelper.generateTempFileFromResource(this.getClass(), "remote.yaml", ".tmp").getAbsolutePath())).toString(), startsWith("Configured "));
    acceptor.configure(Arrays.asList("alias", "x", "g"));
    assertThat(IteratorUtils.list(((Iterator<String>) acceptor.submit(Collections.singletonList("x.addVertex('name','stephen');x.addVertex('name','marko');x.traversal().V()")))), hasSize(2));
    assertThat(((List<Result>) groovysh.getInterp().getContext().getProperty(DriverRemoteAcceptor.RESULT)).stream().map(Result::getString).collect(Collectors.toList()), hasSize(2));
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Test
public void shouldConnectAndSubmitInSession() throws Exception {
    assertThat(acceptor.connect(Arrays.asList(TestHelper.generateTempFileFromResource(this.getClass(), "remote.yaml", ".tmp").getAbsolutePath(), "session")).toString(), startsWith("Configured "));
    assertEquals("2", ((Iterator) acceptor.submit(Collections.singletonList("x=1+1"))).next());
    assertEquals("2", ((List<Result>) groovysh.getInterp().getContext().getProperty(DriverRemoteAcceptor.RESULT)).iterator().next().getString());
    assertEquals("4", ((Iterator) acceptor.submit(Collections.singletonList("x+2"))).next());
    assertEquals("4", ((List<Result>) groovysh.getInterp().getContext().getProperty(DriverRemoteAcceptor.RESULT)).iterator().next().getString());
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Test
public void shouldConnectAndSubmitInNamedSession() throws Exception {
    assertThat(acceptor.connect(Arrays.asList(TestHelper.generateTempFileFromResource(this.getClass(), "remote.yaml", ".tmp").getAbsolutePath(), "session", "AAA")).toString(), startsWith("Configured "));
    assertEquals("2", ((Iterator) acceptor.submit(Collections.singletonList("x=1+1"))).next());
    assertEquals("2", ((List<Result>) groovysh.getInterp().getContext().getProperty(DriverRemoteAcceptor.RESULT)).iterator().next().getString());
    assertEquals("4", ((Iterator) acceptor.submit(Collections.singletonList("x+2"))).next());
    assertEquals("4", ((List<Result>) groovysh.getInterp().getContext().getProperty(DriverRemoteAcceptor.RESULT)).iterator().next().getString());
}
 

import org.apache.tinkerpop.gremlin.driver.Result; //导入依赖的package包/类
@Test
public void shouldEnsureSessionBindingsAreThreadSafe() throws Exception {
    final Cluster cluster = Cluster.build().minInProcessPerConnection(16).maxInProcessPerConnection(64).create();
    final Client client = cluster.connect(name.getMethodName());

    client.submitAsync("a=100;b=1000;c=10000;null");
    final int requests = 10000;
    final List<CompletableFuture<ResultSet>> futures = new ArrayList<>(requests);
    IntStream.range(0, requests).forEach(i -> {
        try {
            futures.add(client.submitAsync("a+b+c"));
        } catch (Exception ex) {
            throw new RuntimeException(ex);
        }
    });

    assertEquals(requests, futures.size());

    int counter = 0;
    for(CompletableFuture<ResultSet> f : futures) {
        final Result r = f.get().all().get(30000, TimeUnit.MILLISECONDS).get(0);
        assertEquals(11100, r.getInt());
        counter++;
    }

    assertEquals(requests, counter);

    cluster.close();
}