Examples with TransactUpdateItemEnhancedRequest software.amazon.awssdk.enhanced.dynamodb.model.TransactUpdateItemEnhancedRequest used on opensource projects

Example 1 with TransactUpdateItemEnhancedRequest

use of software.amazon.awssdk.enhanced.dynamodb.model.TransactUpdateItemEnhancedRequest in project aws-sdk-java-v2 by aws.

the class UpdateItemOperation method generateConditionExpressionIfExist.

/**
 * Retrieves the ConditionExpression from extensions if existing, and retrieves the ConditionExpression from the request
 * if existing. If both exist, they are merged.
 */
private Expression generateConditionExpressionIfExist(WriteModification transformation, Either<UpdateItemEnhancedRequest<T>, TransactUpdateItemEnhancedRequest<T>> request) {
    Expression conditionExpression = null;
    if (transformation != null && transformation.additionalConditionalExpression() != null) {
        conditionExpression = transformation.additionalConditionalExpression();
    }
    Expression operationConditionExpression = request.map(r -> Optional.ofNullable(r.conditionExpression()), r -> Optional.ofNullable(r.conditionExpression())).orElse(null);
    if (operationConditionExpression != null) {
        conditionExpression = operationConditionExpression.and(conditionExpression);
    }
    return conditionExpression;
}

Example 2 with TransactUpdateItemEnhancedRequest

use of software.amazon.awssdk.enhanced.dynamodb.model.TransactUpdateItemEnhancedRequest in project aws-sdk-java-v2 by aws.

the class TransactWriteItemsTest method mixedCommands_returnValuesOnConditionCheckFailureSet_allConditionsFail.

@Test
public void mixedCommands_returnValuesOnConditionCheckFailureSet_allConditionsFail() {
    mappedTable1.putItem(r -> r.item(RECORDS_1.get(0)));
    mappedTable1.putItem(r -> r.item(RECORDS_1.get(1)));
    mappedTable2.putItem(r -> r.item(RECORDS_2.get(0)));
    Expression conditionExpression = Expression.builder().expression("#attribute = :attribute").expressionValues(singletonMap(":attribute", stringValue("99"))).expressionNames(singletonMap("#attribute", "attribute")).build();
    Key key0 = Key.builder().partitionValue(0).build();
    Key key1 = Key.builder().partitionValue(1).build();
    ReturnValuesOnConditionCheckFailure returnValues = ReturnValuesOnConditionCheckFailure.ALL_OLD;
    TransactPutItemEnhancedRequest<Record2> putItemRequest = TransactPutItemEnhancedRequest.builder(Record2.class).conditionExpression(conditionExpression).item(RECORDS_2.get(0)).returnValuesOnConditionCheckFailure(returnValues).build();
    TransactUpdateItemEnhancedRequest<Record1> updateItemRequest = TransactUpdateItemEnhancedRequest.builder(Record1.class).conditionExpression(conditionExpression).item(RECORDS_1.get(0)).returnValuesOnConditionCheckFailure(returnValues).build();
    TransactDeleteItemEnhancedRequest deleteItemRequest = TransactDeleteItemEnhancedRequest.builder().key(key1).conditionExpression(conditionExpression).returnValuesOnConditionCheckFailure(returnValues).build();
    TransactWriteItemsEnhancedRequest transactWriteItemsEnhancedRequest = TransactWriteItemsEnhancedRequest.builder().addPutItem(mappedTable2, putItemRequest).addUpdateItem(mappedTable1, updateItemRequest).addConditionCheck(mappedTable1, ConditionCheck.builder().key(key0).conditionExpression(conditionExpression).returnValuesOnConditionCheckFailure(returnValues).build()).addDeleteItem(mappedTable1, deleteItemRequest).build();
    try {
        enhancedClient.transactWriteItems(transactWriteItemsEnhancedRequest);
        fail("Expected TransactionCanceledException to be thrown");
    } catch (TransactionCanceledException e) {
        List<CancellationReason> cancellationReasons = e.cancellationReasons();
        assertThat(cancellationReasons.size(), is(4));
        cancellationReasons.forEach(r -> assertThat(r.item().isEmpty(), is(false)));
    }
}

Example 3 with TransactUpdateItemEnhancedRequest

use of software.amazon.awssdk.enhanced.dynamodb.model.TransactUpdateItemEnhancedRequest in project aws-sdk-java-v2 by aws.

the class UpdateItemOperation method generateRequest.

@Override
public UpdateItemRequest generateRequest(TableSchema<T> tableSchema, OperationContext operationContext, DynamoDbEnhancedClientExtension extension) {
    if (!TableMetadata.primaryIndexName().equals(operationContext.indexName())) {
        throw new IllegalArgumentException("UpdateItem cannot be executed against a secondary index.");
    }
    T item = request.map(UpdateItemEnhancedRequest::item, TransactUpdateItemEnhancedRequest::item);
    Boolean ignoreNulls = request.map(r -> Optional.ofNullable(r.ignoreNulls()), r -> Optional.ofNullable(r.ignoreNulls())).orElse(null);
    Map<String, AttributeValue> itemMap = tableSchema.itemToMap(item, Boolean.TRUE.equals(ignoreNulls));
    TableMetadata tableMetadata = tableSchema.tableMetadata();
    WriteModification transformation = extension != null ? extension.beforeWrite(DefaultDynamoDbExtensionContext.builder().items(itemMap).operationContext(operationContext).tableMetadata(tableMetadata).tableSchema(tableSchema).operationName(operationName()).build()) : null;
    if (transformation != null && transformation.transformedItem() != null) {
        itemMap = transformation.transformedItem();
    }
    Collection<String> primaryKeys = tableSchema.tableMetadata().primaryKeys();
    Map<String, AttributeValue> keyAttributes = filterMap(itemMap, entry -> primaryKeys.contains(entry.getKey()));
    Map<String, AttributeValue> nonKeyAttributes = filterMap(itemMap, entry -> !primaryKeys.contains(entry.getKey()));
    Expression updateExpression = generateUpdateExpressionIfExist(tableMetadata, transformation, nonKeyAttributes);
    Expression conditionExpression = generateConditionExpressionIfExist(transformation, request);
    Map<String, String> expressionNames = coalesceExpressionNames(updateExpression, conditionExpression);
    Map<String, AttributeValue> expressionValues = coalesceExpressionValues(updateExpression, conditionExpression);
    UpdateItemRequest.Builder requestBuilder = UpdateItemRequest.builder().tableName(operationContext.tableName()).key(keyAttributes).returnValues(ReturnValue.ALL_NEW);
    if (request.left().isPresent()) {
        addPlainUpdateItemParameters(requestBuilder, request.left().get());
    }
    if (updateExpression != null) {
        requestBuilder.updateExpression(updateExpression.expression());
    }
    if (conditionExpression != null) {
        requestBuilder.conditionExpression(conditionExpression.expression());
    }
    if (CollectionUtils.isNotEmpty(expressionNames)) {
        requestBuilder = requestBuilder.expressionAttributeNames(expressionNames);
    }
    if (CollectionUtils.isNotEmpty(expressionValues)) {
        requestBuilder = requestBuilder.expressionAttributeValues(expressionValues);
    }
    return requestBuilder.build();
}