Java SimpleStatement类代码示例

import com.datastax.driver.core.SimpleStatement; //导入依赖的package包/类
@Override
public DBResult queryForLastWriteTimes()
{
	if (cluster == null)
		return null;

	try (Session session = cluster.newSession())
	{
		Statement statement = new SimpleStatement(String.format("SELECT WRITETIME (%s) FROM %s.%s;", columnName, keyspace, tableName));
		statement.setConsistencyLevel(ConsistencyLevel.ALL);
		ResultSet results = session.execute(statement);
		List<Row> allRows = results.all();

		// sort all of the rows accordingly
		allRows.sort(new RowComparator());

		// gather the information we need
		long startTime = allRows.get(0).getLong(0) / 1000;
		long endTime = allRows.get(allRows.size() - 1).getLong(0) / 1000;
		int totalCount = allRows.size();
		return new DBResult(startTime, endTime, totalCount);
	}
}
 

import com.datastax.driver.core.SimpleStatement; //导入依赖的package包/类
private void primeAndExecuteQueries(String[] primed, String[] queries) throws Exception {
  SuccessResult result = getSampleSuccessResult();
  for (String primeQuery : primed) {
    server.prime(when(primeQuery).then(result));
  }

  try (com.datastax.driver.core.Cluster driverCluster =
      defaultBuilder(server.getCluster())
          .withRetryPolicy(FallthroughRetryPolicy.INSTANCE)
          .build()) {
    Session session = driverCluster.connect();
    server.getCluster().clearLogs();
    for (String executeQuery : queries) {
      SimpleStatement stmt = new SimpleStatement(executeQuery);
      stmt.setDefaultTimestamp(100);
      session.execute(stmt);
    }
  }
}
 

import com.datastax.driver.core.SimpleStatement; //导入依赖的package包/类
public void executePut(String query, String consistency){
	logger.debug("in data store handle, executing put:"+query);
	long start = System.currentTimeMillis();
	Statement statement = new SimpleStatement(query);
	if(consistency.equalsIgnoreCase("critical")){
		logger.info("Executing critical put query:"+query);
		statement.setConsistencyLevel(ConsistencyLevel.QUORUM);
	}
	else if (consistency.equalsIgnoreCase("eventual")){
		logger.info("Executing normal put query:"+query);
		statement.setConsistencyLevel(ConsistencyLevel.ONE);
	}
	session.execute(statement); 
	long end = System.currentTimeMillis();
	logger.debug("Time taken for actual put in cassandra:"+(end-start));
}
 

import com.datastax.driver.core.SimpleStatement; //导入依赖的package包/类
private ArgumentMatcher<Statement> getHostStatementMatcher(final Host host, final String query)
        throws Exception {
    return new ArgumentMatcher<Statement>() {
        @Override
        public boolean matches(Object argument) {
            SelectedHostStatement statement = (SelectedHostStatement) argument;

            return ((SimpleStatement)statement.getStatement()).getQueryString().equals(query) &&
                    Objects.equals(statement.getHostCordinator().getAddress(), host.getAddress());
        }

        @Override
        public void describeTo(Description description) {
            description.appendText(format("query:%s host:%s", query, host.getAddress().toString()));
        }
    };
}
 

import com.datastax.driver.core.SimpleStatement; //导入依赖的package包/类
@Test
public void should_generate_simple_statement() throws Exception {
    //Given
    String input = "SELECT * FROM users LIMIT 10;";
    CassandraQueryOptions options = new CassandraQueryOptions(Option.apply(QUORUM),
            Option.<ConsistencyLevel>empty(),
            Option.empty(),
            Option.<RetryPolicy>empty(),
            Option.empty());

    //When
    final SimpleStatement actual = helper.generateSimpleStatement(new SimpleStm(input), options, intrContext);

    //Then
    assertThat(actual).isNotNull();
    assertThat(actual.getQueryString()).isEqualTo("SELECT * FROM users LIMIT 10;");
    assertThat(actual.getConsistencyLevel()).isSameAs(QUORUM);
}
 

