Search in sources :

Example 1 with ImmutableList

use of com.datastax.oss.driver.shaded.guava.common.collect.ImmutableList in project java-driver by datastax.

the class Reflection method buildFromConfigList.

/**
 * @param profileName if null, this is a global policy, use the default profile and look for a
 *     one-arg constructor. If not null, this is a per-profile policy, look for a two-arg
 *     constructor.
 */
public static <ComponentT> ImmutableList<ComponentT> buildFromConfigList(InternalDriverContext context, String profileName, DriverOption classNamesOption, Class<ComponentT> expectedSuperType, String... defaultPackages) {
    DriverExecutionProfile config = (profileName == null) ? context.getConfig().getDefaultProfile() : context.getConfig().getProfile(profileName);
    String configPath = classNamesOption.getPath();
    LOG.debug("Creating a list of {} from config option {}", expectedSuperType.getSimpleName(), configPath);
    if (!config.isDefined(classNamesOption)) {
        LOG.debug("Option is not defined, skipping");
        return ImmutableList.of();
    }
    List<String> classNames = config.getStringList(classNamesOption);
    ImmutableList.Builder<ComponentT> components = ImmutableList.builder();
    for (String className : classNames) {
        components.add(resolveClass(context, profileName, expectedSuperType, configPath, className, defaultPackages));
    }
    return components.build();
}
Also used : DriverExecutionProfile(com.datastax.oss.driver.api.core.config.DriverExecutionProfile) ImmutableList(com.datastax.oss.driver.shaded.guava.common.collect.ImmutableList)

Example 2 with ImmutableList

use of com.datastax.oss.driver.shaded.guava.common.collect.ImmutableList in project java-driver by datastax.

the class GraphTraversalITBase method should_handle_result_object_of_mixed_types.

/**
 * Ensures that a traversal that returns a result of mixed types is interpreted as a {@link Map}
 * with {@link Object} values. Also uses {@link
 * org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversal#by(org.apache.tinkerpop.gremlin.process.traversal.Traversal)}
 * with an anonymous traversal to get inbound 'created' edges and folds them into a list.
 *
 * <p>Executes a vertex traversal that binds label 'a' and 'b' to vertex properties and label 'c'
 * to vertices that have edges from that vertex.
 *
 * @test_category dse:graph
 */
