org.apache.cassandra.thrift.Compression Java Examples

/**
 * Send the batch insert.
 * 
 * @param batch
 *            the CQL batch insert statement
 * @param conn
 *            the connection to use
 * @param compressCQL
 *            true if the CQL should be compressed
 * @throws Exception
 *             if a problem occurs
 */
public static void commitCQLBatch(StringBuilder batch,
		CassandraConnection conn, boolean compressCQL) throws Exception {

	// compress the batch if necessary
	byte[] toSend = null;
	if (compressCQL) {
		toSend = compressQuery(batch.toString(), Compression.GZIP);
	} else {
		toSend = batch.toString().getBytes(
				Charset.forName(CassandraColumnMetaData.UTF8));
	}

	conn.getClient().execute_cql_query(ByteBuffer.wrap(toSend),
			compressCQL ? Compression.GZIP : Compression.NONE);
}
 

/**
 * Compress a CQL query
 * 
 * @param queryStr
 *            the CQL query
 * @param compression
 *            compression option (GZIP is the only option - so far)
 * @return an array of bytes containing the compressed query
 */
public static byte[] compressQuery(String queryStr, Compression compression) {
	byte[] data = queryStr.getBytes(Charset
			.forName(CassandraColumnMetaData.UTF8));

	Deflater compressor = new Deflater();
	compressor.setInput(data);
	compressor.finish();

	ByteArrayOutputStream byteArray = new ByteArrayOutputStream();
	byte[] buffer = new byte[1024];

	while (!compressor.finished()) {
		int size = compressor.deflate(buffer);
		byteArray.write(buffer, 0, size);
	}

	return byteArray.toByteArray();
}
 

/**
 * Compress a CQL query
 * 
 * @param queryStr the CQL query
 * @param compression compression option (GZIP is the only option - so far)
 * @return an array of bytes containing the compressed query
 */
public static byte[] compressQuery(String queryStr, Compression compression) {
  byte[] data = queryStr.getBytes(Charset
      .forName(CassandraColumnMetaData.UTF8));

  Deflater compressor = new Deflater();
  compressor.setInput(data);
  compressor.finish();

  ByteArrayOutputStream byteArray = new ByteArrayOutputStream();
  byte[] buffer = new byte[1024];

  while (!compressor.finished()) {
    int size = compressor.deflate(buffer);
    byteArray.write(buffer, 0, size);
  }

  return byteArray.toByteArray();
}
 

/** @deprecated Remove once Cassandra 1.1 support is no longer necessary. */
public void executeCql3Script_1_1(String script) {
    try {
        _client.set_cql_version(CQL_VERSION);
        for (String cqlStatement : toCqlStatements(script)) {
            if (StringUtils.isNotBlank(cqlStatement)) {
                cqlStatement += ";";
                _log.info("executing cql statement: " + cqlStatement);
                _client.execute_cql_query(ByteBuffer.wrap(cqlStatement.getBytes("UTF-8")), Compression.NONE);
            }
        }
    } catch (Exception e) {
        throw Throwables.propagate(e);
    }
}
 

public void executeCql3Script(String script) {
    try {
        for (String cqlStatement : toCqlStatements(script)) {
            if (StringUtils.isNotBlank(cqlStatement)) {
                cqlStatement += ";";
                _log.info("executing cql3 statement: " + cqlStatement);
                _client.execute_cql3_query(ByteBuffer.wrap(cqlStatement.getBytes("UTF-8")), Compression.NONE, ConsistencyLevel.LOCAL_QUORUM);
            }
        }
    } catch (Exception e) {
        throw Throwables.propagate(e);
    }
}
 

@Override
public <V> V execute(String query, ByteBuffer key, List<Object> queryParams, ResultHandler<V> handler) throws TException
{
    String formattedQuery = formatCqlQuery(query, queryParams, true);
    return handler.simpleNativeHandler().apply(
            client.execute_cql3_query(formattedQuery, key, Compression.NONE, settings.command.consistencyLevel)
    );
}
 

@Override
public <V> V execute(String query, ByteBuffer key, List<Object> queryParams, ResultHandler<V> handler) throws TException
{
    String formattedQuery = formatCqlQuery(query, queryParams, false);
    return handler.simpleNativeHandler().apply(
            client.execute_cql_query(formattedQuery, key, Compression.NONE)
    );
}
 

@Override
public Object createPreparedStatement(String cqlQuery) throws TException
{
    return client.prepare_cql3_query(cqlQuery, Compression.NONE);
}
 

@Override
public Object createPreparedStatement(String cqlQuery) throws TException
{
    return client.prepare_cql_query(cqlQuery, Compression.NONE);
}
 

/**
 * Constructs and executes a CQL TRUNCATE statement.
 * 
 * @param conn
 *            the connection to use
 * @param columnFamily
 *            the name of the column family to truncate.
 * @throws Exception
 *             if a problem occurs.
 */
public static void truncateColumnFamily(CassandraConnection conn,
		String columnFamily) throws Exception {
	String cqlCommand = "TRUNCATE " + columnFamily;

	conn.getClient().execute_cql_query(
			ByteBuffer.wrap(cqlCommand.getBytes()), Compression.NONE);
}