import com.datastax.driver.core.SimpleStatement; //导入依赖的package包/类
@Test
public void should_generate_batch_statement() throws Exception {
    //Given
    Statement st1 = new SimpleStatement("SELECT * FROM users LIMIT 10;");
    Statement st2 = new SimpleStatement("INSERT INTO users(id) VALUES(10);");
    Statement st3 = new SimpleStatement("UPDATE users SET name = 'John DOE' WHERE id=10;");
    CassandraQueryOptions options = new CassandraQueryOptions(Option.apply(QUORUM),
            Option.<ConsistencyLevel>empty(),
            Option.empty(),
            Option.<RetryPolicy>empty(),
            Option.empty());

    //When
    BatchStatement actual = helper.generateBatchStatement(UNLOGGED, options, toScalaList(asList(st1, st2, st3)));

    //Then
    assertThat(actual).isNotNull();
    final List<Statement> statements = new ArrayList<>(actual.getStatements());
    assertThat(statements).hasSize(3);
    assertThat(statements.get(0)).isSameAs(st1);
    assertThat(statements.get(1)).isSameAs(st2);
    assertThat(statements.get(2)).isSameAs(st3);
    assertThat(actual.getConsistencyLevel()).isSameAs(QUORUM);
}
 

import com.datastax.driver.core.SimpleStatement; //导入依赖的package包/类
/**
 * Build schema programmatically
 * <p>
 * DDL equivalent:
 * 
 * <pre>
 * CREATE TABLE messages (
 *   sessionId uuid,
 *   seqNo bigint,
 *   message blob,
 *   PRIMARY KEY  (sessionId, seqNo ) );
 * </pre>
 * 
 * @throws StoreException if the store is not open
 *
 */
public void buildSchema() throws StoreException {
  if (session != null) {
    // Appropriate for a local test only
    session.execute(new SimpleStatement("CREATE KEYSPACE IF NOT EXISTS " + KEYSPACE_NAME
        + " WITH REPLICATION = { 'class' : 'SimpleStrategy', 'replication_factor' : 1 }"));
    System.out.format("Keyspace %s available\n", KEYSPACE_NAME);

    Create create = SchemaBuilder.createTable(KEYSPACE_NAME, TABLE_NAME).ifNotExists()
        .addPartitionKey(SESSION_ID_COLNAME, DataType.uuid())
        .addClusteringColumn(SEQ_NO_COLNAME, DataType.bigint())
        .addColumn(MESSAGE_COLNAME, DataType.blob());

    ResultSet resultSet = session.execute(create);
    System.out.format("Table %s available\n", TABLE_NAME);
  } else {
    throw new StoreException("Schema not created; store not open");
  }
}
 

import com.datastax.driver.core.SimpleStatement; //导入依赖的package包/类
@Test
public void should_generate_simple_statement() throws Exception {
    //Given
    String input = "SELECT * FROM users LIMIT 10;";
    CassandraQueryOptions options = new CassandraQueryOptions(Option.apply(QUORUM),
            Option.<ConsistencyLevel>empty(),
            Option.empty(),
            Option.<RetryPolicy>empty(),
            Option.empty(),
            Option.empty());

    //When
    final SimpleStatement actual = helper.generateSimpleStatement(new SimpleStm(input), options, intrContext);

    //Then
    assertThat(actual).isNotNull();
    assertThat(actual.getQueryString()).isEqualTo("SELECT * FROM users LIMIT 10;");
    assertThat(actual.getConsistencyLevel()).isSameAs(QUORUM);
}
 

import com.datastax.driver.core.SimpleStatement; //导入依赖的package包/类
@Test
public void should_generate_batch_statement() throws Exception {
    //Given
    Statement st1 = new SimpleStatement("SELECT * FROM users LIMIT 10;");
    Statement st2 = new SimpleStatement("INSERT INTO users(id) VALUES(10);");
    Statement st3 = new SimpleStatement("UPDATE users SET name = 'John DOE' WHERE id=10;");
    CassandraQueryOptions options = new CassandraQueryOptions(Option.apply(QUORUM),
            Option.<ConsistencyLevel>empty(),
            Option.empty(),
            Option.<RetryPolicy>empty(),
            Option.empty(),
            Option.empty());

    //When
    BatchStatement actual = helper.generateBatchStatement(UNLOGGED, options, toScalaList(asList(st1, st2, st3)));

    //Then
    assertThat(actual).isNotNull();
    final List<Statement> statements = new ArrayList<>(actual.getStatements());
    assertThat(statements).hasSize(3);
    assertThat(statements.get(0)).isSameAs(st1);
    assertThat(statements.get(1)).isSameAs(st2);
    assertThat(statements.get(2)).isSameAs(st3);
    assertThat(actual.getConsistencyLevel()).isSameAs(QUORUM);
}
 