@Test
public void should_handle_result_object_of_mixed_types() {
    // find all software vertices and select name, language, and find all vertices that created such
    // software.
    GraphResultSet rs = session().execute(newInstance(graphTraversalSource().V().hasLabel("software").as("a", "b", "c").select("a", "b", "c").by("name").by("lang").by(__.in("created").fold())));
    List<GraphNode> results = rs.all();
    assertThat(results.size()).isEqualTo(2);
    // Ensure that we got 'lop' and 'ripple' for property a.
    assertThat(results).extracting(m -> m.getByKey("a").as(Object.class)).containsOnly("lop", "ripple");
    for (GraphNode result : results) {
        // The row should represent a map with a, b, and c keys.
        assertThat(ImmutableList.<Object>copyOf(result.keys())).containsOnlyOnce("a", "b", "c");
        // 'e' should not exist, thus it should be null.
        assertThat(result.getByKey("e")).isNull();
        // both software are written in java.
        assertThat(result.getByKey("b").isNull()).isFalse();
        assertThat(result.getByKey("b").asString()).isEqualTo("java");
        GraphNode c = result.getByKey("c");
        assertThat(c.isList()).isTrue();
        if (result.getByKey("a").asString().equals("lop")) {
            if (isGraphBinary()) {
                // should contain three vertices
                Assertions.assertThat(c.size()).isEqualTo(3);
            } else {
                // 'c' should contain marko, josh, peter.
                // Ensure we have three vertices.
                assertThat(c.size()).isEqualTo(3);
                List<Vertex> vertices = Lists.newArrayList(c.getByIndex(0).asVertex(), c.getByIndex(1).asVertex(), c.getByIndex(2).asVertex());
                assertThat(vertices).extracting(vertex -> vertex.property("name").value()).containsOnly("marko", "josh", "peter");
            }
        } else {
            if (isGraphBinary()) {
                // has only one label
                Assertions.assertThat(c.size()).isEqualTo(1);
            } else {
                // ripple, 'c' should contain josh.
                // Ensure we have 1 vertex.
                assertThat(c.size()).isEqualTo(1);
                Vertex vertex = c.getByIndex(0).asVertex();
                assertThat(vertex).hasProperty("name", "josh");
            }
        }
    }
}
Also used : GraphNode(com.datastax.dse.driver.api.core.graph.GraphNode) AsyncGraphResultSet(com.datastax.dse.driver.api.core.graph.AsyncGraphResultSet) Graph(org.apache.tinkerpop.gremlin.structure.Graph) GenericType(com.datastax.oss.driver.api.core.type.reflect.GenericType) GraphTestUtils.assertThatContainsLabel(com.datastax.dse.driver.internal.core.graph.GraphTestUtils.assertThatContainsLabel) Tree(org.apache.tinkerpop.gremlin.process.traversal.step.util.Tree) GraphResultSet(com.datastax.dse.driver.api.core.graph.GraphResultSet) Version(com.datastax.oss.driver.api.core.Version) Lists(com.datastax.oss.driver.shaded.guava.common.collect.Lists) Assertions.assertThatThrownBy(org.assertj.core.api.Assertions.assertThatThrownBy) GraphStatement(com.datastax.dse.driver.api.core.graph.GraphStatement) CqlSession(com.datastax.oss.driver.api.core.CqlSession) Map(java.util.Map) TinkerGraphAssertions.assertThat(com.datastax.dse.driver.api.core.graph.TinkerGraphAssertions.assertThat) Assertions(org.assertj.core.api.Assertions) Path(org.apache.tinkerpop.gremlin.process.traversal.Path) InvalidQueryException(com.datastax.oss.driver.api.core.servererrors.InvalidQueryException) Edge(org.apache.tinkerpop.gremlin.structure.Edge) TinkerPathAssert.validatePathObjects(com.datastax.dse.driver.api.core.graph.TinkerPathAssert.validatePathObjects) org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.__(org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.__) Test(org.junit.Test) Vertex(org.apache.tinkerpop.gremlin.structure.Vertex) Collectors(java.util.stream.Collectors) GraphTraversal(org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversal) ImmutableList(com.datastax.oss.driver.shaded.guava.common.collect.ImmutableList) CustomCcmRule(com.datastax.oss.driver.api.testinfra.ccm.CustomCcmRule) ExecutionException(java.util.concurrent.ExecutionException) Assumptions(org.assertj.core.api.Assumptions) List(java.util.List) Direction(org.apache.tinkerpop.gremlin.structure.Direction) Assertions.fail(org.assertj.core.api.Assertions.fail) CompletionStage(java.util.concurrent.CompletionStage) GraphTestUtils.assertThatContainsProperties(com.datastax.dse.driver.internal.core.graph.GraphTestUtils.assertThatContainsProperties) SocialTraversalSource(com.datastax.dse.driver.api.core.graph.SocialTraversalSource) FluentGraphStatement.newInstance(com.datastax.dse.driver.api.core.graph.FluentGraphStatement.newInstance) FluentGraphStatement(com.datastax.dse.driver.api.core.graph.FluentGraphStatement) ScriptGraphStatement(com.datastax.dse.driver.api.core.graph.ScriptGraphStatement) GraphTraversalSource(org.apache.tinkerpop.gremlin.process.traversal.dsl.graph.GraphTraversalSource) Vertex(org.apache.tinkerpop.gremlin.structure.Vertex) AsyncGraphResultSet(com.datastax.dse.driver.api.core.graph.AsyncGraphResultSet) GraphResultSet(com.datastax.dse.driver.api.core.graph.GraphResultSet) GraphNode(com.datastax.dse.driver.api.core.graph.GraphNode) Test(org.junit.Test)

Aggregations

ImmutableList (com.datastax.oss.driver.shaded.guava.common.collect.ImmutableList)2 AsyncGraphResultSet (com.datastax.dse.driver.api.core.graph.AsyncGraphResultSet)1 FluentGraphStatement (com.datastax.dse.driver.api.core.graph.FluentGraphStatement)1 FluentGraphStatement.newInstance (com.datastax.dse.driver.api.core.graph.FluentGraphStatement.newInstance)1 GraphNode (com.datastax.dse.driver.api.core.graph.GraphNode)1 GraphResultSet (com.datastax.dse.driver.api.core.graph.GraphResultSet)1 GraphStatement (com.datastax.dse.driver.api.core.graph.GraphStatement)1 ScriptGraphStatement (com.datastax.dse.driver.api.core.graph.ScriptGraphStatement)1 SocialTraversalSource (com.datastax.dse.driver.api.core.graph.SocialTraversalSource)1 TinkerGraphAssertions.assertThat (com.datastax.dse.driver.api.core.graph.TinkerGraphAssertions.assertThat)1 TinkerPathAssert.validatePathObjects (com.datastax.dse.driver.api.core.graph.TinkerPathAssert.validatePathObjects)1 GraphTestUtils.assertThatContainsLabel (com.datastax.dse.driver.internal.core.graph.GraphTestUtils.assertThatContainsLabel)1 GraphTestUtils.assertThatContainsProperties (com.datastax.dse.driver.internal.core.graph.GraphTestUtils.assertThatContainsProperties)1 CqlSession (com.datastax.oss.driver.api.core.CqlSession)1 Version (com.datastax.oss.driver.api.core.Version)1 DriverExecutionProfile (com.datastax.oss.driver.api.core.config.DriverExecutionProfile)1 InvalidQueryException (com.datastax.oss.driver.api.core.servererrors.InvalidQueryException)1 GenericType (com.datastax.oss.driver.api.core.type.reflect.GenericType)1 CustomCcmRule (com.datastax.oss.driver.api.testinfra.ccm.CustomCcmRule)1 Lists (com.datastax.oss.driver.shaded.guava.common.collect.Lists)1