Examples with SimpleStatement com.datastax.driver.core.SimpleStatement used on opensource projects

Example 31 with SimpleStatement

use of com.datastax.driver.core.SimpleStatement in project cassandra by apache.

the class QueryEventsTest method batchTest.

@Test
public void batchTest() {
    createTable("create table %s (id int primary key, v int)");
    BatchMockListener listener = new BatchMockListener(getCurrentColumnFamilyStore());
    QueryEvents.instance.registerListener(listener);
    Session session = sessionNet();
    com.datastax.driver.core.BatchStatement batch = new com.datastax.driver.core.BatchStatement(com.datastax.driver.core.BatchStatement.Type.UNLOGGED);
    String q1 = formatQuery("insert into %s (id, v) values (?, ?)");
    PreparedStatement ps = session.prepare(q1);
    batch.add(ps.bind(1, 1));
    batch.add(ps.bind(2, 2));
    String q2 = formatQuery("insert into %s (id, v) values (1,1)");
    batch.add(new SimpleStatement(formatQuery("insert into %s (id, v) values (1,1)")));
    session.execute(batch);
    listener.verify(newArrayList("prepareSuccess", "batchSuccess"), newArrayList(1, 1));
    assertEquals(3, listener.queries.size());
    assertEquals(BatchStatement.Type.UNLOGGED, listener.batchType);
    assertEquals(newArrayList(q1, q1, q2), listener.queries);
    assertEquals(newArrayList(newArrayList(ByteBufferUtil.bytes(1), ByteBufferUtil.bytes(1)), newArrayList(ByteBufferUtil.bytes(2), ByteBufferUtil.bytes(2)), newArrayList(newArrayList())), listener.values);
    batch.add(new SimpleStatement("insert into abc.def (id, v) values (1,1)"));
    try {
        session.execute(batch);
        fail("Batch should fail");
    } catch (Exception e) {
    // ok
    }
    listener.verify(newArrayList("prepareSuccess", "batchSuccess", "batchFailure"), newArrayList(1, 1, 1));
    assertEquals(3, listener.queries.size());
    assertEquals(BatchStatement.Type.UNLOGGED, listener.batchType);
    assertEquals(newArrayList(q1, q1, q2), listener.queries);
    assertEquals(newArrayList(newArrayList(ByteBufferUtil.bytes(1), ByteBufferUtil.bytes(1)), newArrayList(ByteBufferUtil.bytes(2), ByteBufferUtil.bytes(2)), newArrayList(newArrayList()), newArrayList(newArrayList())), listener.values);
}

Example 32 with SimpleStatement

use of com.datastax.driver.core.SimpleStatement in project cassandra by apache.

the class PagingTest method testPaging.

/**
 * Makes sure that we don't drop any live rows when paging with DISTINCT queries
 *
 * * We need to have more rows than fetch_size
 * * The node must have a token within the first page (so that the range gets split up in StorageProxy#getRestrictedRanges)
 *   - This means that the second read in the second range will read back too many rows
 * * The extra rows are dropped (so that we only return fetch_size rows to client)
 * * This means that the last row recorded in AbstractQueryPager#recordLast is a non-live one
 * * For the next page, the first row returned will be the same non-live row as above
 * * The bug in CASSANDRA-14956 caused us to drop that non-live row + the first live row in the next page
 */
@Test
public void testPaging() throws InterruptedException {
    String table = KEYSPACE + ".paging";
    String createTableStatement = "CREATE TABLE IF NOT EXISTS " + table + " (id int, id2 int, id3 int, val text, PRIMARY KEY ((id, id2), id3));";
    String dropTableStatement = "DROP TABLE IF EXISTS " + table + ';';
    // custom snitch to avoid merging ranges back together after StorageProxy#getRestrictedRanges splits them up
    IEndpointSnitch snitch = new AbstractEndpointSnitch() {

        private IEndpointSnitch oldSnitch = DatabaseDescriptor.getEndpointSnitch();

        public int compareEndpoints(InetAddressAndPort target, Replica a1, Replica a2) {
            return oldSnitch.compareEndpoints(target, a1, a2);
        }

        public String getRack(InetAddressAndPort endpoint) {
            return oldSnitch.getRack(endpoint);
        }

        public String getDatacenter(InetAddressAndPort endpoint) {
            return oldSnitch.getDatacenter(endpoint);
        }

        @Override
        public boolean isWorthMergingForRangeQuery(ReplicaCollection merged, ReplicaCollection l1, ReplicaCollection l2) {
            return false;
        }
    };
    DatabaseDescriptor.setEndpointSnitch(snitch);
    StorageService.instance.getTokenMetadata().clearUnsafe();
    StorageService.instance.getTokenMetadata().updateNormalToken(new LongToken(5097162189738624638L), FBUtilities.getBroadcastAddressAndPort());
    session.execute(createTableStatement);
    for (int i = 0; i < 110; i++) {
        // removing row with idx 10 causes the last row in the first page read to be empty
        String ttlClause = i == 10 ? "USING TTL 1" : "";
        session.execute(String.format("INSERT INTO %s (id, id2, id3, val) VALUES (%d, %d, %d, '%d') %s", table, i, i, i, i, ttlClause));
    }
    Thread.sleep(1500);
    Statement stmt = new SimpleStatement(String.format("SELECT DISTINCT token(id, id2), id, id2 FROM %s", table));
    stmt.setFetchSize(100);
    ResultSet res = session.execute(stmt);
    stmt.setFetchSize(200);
    ResultSet res2 = session.execute(stmt);
    Iterator<Row> iter1 = res.iterator();
    Iterator<Row> iter2 = res2.iterator();
    while (iter1.hasNext() && iter2.hasNext()) {
        Row row1 = iter1.next();
        Row row2 = iter2.next();
        assertEquals(row1.getInt("id"), row2.getInt("id"));
    }
    assertFalse(iter1.hasNext());
    assertFalse(iter2.hasNext());
    session.execute(dropTableStatement);
}

