Search in sources :

Example 1 with DefaultRows

use of com.datastax.oss.protocol.internal.response.result.DefaultRows in project java-driver by datastax.

the class DseTestFixtures method tenDseRows.

// Returns 10 rows, each with a single "message" column with the value "hello, world"
public static Rows tenDseRows(int page, boolean last) {
    DseRowsMetadata metadata = new DseRowsMetadata(ImmutableList.of(new ColumnSpec("ks", "table", "message", 0, RawType.PRIMITIVES.get(ProtocolConstants.DataType.VARCHAR))), last ? null : ByteBuffer.wrap(new byte[] { (byte) page }), new int[] {}, null, page, last);
    Queue<List<ByteBuffer>> data = new ArrayDeque<>();
    for (int i = 0; i < 10; i++) {
        data.add(ImmutableList.of(Bytes.fromHexString("0x68656C6C6F2C20776F726C64")));
    }
    return new DefaultRows(metadata, data);
}
Also used : ColumnSpec(com.datastax.oss.protocol.internal.response.result.ColumnSpec) DefaultRows(com.datastax.oss.protocol.internal.response.result.DefaultRows) ImmutableList(com.datastax.oss.driver.shaded.guava.common.collect.ImmutableList) List(java.util.List) DseRowsMetadata(com.datastax.dse.protocol.internal.response.result.DseRowsMetadata) ArrayDeque(java.util.ArrayDeque)

Example 2 with DefaultRows

use of com.datastax.oss.protocol.internal.response.result.DefaultRows in project java-driver by datastax.

the class GraphTestUtils method singleGraphRow.

public static Message singleGraphRow(GraphProtocol graphProtocol, GraphBinaryModule module) throws IOException {
    Vertex value = DetachedVertex.build().setId(1).setLabel("person").addProperty(DetachedVertexProperty.build().setId(11).setLabel("name").setValue("marko").create()).create();
    DseRowsMetadata metadata = new DseRowsMetadata(ImmutableList.of(new ColumnSpec("ks", "table", "gremlin", 0, graphProtocol.isGraphBinary() ? RawType.PRIMITIVES.get(ProtocolConstants.DataType.BLOB) : RawType.PRIMITIVES.get(ProtocolConstants.DataType.VARCHAR))), null, new int[] {}, null, 1, true);
    Queue<List<ByteBuffer>> data = new ArrayDeque<>();
    data.add(ImmutableList.of(serialize(graphProtocol.isGraphBinary() ? // GraphBinary returns results directly inside a Traverser
    new DefaultRemoteTraverser<>(value, 1) : ImmutableMap.of("result", value), graphProtocol, module)));
    return new DefaultRows(metadata, data);
}
Also used : Vertex(org.apache.tinkerpop.gremlin.structure.Vertex) DetachedVertex(org.apache.tinkerpop.gremlin.structure.util.detached.DetachedVertex) ColumnSpec(com.datastax.oss.protocol.internal.response.result.ColumnSpec) DefaultRows(com.datastax.oss.protocol.internal.response.result.DefaultRows) ImmutableList(com.datastax.oss.driver.shaded.guava.common.collect.ImmutableList) List(java.util.List) DseRowsMetadata(com.datastax.dse.protocol.internal.response.result.DseRowsMetadata) DefaultRemoteTraverser(org.apache.tinkerpop.gremlin.process.remote.traversal.DefaultRemoteTraverser) ArrayDeque(java.util.ArrayDeque)

Example 3 with DefaultRows

use of com.datastax.oss.protocol.internal.response.result.DefaultRows in project java-driver by datastax.

the class CqlRequestHandlerTestBase method singleRow.

// Returns a single row, with a single "message" column with the value "hello, world"
protected static Message singleRow() {
    RowsMetadata metadata = new RowsMetadata(ImmutableList.of(new ColumnSpec("ks", "table", "message", 0, RawType.PRIMITIVES.get(ProtocolConstants.DataType.VARCHAR))), null, new int[] {}, null);
    Queue<List<ByteBuffer>> data = new ArrayDeque<>();
    data.add(ImmutableList.of(Bytes.fromHexString("0x68656C6C6F2C20776F726C64")));
    return new DefaultRows(metadata, data);
}
Also used : RowsMetadata(com.datastax.oss.protocol.internal.response.result.RowsMetadata) ColumnSpec(com.datastax.oss.protocol.internal.response.result.ColumnSpec) DefaultRows(com.datastax.oss.protocol.internal.response.result.DefaultRows) ImmutableList(com.datastax.oss.driver.shaded.guava.common.collect.ImmutableList) List(java.util.List) ArrayDeque(java.util.ArrayDeque)

