/** {@inheritDoc} */ @Override public PreparedStatement setConsistencyLevel(ConsistencyLevel consistency) { return st.setConsistencyLevel(consistency); }
@Inject public CassandraExecutionDAO(Session session, ObjectMapper objectMapper, CassandraConfiguration config, Statements statements) { super(session, objectMapper, config); this.insertWorkflowStatement = session.prepare(statements.getInsertWorkflowStatement()).setConsistencyLevel(config.getWriteConsistencyLevel()); this.insertTaskStatement = session.prepare(statements.getInsertTaskStatement()).setConsistencyLevel(config.getWriteConsistencyLevel()); this.selectTotalStatement = session.prepare(statements.getSelectTotalStatement()).setConsistencyLevel(config.getReadConsistencyLevel()); this.selectTaskStatement = session.prepare(statements.getSelectTaskStatement()).setConsistencyLevel(config.getReadConsistencyLevel()); this.selectWorkflowStatement = session.prepare(statements.getSelectWorkflowStatement()).setConsistencyLevel(config.getReadConsistencyLevel()); this.selectWorkflowWithTasksStatement = session.prepare(statements.getSelectWorkflowWithTasksStatement()).setConsistencyLevel(config.getReadConsistencyLevel()); this.selectTaskLookupStatement = session.prepare(statements.getSelectTaskFromLookupTableStatement()).setConsistencyLevel(config.getReadConsistencyLevel()); this.updateWorkflowStatement = session.prepare(statements.getUpdateWorkflowStatement()).setConsistencyLevel(config.getWriteConsistencyLevel()); this.updateTotalTasksStatement = session.prepare(statements.getUpdateTotalTasksStatement()).setConsistencyLevel(config.getWriteConsistencyLevel()); this.updateTotalPartitionsStatement = session.prepare(statements.getUpdateTotalPartitionsStatement()).setConsistencyLevel(config.getWriteConsistencyLevel()); this.updateTaskLookupStatement = session.prepare(statements.getUpdateTaskLookupStatement()).setConsistencyLevel(config.getWriteConsistencyLevel()); this.deleteWorkflowStatement = session.prepare(statements.getDeleteWorkflowStatement()).setConsistencyLevel(config.getWriteConsistencyLevel()); this.deleteTaskStatement = session.prepare(statements.getDeleteTaskStatement()).setConsistencyLevel(config.getWriteConsistencyLevel()); this.deleteTaskLookupStatement = session.prepare(statements.getDeleteTaskLookupStatement()).setConsistencyLevel(config.getWriteConsistencyLevel()); }
stmt = session.prepare(QueryBuilder.delete().from(table) .where(QueryBuilder.eq(YCSB_KEY, QueryBuilder.bindMarker()))); stmt.setConsistencyLevel(writeConsistencyLevel); if (trace) { stmt.enableTracing();
stmt.setConsistencyLevel(readConsistencyLevel); if (trace) { stmt.enableTracing();
stmt.setConsistencyLevel(writeConsistencyLevel); if (trace) { stmt.enableTracing();
stmt.setConsistencyLevel(writeConsistencyLevel); if (trace) { stmt.enableTracing();
.where(QueryBuilder.eq(YCSB_KEY, QueryBuilder.bindMarker())) .limit(1)); stmt.setConsistencyLevel(readConsistencyLevel); if (trace) { stmt.enableTracing();
protected void init(int n, boolean batch, ConsistencyLevel cl) { write(n, batch, cl); prepared = session().prepare("SELECT * FROM " + tableName + " WHERE k = ?").setConsistencyLevel(cl); }
@Override public PreparedStatement apply(PreparedStatement prepared) { ProtocolVersion protocolVersion = getCluster().getConfiguration().getProtocolOptions().getProtocolVersion(); ByteBuffer routingKey = statement.getRoutingKey(protocolVersion, codecRegistry); if (routingKey != null) prepared.setRoutingKey(routingKey); if (statement.getConsistencyLevel() != null) prepared.setConsistencyLevel(statement.getConsistencyLevel()); if (statement.getSerialConsistencyLevel() != null) prepared.setSerialConsistencyLevel(statement.getSerialConsistencyLevel()); if (statement.isTracing()) prepared.enableTracing(); prepared.setRetryPolicy(statement.getRetryPolicy()); prepared.setOutgoingPayload(statement.getOutgoingPayload()); prepared.setIdempotent(statement.isIdempotent()); return prepared; } });
private void prepare(String cql, RetryPolicy petryPolicy) { logger.info("Preparing cql stmt {}", cql); PreparedStatement pstmt = session.prepare(cql); pstmt.setConsistencyLevel(consistencyLevel); pstmt.setRetryPolicy(petryPolicy); pstmt.setIdempotent(true); preparedStatements.put(cql, pstmt); }
private PreparedStatement initQuery(String cql) { return session.getSession().prepare(cql).setConsistencyLevel(ConsistencyLevel.LOCAL_QUORUM); }
public void init(RxSession session) { this.session = session; findConfigurationGroup = session.getSession() .prepare("SELECT name, value FROM sys_config WHERE config_id = ?") .setConsistencyLevel(ConsistencyLevel.LOCAL_QUORUM); findConfigurationValue = session.getSession() .prepare("SELECT value FROM sys_config WHERE config_id = ? AND name= ?") .setConsistencyLevel(ConsistencyLevel.LOCAL_QUORUM); updateConfigurationValue = session.getSession().prepare( "INSERT INTO sys_config (config_id, name, value) VALUES (?, ?, ?)") .setConsistencyLevel(ConsistencyLevel.LOCAL_QUORUM); deleteConfigurationValue = session.getSession().prepare( "DELETE FROM sys_config WHERE config_id =? and name = ?") .setConsistencyLevel(ConsistencyLevel.LOCAL_QUORUM); deleteConfiguration = session.getSession().prepare( "DELETE FROM sys_config WHERE config_id = ?") .setConsistencyLevel(ConsistencyLevel.LOCAL_QUORUM); }
/** * Create and configure a Prepared CQL statement */ protected PreparedStatement prepareStatement(String cql) { PreparedStatement preparedStatement = getSessionHolder().getSession().prepare(cql); if (consistencyLevel != null) { preparedStatement.setConsistencyLevel(consistencyLevel); } return preparedStatement; }
@Override public CqlPreparedStatement asPreparedStatement() { PreparedStatement pStmt = session.prepare(cqlQuery); pStmt.setConsistencyLevel(ConsistencyLevelTransform.getConsistencyLevel(cLevel)); return new DirectCqlPreparedStatement(session, pStmt); }
private void configureBehavior(String expected) { when(session.prepare(expected == null ? anyString() : expected)).thenReturn(prepared); when(prepared.setConsistencyLevel(any(ConsistencyLevel.class))).thenReturn(prepared); when(prepared.bind()).thenReturn(bound); when(bound.setString(anyInt(), anyString())).thenReturn(bound); }
/** * Prepare the given CQL Statement (or {@link com.datastax.driver.core.PreparedStatement}), applying statement * settings such as retry policy and consistency level. * * @param stmt the CQL Statement to prepare * @see #setRetryPolicy(RetryPolicy) * @see #setConsistencyLevel(ConsistencyLevel) */ protected void applyStatementSettings(PreparedStatement stmt) { ConsistencyLevel consistencyLevel = getConsistencyLevel(); if (consistencyLevel != null) { stmt.setConsistencyLevel(consistencyLevel); } RetryPolicy retryPolicy = getRetryPolicy(); if (retryPolicy != null) { stmt.setRetryPolicy(retryPolicy); } }
/** * Add common {@link QueryOptions} to Cassandra {@link PreparedStatement}s. * * @param preparedStatement the Cassandra {@link PreparedStatement}, must not be {@literal null}. * @param queryOptions query options (e.g. consistency level) to add to the Cassandra {@link PreparedStatement}. */ public static PreparedStatement addPreparedStatementOptions(PreparedStatement preparedStatement, QueryOptions queryOptions) { Assert.notNull(preparedStatement, "PreparedStatement must not be null"); if (queryOptions.getConsistencyLevel() != null) { preparedStatement.setConsistencyLevel(queryOptions.getConsistencyLevel()); } if (queryOptions.getRetryPolicy() != null) { preparedStatement.setRetryPolicy(queryOptions.getRetryPolicy()); } return preparedStatement; }
@Override public PreparedStatement apply(PreparedStatement prepared) { ByteBuffer routingKey = statement.getRoutingKey(); if (routingKey != null) prepared.setRoutingKey(routingKey); prepared.setConsistencyLevel(statement.getConsistencyLevel()); if (statement.isTracing()) prepared.enableTracing(); prepared.setRetryPolicy(statement.getRetryPolicy()); return prepared; } });
protected void init(int n, boolean batch, ConsistencyLevel cl) { write(n, batch, cl); prepared = session().prepare("SELECT * FROM " + tableName + " WHERE k = ?").setConsistencyLevel(cl); }
@Override public PreparedStatement apply(PreparedStatement prepared) { ProtocolVersion protocolVersion = getCluster().getConfiguration().getProtocolOptions().getProtocolVersion(); ByteBuffer routingKey = statement.getRoutingKey(protocolVersion, codecRegistry); if (routingKey != null) prepared.setRoutingKey(routingKey); if (statement.getConsistencyLevel() != null) prepared.setConsistencyLevel(statement.getConsistencyLevel()); if (statement.getSerialConsistencyLevel() != null) prepared.setSerialConsistencyLevel(statement.getSerialConsistencyLevel()); if (statement.isTracing()) prepared.enableTracing(); prepared.setRetryPolicy(statement.getRetryPolicy()); prepared.setOutgoingPayload(statement.getOutgoingPayload()); prepared.setIdempotent(statement.isIdempotent()); return prepared; } });