import com.datastax.driver.core.SimpleStatement; //导入依赖的package包/类
static void load(SessionContext sessionContext, List<String> cqlStatements) {
    if (!cqlStatements.isEmpty()) {
        sessionContext.checkClusterHealth();
    }
    try {
        cqlStatements.stream()
                .map(stringStatement -> new SimpleStatement(stringStatement).setConsistencyLevel(sessionContext.getWriteConsistencyLevel()))
                .forEach(statement -> {
                    LOGGER.debug("Executing cql statement {}", statement);
                    sessionContext.getSession().execute(statement);
                });
    } catch (DriverException e) {
        LOGGER.error("Failed to execute cql statements {}: {}", cqlStatements, e.getMessage());
        throw e;
    }
}
 

import com.datastax.driver.core.SimpleStatement; //导入依赖的package包/类
public int[] executeBatch() throws SQLException
  {
  	int[] returnCounts= new int[batchQueries.size()];
  	List<ResultSetFuture> futures = new ArrayList<ResultSetFuture>();
  	if (logger.isTraceEnabled() || this.connection.debugMode) logger.debug("CQL statements: "+ batchQueries.size());
  	for(String q:batchQueries){
  		if (logger.isTraceEnabled() || this.connection.debugMode) logger.debug("CQL: "+ q);
  		SimpleStatement stmt = new SimpleStatement(q);
  		stmt.setConsistencyLevel(this.connection.defaultConsistencyLevel);
  		ResultSetFuture resultSetFuture = this.connection.getSession().executeAsync(stmt);
  		futures.add(resultSetFuture);
  	}

  	int i=0;
for (ResultSetFuture future : futures){
	future.getUninterruptibly();
	returnCounts[i]=1;
	i++;
}
      
      return returnCounts;
  }
 

import com.datastax.driver.core.SimpleStatement; //导入依赖的package包/类
public ResultSet getRows(String keyspace, String table, int limit,
		String startKey, ConsistencyLevel level) {
	StringBuilder command = new StringBuilder();
	command.append("SELECT * from ").append(keyspace).append(".")
			.append(table);
	if (startKey != null) {
		command.append(" WHERE token(key)>token('").append(startKey)
				.append("')");
	}

	if (limit > 0) {
		command.append(" LIMIT ").append(Integer.toString(limit));
	}
	command.append(";");
	SimpleStatement ss = new SimpleStatement(command.toString());
	ss.setConsistencyLevel(level);
	ResultSet result = executeQuery(ss);
	return result;
}
 

import com.datastax.driver.core.SimpleStatement; //导入依赖的package包/类
public ResultSet getColumnFromRow(String keyspace, String tableName,
		String key, String column, ConsistencyLevel level) {

	StringBuilder command = new StringBuilder();

	// Build SELECT query
	command.append("SELECT ").append(column).append(" FROM ")
			.append((keyspace == null) ? keyspaceName : keyspace)
			.append(".").append(tableName).append(" WHERE ").append(KEY)
			.append(" = '").append(key).append("';");

	SimpleStatement ss = new SimpleStatement(command.toString());
	ss.setConsistencyLevel(level);

	ResultSet result = executeQuery(ss);

	return result;
}
 

import com.datastax.driver.core.SimpleStatement; //导入依赖的package包/类
public ResultSet getColumnsFromRow(String keyspace, String tableName,
		String key, ArrayList<String> columns, ConsistencyLevel level) {

	StringBuilder command = new StringBuilder();
	StringBuilder cols = new StringBuilder();

	// Build column list
	cols.append(columns.get(0));
	for (int i = 1; i < columns.size(); i++) {
		cols.append(",").append(columns.get(i));
	}

	// Build SELECT query
	command.append("SELECT ").append(cols.toString()).append(" FROM ")
			.append((keyspace == null) ? keyspaceName : keyspace)
			.append(".").append(tableName).append(" WHERE ").append(KEY)
			.append(" = '").append(key).append("';");

	SimpleStatement ss = new SimpleStatement(command.toString());
	ss.setConsistencyLevel(level);

	ResultSet result = executeQuery(ss);

	return result;
}
 

import com.datastax.driver.core.SimpleStatement; //导入依赖的package包/类
public ResultSet getRowByVersion(String keyspace, String tableName,
		int version, ConsistencyLevel level) {
	Long start = System.nanoTime();
	StringBuilder command = new StringBuilder();

	command.append("SELECT * FROM ")
			.append((keyspace == null) ? keyspaceName : keyspace)
			.append(".").append(tableName).append(" WHERE ")
			.append(VERSION).append(" = ").append(version).append(";");

	SimpleStatement ss = new SimpleStatement(command.toString());
	ss.setConsistencyLevel(level);

	ResultSet result = executeQuery(ss);
	IOStats.getRowByVersion(((double) System.nanoTime() - (double) start) / 1000000);

	return result;
}