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

Example 46 with Update

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

the class StatementFactoryUnitTests method shouldCreateSetUpdateWithTimestamp.

// DATACASS-656
@Test
void shouldCreateSetUpdateWithTimestamp() {
    WriteOptions options = WriteOptions.builder().timestamp(1234).build();
    Query query = Query.query(Criteria.where("foo").is("bar")).queryOptions(options);
    StatementBuilder<com.datastax.oss.driver.api.querybuilder.update.Update> update = statementFactory.update(query, Update.empty().set("firstName", "baz"), personEntity);
    assertThat(update.build(ParameterHandling.INLINE).getQuery()).isEqualTo("UPDATE person USING TIMESTAMP 1234 SET first_name='baz' WHERE foo='bar'");
}

Example 47 with Update

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

the class AsyncCassandraTemplateUnitTests method updateShouldApplyUpdateQueryWitLwt.

// DATACASS-575
@Test
void updateShouldApplyUpdateQueryWitLwt() {
    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);
    verify(session).executeAsync(statementCaptor.capture());
    assertThat(render(statementCaptor.getValue())).isEqualTo("UPDATE users SET firstname='Walter' WHERE id='heisenberg' IF firstname='Walter' AND lastname='White'");
}

Example 48 with Update

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

the class AsyncCassandraTemplateUnitTests method updateShouldApplyUpdateQuery.

// DATACASS-575
@Test
void updateShouldApplyUpdateQuery() {
    Query query = Query.query(where("id").is("heisenberg"));
    Update update = Update.update("firstname", "Walter");
    template.update(query, update, User.class);
    verify(session).executeAsync(statementCaptor.capture());
    assertThat(render(statementCaptor.getValue())).isEqualTo("UPDATE users SET firstname='Walter' WHERE id='heisenberg'");
}

Example 49 with Update

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

the class StatementFactory method update.

/**
 * Create an {@literal UPDATE} statement by mapping {@link Query} to {@link Update}.
 *
 * @param query must not be {@literal null}.
 * @param update must not be {@literal null}.
 * @param persistentEntity must not be {@literal null}.
 * @param tableName must not be {@literal null}.
 * @return the update builder.
 * @since 2.1
 */
StatementBuilder<com.datastax.oss.driver.api.querybuilder.update.Update> update(Query query, Update update, CassandraPersistentEntity<?> persistentEntity, CqlIdentifier tableName) {
    Assert.notNull(query, "Query must not be null");
    Assert.notNull(update, "Update must not be null");
    Assert.notNull(persistentEntity, "CassandraPersistentEntity must not be null");
    Assert.notNull(tableName, "Table name must not be null");
    Filter filter = getQueryMapper().getMappedObject(query, persistentEntity);
    Update mappedUpdate = getUpdateMapper().getMappedObject(update, persistentEntity);
    StatementBuilder<com.datastax.oss.driver.api.querybuilder.update.Update> builder = update(tableName, mappedUpdate, filter);
    query.getQueryOptions().filter(UpdateOptions.class::isInstance).map(UpdateOptions.class::cast).map(UpdateOptions::getIfCondition).ifPresent(criteriaDefinitions -> applyUpdateIfCondition(builder, criteriaDefinitions));
    query.getQueryOptions().filter(WriteOptions.class::isInstance).map(WriteOptions.class::cast).ifPresent(writeOptions -> builder.apply(statement -> addWriteOptions(statement, writeOptions)));
    query.getQueryOptions().ifPresent(options -> builder.transform(statementBuilder -> QueryOptionsUtil.addQueryOptions(statementBuilder, options)));
    return builder;
}

Example 50 with Update

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

the class StatementFactory method update.

/**
 * Create an {@literal UPDATE} statement by mapping {@code objectToUpdate} to {@link Update} considering
 * {@link UpdateOptions}.
 *
 * @param objectToUpdate must not be {@literal null}.
 * @param options must not be {@literal null}.
 * @param entity must not be {@literal null}.
 * @param tableName must not be {@literal null}.
 * @return the update builder.
 */
StatementBuilder<com.datastax.oss.driver.api.querybuilder.update.Update> update(Object objectToUpdate, WriteOptions options, CassandraPersistentEntity<?> entity, CqlIdentifier tableName) {
    Assert.notNull(tableName, "TableName must not be null");
    Assert.notNull(objectToUpdate, "Object to builder must not be null");
    Assert.notNull(options, "WriteOptions must not be null");
    Assert.notNull(entity, "CassandraPersistentEntity must not be null");
    Where where = new Where();
    cassandraConverter.write(objectToUpdate, where, entity);
    Map<CqlIdentifier, Object> object = new LinkedHashMap<>();
    cassandraConverter.write(objectToUpdate, object, entity);
    where.forEach((cqlIdentifier, o) -> object.remove(cqlIdentifier));
    StatementBuilder<com.datastax.oss.driver.api.querybuilder.update.Update> builder = StatementBuilder.of(QueryBuilder.update(tableName).set().where()).bind((statement, factory) -> ((UpdateWithAssignments) statement).set(toAssignments(object, factory)).where(toRelations(where, factory))).apply(update -> addWriteOptions(update, options));
    Optional.of(options).filter(UpdateOptions.class::isInstance).map(UpdateOptions.class::cast).map(UpdateOptions::getIfCondition).ifPresent(criteriaDefinitions -> applyUpdateIfCondition(builder, criteriaDefinitions));
    builder.transform(statement -> QueryOptionsUtil.addQueryOptions(statement, options));
    return builder;
}