Example 33 with SimpleStatement

use of com.datastax.driver.core.SimpleStatement in project cassandra by apache.

the class CqlInputFormat method queryTableEstimates.

private static ResultSet queryTableEstimates(Session session, Host host, String keyspace, String table, TokenRange tokenRange) {
    try {
        String query = String.format("SELECT mean_partition_size, partitions_count " + "FROM %s.%s " + "WHERE keyspace_name = ? AND table_name = ? AND range_type = '%s' AND range_start = ? AND range_end = ?", SchemaConstants.SYSTEM_KEYSPACE_NAME, SystemKeyspace.TABLE_ESTIMATES, SystemKeyspace.TABLE_ESTIMATES_TYPE_LOCAL_PRIMARY);
        Statement stmt = new SimpleStatement(query, keyspace, table, tokenRange.getStart().toString(), tokenRange.getEnd().toString()).setHost(host);
        return session.execute(stmt);
    } catch (InvalidQueryException e) {
        // if the table doesn't exist, fall back to old table.  This is likely to return no records in a multi
        // DC setup, but should work fine in a single DC setup.
        String query = String.format("SELECT mean_partition_size, partitions_count " + "FROM %s.%s " + "WHERE keyspace_name = ? AND table_name = ? AND range_start = ? AND range_end = ?", SchemaConstants.SYSTEM_KEYSPACE_NAME, SystemKeyspace.LEGACY_SIZE_ESTIMATES);
        Statement stmt = new SimpleStatement(query, keyspace, table, tokenRange.getStart().toString(), tokenRange.getEnd().toString()).setHost(host);
        return session.execute(stmt);
    }
}

Example 34 with SimpleStatement

use of com.datastax.driver.core.SimpleStatement in project cassandra by apache.

the class FQLReplayTest method testFQLQuerySingleToStatement.

@Test
public void testFQLQuerySingleToStatement() {
    List<ByteBuffer> values = new ArrayList<>();
    for (int i = 0; i < 10; i++) values.add(ByteBufferUtil.bytes(i));
    FQLQuery.Single single = new FQLQuery.Single("xyz", QueryOptions.DEFAULT.getProtocolVersion().asInt(), QueryOptions.forInternalCalls(values), 1234, 12345, 54321, "select * from aaa", values);
    Statement stmt = single.toStatement();
    assertEquals(stmt.getDefaultTimestamp(), 12345);
    assertTrue(stmt instanceof SimpleStatement);
    SimpleStatement simpleStmt = (SimpleStatement) stmt;
    assertEquals("select * from aaa", simpleStmt.getQueryString(CodecRegistry.DEFAULT_INSTANCE));
    assertArrayEquals(values.toArray(), simpleStmt.getValues(com.datastax.driver.core.ProtocolVersion.fromInt(QueryOptions.DEFAULT.getProtocolVersion().asInt()), CodecRegistry.DEFAULT_INSTANCE));
}

Example 35 with SimpleStatement

use of com.datastax.driver.core.SimpleStatement in project cassandra by apache.

the class SASICQLTest method testStringLikePrefix.

/**
 * Tests query condition like_prefix on string columns.
 */
@Test
public void testStringLikePrefix() throws Throwable {
    for (String mode : new String[] { "PREFIX", "CONTAINS" }) {
        for (boolean forceFlush : new boolean[] { false, true }) {
            try {
                createTable("CREATE TABLE %s (pk int primary key, v text);");
                createIndex(String.format("CREATE CUSTOM INDEX ON %%s (v) USING 'org.apache.cassandra.index.sasi.SASIIndex' WITH OPTIONS = {'mode': '%s'};", mode));
                execute("INSERT INTO %s (pk, v) VALUES (?, ?);", 0, "a");
                execute("INSERT INTO %s (pk, v) VALUES (?, ?);", 1, "abc");
                execute("INSERT INTO %s (pk, v) VALUES (?, ?);", 2, "ac");
                flush(forceFlush);
                Session session = sessionNet();
                SimpleStatement stmt = new SimpleStatement("SELECT * FROM " + KEYSPACE + '.' + currentTable() + " WHERE v LIKE 'ab%'");
                stmt.setFetchSize(5);
                List<Row> rs = session.execute(stmt).all();
                Assert.assertEquals(1, rs.size());
                Assert.assertEquals(1, rs.get(0).getInt("pk"));
            } catch (Throwable th) {
                throw new AssertionError(String.format("Failure with mode:%s and flush:%s ", mode, forceFlush), th);
            }
        }
    }
}