/* * Copyright 2011-2023 Amazon.com, Inc. or its affiliates. All Rights Reserved. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at: * * http://aws.amazon.com/apache2.0 * * This file is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES * OR CONDITIONS OF ANY KIND, either express or implied. See the * License for the specific language governing permissions and * limitations under the License. */ package com.amazonaws.services.dynamodbv2.datamodeling; import com.amazonaws.metrics.RequestMetricCollector; import com.amazonaws.services.dynamodbv2.model.KeysAndAttributes; import com.amazonaws.services.dynamodbv2.model.WriteRequest; import java.util.List; import java.util.Map; import java.util.Random; /** * Immutable configuration object for service call behavior. An instance of this * configuration is supplied to every {@link DynamoDBMapper} at construction; if * not provided explicitly, {@link DynamoDBMapperConfig#DEFAULT} is used. New * instances can be given to the mapper object on individual save, load, and * delete operations to override the defaults. For example: * *
* DynamoDBMapper mapper = new DynamoDBMapper(dynamoDBClient); * // Force this read to be consistent * DomainClass obj = mapper.load(DomainClass.class, key, ConsistentReads.CONSISTENT.config()); * // Force this save operation to use putItem rather than updateItem * mapper.save(obj, SaveBehavior.CLOBBER.config()); * // Save the object into a different table * mapper.save(obj, new TableNameOverride("AnotherTable").config()); * // Delete the object even if the version field is out of date * mapper.delete(obj, SaveBehavior.CLOBBER.config()); **/ public class DynamoDBMapperConfig { /** * Default configuration; these defaults are also applied by the mapper * when only partial configurations are specified. * * @see SaveBehavior#UPDATE * @see ConsistentReads#EVENTUAL * @see PaginationLoadingStrategy#LAZY_LOADING * @see DefaultTableNameResolver#INSTANCE * @see DefaultBatchWriteRetryStrategy#INSTANCE * @see DefaultBatchLoadRetryStrategy#INSTANCE * @see DynamoDBTypeConverterFactory#standard * @see ConversionSchemas#DEFAULT */ public static final DynamoDBMapperConfig DEFAULT = builder() .withSaveBehavior(SaveBehavior.UPDATE) .withConsistentReads(ConsistentReads.EVENTUAL) .withPaginationLoadingStrategy(PaginationLoadingStrategy.LAZY_LOADING) .withTableNameResolver(DefaultTableNameResolver.INSTANCE) .withBatchWriteRetryStrategy(DefaultBatchWriteRetryStrategy.INSTANCE) .withBatchLoadRetryStrategy(DefaultBatchLoadRetryStrategy.INSTANCE) .withTypeConverterFactory(DynamoDBTypeConverterFactory.standard()) .withConversionSchema(ConversionSchemas.DEFAULT) .build(); /** * Creates a new empty builder. */ public static final Builder builder() { return new Builder(false); } /** * A fluent builder for DynamoDBMapperConfig objects. */ public static class Builder { private SaveBehavior saveBehavior; private ConsistentReads consistentReads; private TableNameOverride tableNameOverride; private TableNameResolver tableNameResolver; private ObjectTableNameResolver objectTableNameResolver; private PaginationLoadingStrategy paginationLoadingStrategy; private RequestMetricCollector requestMetricCollector; private ConversionSchema conversionSchema; private BatchWriteRetryStrategy batchWriteRetryStrategy; private BatchLoadRetryStrategy batchLoadRetryStrategy; private DynamoDBTypeConverterFactory typeConverterFactory; /** * Creates a new builder initialized with the {@link #DEFAULT} values. */ public Builder() { this(true); } /** * Creates a new builder, optionally initialized with the defaults. */ private Builder(final boolean defaults) { if (defaults == true) { saveBehavior = DEFAULT.getSaveBehavior(); consistentReads = DEFAULT.getConsistentReads(); paginationLoadingStrategy = DEFAULT.getPaginationLoadingStrategy(); conversionSchema = DEFAULT.getConversionSchema(); batchWriteRetryStrategy = DEFAULT.getBatchWriteRetryStrategy(); batchLoadRetryStrategy = DEFAULT.getBatchLoadRetryStrategy(); } } /** * Merges any non-null configuration values for the specified overrides. */ private final Builder merge(final DynamoDBMapperConfig o) { if (o == null) return this; if (o.saveBehavior != null) saveBehavior = o.saveBehavior; if (o.consistentReads != null) consistentReads = o.consistentReads; if (o.tableNameOverride != null) tableNameOverride = o.tableNameOverride; if (o.tableNameResolver != null) tableNameResolver = o.tableNameResolver; if (o.objectTableNameResolver != null) objectTableNameResolver = o.objectTableNameResolver; if (o.paginationLoadingStrategy != null) paginationLoadingStrategy = o.paginationLoadingStrategy; if (o.requestMetricCollector != null) requestMetricCollector = o.requestMetricCollector; if (o.conversionSchema != null) conversionSchema = o.conversionSchema; if (o.batchWriteRetryStrategy != null) batchWriteRetryStrategy = o.batchWriteRetryStrategy; if (o.batchLoadRetryStrategy != null) batchLoadRetryStrategy = o.batchLoadRetryStrategy; if (o.typeConverterFactory != null) typeConverterFactory = o.typeConverterFactory; return this; } /** * @return the currently-configured save behavior */ public SaveBehavior getSaveBehavior() { return saveBehavior; } /** * @param value the new save behavior */ public void setSaveBehavior(SaveBehavior value) { saveBehavior = value; } /** * @param value the new save behavior * @return this builder */ public Builder withSaveBehavior(SaveBehavior value) { setSaveBehavior(value); return this; } /** * Returns the consistent read behavior. Currently * this value is applied only in load and batch load operations of the * DynamoDBMapper. * @return the currently-configured consistent read behavior. */ public ConsistentReads getConsistentReads() { return consistentReads; } /** * Sets the consistent read behavior. Currently * this value is applied only in load and batch load operations of the * DynamoDBMapper. * @param value the new consistent read behavior. */ public void setConsistentReads(ConsistentReads value) { consistentReads = value; } /** * Sets the consistent read behavior. Currently * this value is applied only in load and batch load operations of the * DynamoDBMapper. * @param value the new consistent read behavior * @return this builder. * */ public Builder withConsistentReads(ConsistentReads value) { setConsistentReads(value); return this; } /** * @return the current table name override */ public TableNameOverride getTableNameOverride() { return tableNameOverride; } /** * @param value the new table name override */ public void setTableNameOverride(TableNameOverride value) { tableNameOverride = value; } /** * @param value the new table name override * @return this builder */ public Builder withTableNameOverride(TableNameOverride value) { setTableNameOverride(value); return this; } /** * @return the current table name resolver */ public TableNameResolver getTableNameResolver() { return tableNameResolver; } /** * @param value the new table name resolver */ public void setTableNameResolver(TableNameResolver value) { tableNameResolver = value; } /** * @param value the new table name resolver * @return this builder */ public Builder withTableNameResolver(TableNameResolver value) { setTableNameResolver(value); return this; } /** * @return the current object table name resolver */ public ObjectTableNameResolver getObjectTableNameResolver() { return objectTableNameResolver; } /** * @param value the new object table name resolver */ public void setObjectTableNameResolver(ObjectTableNameResolver value) { objectTableNameResolver = value; } /** * @param value the new object table name resolver * @return this builder */ public Builder withObjectTableNameResolver(ObjectTableNameResolver value) { setObjectTableNameResolver(value); return this; } /** * @return the currently-configured pagination loading strategy */ public PaginationLoadingStrategy getPaginationLoadingStrategy() { return paginationLoadingStrategy; } /** * @param value the new pagination loading strategy */ public void setPaginationLoadingStrategy( PaginationLoadingStrategy value) { paginationLoadingStrategy = value; } /** * @param value the new pagination loading strategy * @return this builder */ public Builder withPaginationLoadingStrategy( PaginationLoadingStrategy value) { setPaginationLoadingStrategy(value); return this; } /** * @return the currently-configured request metric collector */ public RequestMetricCollector getRequestMetricCollector() { return requestMetricCollector; } /** * @param value the new request metric collector */ public void setRequestMetricCollector(RequestMetricCollector value) { requestMetricCollector = value; } /** * @param value the new request metric collector * @return this builder */ public Builder withRequestMetricCollector(RequestMetricCollector value) { setRequestMetricCollector(value); return this; } /** * @return the current conversion schema */ public ConversionSchema getConversionSchema() { return conversionSchema; } /** * @param value the new conversion schema */ public void setConversionSchema(ConversionSchema value) { conversionSchema = value; } /** * @param value the new conversion schema * @return this builder */ public Builder withConversionSchema(ConversionSchema value) { setConversionSchema(value); return this; } /** * @return the current BatchWriteRetryStrategy */ public BatchWriteRetryStrategy getBatchWriteRetryStrategy() { return batchWriteRetryStrategy; } /** * @param value the new BatchWriteRetryStrategy */ public void setBatchWriteRetryStrategy( BatchWriteRetryStrategy value) { this.batchWriteRetryStrategy = value; } /** * @param value the new BatchWriteRetryStrategy * @return this builder */ public Builder withBatchWriteRetryStrategy( BatchWriteRetryStrategy value) { setBatchWriteRetryStrategy(value); return this; } public BatchLoadRetryStrategy getBatchLoadRetryStrategy() { return batchLoadRetryStrategy; } /** * @param value the new BatchLoadRetryStrategy */ public void setBatchLoadRetryStrategy( BatchLoadRetryStrategy value) { this.batchLoadRetryStrategy = value; } /** * @param value the new BatchLoadRetryStrategy * @return this builder */ public Builder withBatchLoadRetryStrategy( BatchLoadRetryStrategy value) { //set the no retry strategy if the user overrides the default with null if (value == null) { value = NoRetryBatchLoadRetryStrategy.INSTANCE; } setBatchLoadRetryStrategy(value); return this; } /** * @return the current type-converter factory */ public final DynamoDBTypeConverterFactory getTypeConverterFactory() { return typeConverterFactory; } /** * @param value the new type-converter factory */ public final void setTypeConverterFactory(DynamoDBTypeConverterFactory value) { this.typeConverterFactory = value; } /** * The type-converter factory for scalar conversions. *
To override standard type-conversions,
** DynamoDBMapperConfig config = DynamoDBMapperConfig.builder() * .withTypeConverterFactory(DynamoDBTypeConverterFactory.standard().override() * .with(String.class, MyObject.class, new StringToMyObjectConverter()) * .build()) * .build(); **
Then, on the property, specify the attribute binding,
** @DynamoDBTyped(DynamoDBAttributeType.S) * public MyObject getMyObject() ** @param value the new type-converter factory * @return this builder */ public final Builder withTypeConverterFactory(DynamoDBTypeConverterFactory value) { setTypeConverterFactory(value); return this; } /** * Builds a new {@code DynamoDBMapperConfig} object. * * @return the new, immutable config object */ public DynamoDBMapperConfig build() { return new DynamoDBMapperConfig(this); } } /** * Enumeration of behaviors for the save operation. */ public static enum SaveBehavior { /** * UPDATE will not affect unmodeled attributes on a save operation and a * null value for the modeled attribute will remove it from that item in * DynamoDB. *
* Because of the limitation of updateItem request, the implementation * of UPDATE will send a putItem request when a key-only object is being * saved, and it will send another updateItem request if the given * key(s) already exists in the table. *
* By default, the mapper uses UPDATE. */ UPDATE, /** * UPDATE_SKIP_NULL_ATTRIBUTES is similar to UPDATE, except that it * ignores any null value attribute(s) and will NOT remove them from * that item in DynamoDB. It also guarantees to send only one single * updateItem request, no matter the object is key-only or not. */ UPDATE_SKIP_NULL_ATTRIBUTES, /** * CLOBBER will clear and replace all attributes on save, including unmodeled * ones, and will also disregard versioned field constraints on conditional writes * as well as overwriting auto-generated values regardless of existing values. * If versioning is required, use {@link #PUT}. */ CLOBBER, /** * PUT will clear and replace all attributes on save, including unmodeled * ones, but fails if values do not match what is persisted on conditional writes * and does not overwrite auto-generated values. */ PUT, /** * APPEND_SET treats scalar attributes (String, Number, Binary) the same * as UPDATE_SKIP_NULL_ATTRIBUTES does. However, for set attributes, it * will append to the existing attribute value, instead of overriding * it. Caller needs to make sure that the modeled attribute type matches * the existing set type, otherwise it would result in a service * exception. */ APPEND_SET; public final DynamoDBMapperConfig config() { return builder().withSaveBehavior(this).build(); } }; /** * Enumeration of consistent read behavior. *
* CONSISTENT uses consistent reads, EVENTUAL does not. Consistent reads * have implications for performance and billing; see the service * documentation for details. *
* By default, the mapper uses eventual consistency. */ public static enum ConsistentReads { CONSISTENT, EVENTUAL; public final DynamoDBMapperConfig config() { return builder().withConsistentReads(this).build(); } }; /** * Enumeration of pagination loading strategy. */ public enum PaginationLoadingStrategy { /** * Paginated list is lazily loaded when possible, and all loaded results are kept in the memory. Data will * only be fetched when accessed so this can be more performant than {@link #EAGER_LOADING} when not all * data is used. Calls to methods such as {@link List#size()} will cause all results to be fetched from * the service. *
* By default, the mapper uses LAZY_LOADING. */ LAZY_LOADING, /** * Only supports using iterator to read from the paginated list. All other list operations will return * UnsupportedOperationException immediately. During the iteration, the list will clear all the * previous results before loading the next page, so that the list will keep at most one page of the loaded results in * memory. This also means the list could only be iterated once. *
* Use this configuration to reduce the memory overhead when handling
* large DynamoDB items. This is the most performant option when you only need to iterate the results
* of a query.
*/
ITERATION_ONLY,
/**
* Paginated list will eagerly load all the paginated results from DynamoDB as soon as the list is initialized. This may
* make several service calls when the list is created and is not recommended for large data sets. The benefit of using
* eager loading is that service call penalties are paid up front and you get predictable latencies when accessing the
* list afterwards since all contents are in memory.
*/
EAGER_LOADING;
public final DynamoDBMapperConfig config() {
return builder().withPaginationLoadingStrategy(this).build();
}
}
/**
* Allows overriding the table name declared on a domain class by the
* {@link DynamoDBTable} annotation.
*/
public static final class TableNameOverride {
private final String tableNameOverride;
private final String tableNamePrefix;
/**
* Returns a new {@link TableNameOverride} object that will prepend the
* given string to every table name.
*/
public static TableNameOverride withTableNamePrefix(
String tableNamePrefix) {
return new TableNameOverride(null, tableNamePrefix);
}
/**
* Returns a new {@link TableNameOverride} object that will replace
* every table name in requests with the given string.
*/
public static TableNameOverride withTableNameReplacement(
String tableNameReplacement) {
return new TableNameOverride(tableNameReplacement, null);
}
private TableNameOverride(String tableNameOverride, String tableNamePrefix) {
this.tableNameOverride = tableNameOverride;
this.tableNamePrefix = tableNamePrefix;
}
/**
* @see TableNameOverride#withTableNameReplacement(String)
*/
public TableNameOverride(String tableNameOverride) {
this(tableNameOverride, null);
}
/**
* Returns the table name to use for all requests. Exclusive with
* {@link TableNameOverride#getTableNamePrefix()}
*
* @see DynamoDBMapperConfig#getTableNameOverride()
*/
public String getTableName() {
return tableNameOverride;
}
/**
* Returns the table name prefix to prepend the table name for all
* requests. Exclusive with {@link TableNameOverride#getTableName()}
*
* @see DynamoDBMapperConfig#getTableNameOverride()
*/
public String getTableNamePrefix() {
return tableNamePrefix;
}
public final DynamoDBMapperConfig config() {
return builder().withTableNameOverride(this).build();
}
}
/**
* Interface for a strategy used to determine the table name of an object based on its class.
* This resolver is used when an object isn't available such as in
* {@link DynamoDBMapper#query(Class, DynamoDBQueryExpression)}
*
* @see ObjectTableNameResolver
* @author Raniz
*/
public static interface TableNameResolver {
/**
* Get the table name for a class. This method is used when an object is not available
* such as when creating requests for scan or query operations.
*
* @param clazz The class to get the table name for
* @param config The {@link DynamoDBMapperConfig}
* @return The table name to use for instances of clazz
*/
public String getTableName(Class> clazz, DynamoDBMapperConfig config);
}
/**
* Interface for a strategy used to determine the table name of an object based on its class.
* This resolver is used when an object is available such as in
* {@link DynamoDBMapper#batchSave(Object...)}
*
* If no table name resolver for objects is set, {@link DynamoDBMapper} reverts to using the
* {@link TableNameResolver} on each object's class.
*
* @see TableNameResolver
* @author Raniz
*/
public static interface ObjectTableNameResolver {
/**
* Get the table name for an object.
*
* @param object The object to get the table name for
* @param config The {@link DynamoDBMapperConfig}
* @return The table name to use for object
*/
public String getTableName(Object object, DynamoDBMapperConfig config);
}
/**
* Default implementation of {@link TableNameResolver} that mimics the behavior
* of DynamoDBMapper before the addition of {@link TableNameResolver}.
*
* @author Raniz
*/
public static class DefaultTableNameResolver implements TableNameResolver {
public static final DefaultTableNameResolver INSTANCE = new DefaultTableNameResolver();
@Override
public String getTableName(Class> clazz, DynamoDBMapperConfig config) {
final TableNameOverride override = config.getTableNameOverride();
if (override != null) {
final String tableName = override.getTableName();
if (tableName != null) {
return tableName;
}
}
final StandardBeanProperties.Beans> beans = StandardBeanProperties.of(clazz);
if (beans.properties().tableName() == null) {
throw new DynamoDBMappingException(clazz + " not annotated with @DynamoDBTable");
}
final String prefix = override == null ? null : override.getTableNamePrefix();
return prefix == null ? beans.properties().tableName() : prefix + beans.properties().tableName();
}
public final DynamoDBMapperConfig config() {
return builder().withTableNameResolver(this).build();
}
}
/**
* DynamoDBMapper#batchWrite takes arbitrary number of save/delete requests
* and breaks them into smaller chunks that can be accepted by the service
* API. Each chunk will be sent to DynamoDB via the BatchWriteItem API, and
* if it fails because the table's provisioned throughput is exceeded or an
* internal processing failure occurs, the failed requests are returned in
* the UnprocessedItems response parameter. This interface allows you to
* control the retry strategy when such scenario occurs.
*
* @see DynamoDBMapper#batchWrite(Iterable, Iterable, DynamoDBMapperConfig)
* @see
* Used internally to merge the {@link DynamoDBMapperConfig} provided at
* construction with an overriding object for a particular operation.
*
* @param defaults
* The default mapper configuration values.
* @param overrides
* The overridden mapper configuration values. Any non-null
* config settings will be applied to the returned object.
* @see DynamoDBMapperConfig#builder()
*/
@Deprecated
public DynamoDBMapperConfig(
DynamoDBMapperConfig defaults,
DynamoDBMapperConfig overrides) {
this(builder().merge(defaults).merge(overrides));
}
public BatchLoadRetryStrategy getBatchLoadRetryStrategy() {
return batchLoadRetryStrategy;
}
/**
* Returns the save behavior for this configuration.
*/
public SaveBehavior getSaveBehavior() {
return saveBehavior;
}
/**
* Returns the consistent read behavior for this configuration.
*/
public ConsistentReads getConsistentReads() {
return consistentReads;
}
/**
* Returns the table name override for this configuration. This value will
* override the table name specified in a {@link DynamoDBTable} annotation,
* either by replacing the table name entirely or else by pre-pending a
* string to each table name. This is useful for partitioning data in
* multiple tables at runtime.
*
* @see TableNameOverride#withTableNamePrefix(String)
* @see TableNameOverride#withTableNameReplacement(String)
*/
public TableNameOverride getTableNameOverride() {
return tableNameOverride;
}
/**
* Returns the table name resolver for this configuration. This value will
* be used to determine the table name for classes. It can be
* used for more powerful customization of table name than is possible using
* only {@link TableNameOverride}.
*
* @see TableNameResolver#getTableName(Class, DynamoDBMapperConfig)
*/
public TableNameResolver getTableNameResolver() {
return tableNameResolver;
}
/**
* Returns the object table name resolver for this configuration. This value will
* be used to determine the table name for objects. It can be
* used for more powerful customization of table name than is possible using
* only {@link TableNameOverride}.
*
* @see ObjectTableNameResolver#getTableName(Object, DynamoDBMapperConfig)
*/
public ObjectTableNameResolver getObjectTableNameResolver() {
return objectTableNameResolver;
}
/**
* Returns the pagination loading strategy for this configuration.
*/
public PaginationLoadingStrategy getPaginationLoadingStrategy() {
return paginationLoadingStrategy;
}
/**
* Returns the request metric collector or null if not specified.
*/
public RequestMetricCollector getRequestMetricCollector() {
return requestMetricCollector;
}
/**
* @return the conversion schema for this config object
*/
public ConversionSchema getConversionSchema() {
return conversionSchema;
}
/**
* @return the BatchWriteRetryStrategy for this config object
*/
public BatchWriteRetryStrategy getBatchWriteRetryStrategy() {
return batchWriteRetryStrategy;
}
/**
* @return the current type-converter factory
*/
public final DynamoDBTypeConverterFactory getTypeConverterFactory() {
return typeConverterFactory;
}
}