software.amazon.awssdk.services.dynamodb.model.ExpectedAttributeValue Java Examples

@Override
public Map<String, ExpectedAttributeValue> getDynamoLeaseOwnerExpectation(final Lease lease) {
    Map<String, ExpectedAttributeValue> result = new HashMap<>();

    ExpectedAttributeValue.Builder eavBuilder = ExpectedAttributeValue.builder();
    
    if (lease.leaseOwner() == null) {
        eavBuilder = eavBuilder.exists(false);
    } else {
        eavBuilder = eavBuilder.value(DynamoUtils.createAttributeValue(lease.leaseOwner()));
    }
    
    result.put(LEASE_OWNER_KEY, eavBuilder.build());

    return result;
}
 

/**
 * Create a new MetaStore with specified table name and extra data supplier.
 *
 * @param ddb Interface for accessing DynamoDB.
 * @param tableName DynamoDB table name for this {@link MetaStore}.
 * @param encryptor used to perform crypto operations on the record attributes
 * @param extraDataSupplier provides extra data that should be stored along with the material.
 */
public MetaStore(final DynamoDbClient ddb, final String tableName,
                 final DynamoDbEncryptor encryptor, final ExtraDataSupplier extraDataSupplier) {
    this.ddb = checkNotNull(ddb, "ddb must not be null");
    this.tableName = checkNotNull(tableName, "tableName must not be null");
    this.encryptor = checkNotNull(encryptor, "encryptor must not be null");
    this.extraDataSupplier = checkNotNull(extraDataSupplier, "extraDataSupplier must not be null");

    final Map<String, ExpectedAttributeValue> tmpExpected = new HashMap<>();
    tmpExpected.put(DEFAULT_HASH_KEY, ExpectedAttributeValue.builder().exists(false).build());
    tmpExpected.put(DEFAULT_RANGE_KEY, ExpectedAttributeValue.builder().exists(false).build());
    doesNotExist = Collections.unmodifiableMap(tmpExpected);

    this.encryptionConfiguration = DynamoDbEncryptionConfiguration.builder()
        .encryptionContext(EncryptionContext.builder()
                               .tableName(this.tableName)
                               .hashKeyName(DEFAULT_HASH_KEY)
                               .rangeKeyName(DEFAULT_RANGE_KEY)
                               .build())
        // All fields default to ENCRYPT_AND_SIGN with 'sign only' fields being explicitly overridden
        .defaultEncryptionAction(EncryptionAction.ENCRYPT_AND_SIGN)
        .addEncryptionActionOverrides(getSignedOnlyFields(extraDataSupplier).stream()
                                     .collect(Collectors.toMap(Function.identity(),
                                                               ignored -> EncryptionAction.SIGN_ONLY)))
        .build();
    ;
}
 

public Map<String, ExpectedAttributeValue> getDynamoLeaseCounterExpectation(final Long leaseCounter) {
    Map<String, ExpectedAttributeValue> result = new HashMap<>();

    ExpectedAttributeValue eav = ExpectedAttributeValue.builder().value(DynamoUtils.createAttributeValue(leaseCounter)).build();
    result.put(LEASE_COUNTER_KEY, eav);

    return result;
}
 

@Override
public Map<String, ExpectedAttributeValue> getDynamoNonexistantExpectation() {
    Map<String, ExpectedAttributeValue> result = new HashMap<>();

    ExpectedAttributeValue expectedAV = ExpectedAttributeValue.builder().exists(false).build();
    result.put(LEASE_KEY_KEY, expectedAV);

    return result;
}
 

@Override
public Map<String, ExpectedAttributeValue> getDynamoLeaseCounterExpectation(final Lease lease) {
    return getDynamoLeaseCounterExpectation(lease.leaseCounter());
}
 

/**
 * @param lease
 * @return the attribute value map asserting that a lease counter is what we expect.
 */
Map<String, ExpectedAttributeValue> getDynamoLeaseCounterExpectation(Lease lease);
 

/**
 * @param lease
 * @return the attribute value map asserting that the lease owner is what we expect.
 */
Map<String, ExpectedAttributeValue> getDynamoLeaseOwnerExpectation(Lease lease);
 

/**
 * @return the attribute value map asserting that a lease does not exist.
 */
Map<String, ExpectedAttributeValue> getDynamoNonexistantExpectation();