Examples with Query org.springframework.data.cassandra.core.query.Query used on opensource projects

Example 6 with Query

use of org.springframework.data.cassandra.core.query.Query in project spring-data-cassandra by spring-projects.

the class ReactiveCassandraTemplateUnitTests method updateShouldApplyUpdateQueryWitLwt.

// DATACASS-575
@Test
void updateShouldApplyUpdateQueryWitLwt() {
    when(reactiveResultSet.rows()).thenReturn(Flux.just(row));
    Filter ifCondition = Filter.from(where("firstname").is("Walter"), where("lastname").is("White"));
    Query query = Query.query(where("id").is("heisenberg")).queryOptions(UpdateOptions.builder().ifCondition(ifCondition).build());
    Update update = Update.update("firstname", "Walter");
    // 
    template.update(query, update, User.class).as(// 
    StepVerifier::create).expectNextCount(// 
    1).verifyComplete();
    verify(session).execute(statementCaptor.capture());
    assertThat(render(statementCaptor.getValue())).isEqualTo("UPDATE users SET firstname='Walter' WHERE id='heisenberg' IF firstname='Walter' AND lastname='White'");
}

Example 7 with Query

use of org.springframework.data.cassandra.core.query.Query in project spring-data-cassandra by spring-projects.

the class ReactiveCassandraTemplateIntegrationTests method deleteByQueryShouldRemoveEntityWithLwt.

// DATACASS-606
@Test
void deleteByQueryShouldRemoveEntityWithLwt() {
    DeleteOptions lwtOptions = DeleteOptions.builder().withIfExists().build();
    User user = new User("heisenberg", "Walter", "White");
    template.insert(user).as(StepVerifier::create).expectNextCount(1).verifyComplete();
    Query query = Query.query(where("id").is("heisenberg")).queryOptions(lwtOptions);
    // 
    template.delete(query, User.class).as(// 
    StepVerifier::create).expectNext(// 
    true).verifyComplete();
    // 
    template.delete(query, User.class).as(// 
    StepVerifier::create).expectNext(// 
    false).verifyComplete();
}

Example 8 with Query

use of org.springframework.data.cassandra.core.query.Query in project spring-data-cassandra by spring-projects.

the class StatementFactoryUnitTests method shouldMapDeleteQueryWithColumns.

// DATACASS-343
@Test
void shouldMapDeleteQueryWithColumns() {
    Query query = Query.empty().columns(Columns.from("age"));
    StatementBuilder<Delete> delete = statementFactory.delete(query, converter.getMappingContext().getRequiredPersistentEntity(Group.class));
    assertThat(delete.build(ParameterHandling.INLINE).getQuery()).isEqualTo("DELETE age FROM group");
}

Example 9 with Query

use of org.springframework.data.cassandra.core.query.Query in project spring-data-cassandra by spring-projects.

the class StatementFactoryUnitTests method shouldMapSelectQueryWithColumnsAndCriteria.

// DATACASS-343
@Test
void shouldMapSelectQueryWithColumnsAndCriteria() {
    Query query = Query.query(Criteria.where("foo").is("bar")).columns(Columns.from("age"));
    StatementBuilder<Select> select = statementFactory.select(query, groupEntity);
    assertThat(select.build(ParameterHandling.INLINE).getQuery()).isEqualTo("SELECT age FROM group WHERE foo='bar'");
}

Example 10 with Query

use of org.springframework.data.cassandra.core.query.Query in project spring-data-cassandra by spring-projects.

the class StatementFactoryUnitTests method updateShouldApplyQueryOptions.

// DATACASS-708
@Test
void updateShouldApplyQueryOptions() {
    UpdateOptions queryOptions = // 
    UpdateOptions.builder().executionProfile(// 
    "foo").serialConsistencyLevel(// 
    DefaultConsistencyLevel.QUORUM).build();
    Query query = Query.query(Criteria.where("foo").is("bar")).queryOptions(queryOptions);
    StatementBuilder<com.datastax.oss.driver.api.querybuilder.update.Update> update = statementFactory.update(query, Update.empty().set("firstName", "baz"), personEntity);
    SimpleStatement statement = update.build();
    assertThat(statement.getExecutionProfileName()).isEqualTo("foo");
    assertThat(statement.getSerialConsistencyLevel()).isEqualTo(DefaultConsistencyLevel.QUORUM);
}