Example 4 with DefaultRows

use of com.datastax.oss.protocol.internal.response.result.DefaultRows in project java-driver by datastax.

the class GraphTestUtils method tenGraphRows.

// Returns 10 rows, each with a vertex
public static Rows tenGraphRows(GraphProtocol graphProtocol, GraphBinaryModule module, int page, boolean last) throws IOException {
    DseRowsMetadata metadata = new DseRowsMetadata(ImmutableList.of(new ColumnSpec("ks", "table", "gremlin", 0, graphProtocol.isGraphBinary() ? RawType.PRIMITIVES.get(ProtocolConstants.DataType.BLOB) : RawType.PRIMITIVES.get(ProtocolConstants.DataType.VARCHAR))), null, new int[] {}, null, page, last);
    Queue<List<ByteBuffer>> data = new ArrayDeque<>();
    int start = (page - 1) * 10;
    for (int i = start; i < start + 10; i++) {
        Vertex v = DetachedVertex.build().setId("vertex" + i).setLabel("person").addProperty(DetachedVertexProperty.build().setId("property" + i).setLabel("name").setValue("user" + i).create()).create();
        data.add(ImmutableList.of(serialize(graphProtocol.isGraphBinary() ? // GraphBinary returns results directly inside a Traverser
        new DefaultRemoteTraverser<>(v, 1) : ImmutableMap.of("result", v), graphProtocol, module)));
    }
    return new DefaultRows(metadata, data);
}
Also used : Vertex(org.apache.tinkerpop.gremlin.structure.Vertex) DetachedVertex(org.apache.tinkerpop.gremlin.structure.util.detached.DetachedVertex) ColumnSpec(com.datastax.oss.protocol.internal.response.result.ColumnSpec) DefaultRows(com.datastax.oss.protocol.internal.response.result.DefaultRows) ImmutableList(com.datastax.oss.driver.shaded.guava.common.collect.ImmutableList) List(java.util.List) DseRowsMetadata(com.datastax.dse.protocol.internal.response.result.DseRowsMetadata) DefaultRemoteTraverser(org.apache.tinkerpop.gremlin.process.remote.traversal.DefaultRemoteTraverser) ArrayDeque(java.util.ArrayDeque)

Example 5 with DefaultRows

use of com.datastax.oss.protocol.internal.response.result.DefaultRows in project java-driver by datastax.

the class DseTestFixtures method singleDseRow.

// Returns a single row, with a single "message" column with the value "hello, world"
public static Rows singleDseRow() {
    DseRowsMetadata metadata = new DseRowsMetadata(ImmutableList.of(new ColumnSpec("ks", "table", "message", 0, RawType.PRIMITIVES.get(ProtocolConstants.DataType.VARCHAR))), null, new int[] {}, null, 1, true);
    Queue<List<ByteBuffer>> data = new ArrayDeque<>();
    data.add(ImmutableList.of(Bytes.fromHexString("0x68656C6C6F2C20776F726C64")));
    return new DefaultRows(metadata, data);
}
Also used : ColumnSpec(com.datastax.oss.protocol.internal.response.result.ColumnSpec) DefaultRows(com.datastax.oss.protocol.internal.response.result.DefaultRows) ImmutableList(com.datastax.oss.driver.shaded.guava.common.collect.ImmutableList) List(java.util.List) DseRowsMetadata(com.datastax.dse.protocol.internal.response.result.DseRowsMetadata) ArrayDeque(java.util.ArrayDeque)

Aggregations

ImmutableList (com.datastax.oss.driver.shaded.guava.common.collect.ImmutableList)7 ColumnSpec (com.datastax.oss.protocol.internal.response.result.ColumnSpec)7 DefaultRows (com.datastax.oss.protocol.internal.response.result.DefaultRows)7 List (java.util.List)7 ArrayDeque (java.util.ArrayDeque)6 DseRowsMetadata (com.datastax.dse.protocol.internal.response.result.DseRowsMetadata)4 RowsMetadata (com.datastax.oss.protocol.internal.response.result.RowsMetadata)3 DefaultRemoteTraverser (org.apache.tinkerpop.gremlin.process.remote.traversal.DefaultRemoteTraverser)2 Vertex (org.apache.tinkerpop.gremlin.structure.Vertex)2 DetachedVertex (org.apache.tinkerpop.gremlin.structure.util.detached.DetachedVertex)2