DynamodbTableNameResolver.java

public class DynamodbTableNameResolver extends DynamoDBMapperConfig.DefaultTableNameResolver {
    private String region;
    private String activeProfile;
    private Environment environment = null;

    public DynamodbTableNameResolver(String region, String activeProfile, Environment environment) {
        this.region = region;
        this.activeProfile = activeProfile;
        this.environment = environment;
    }

    private static final String DELIMITER = "--";

    @Override
    public String getTableName(Class<?> clazz, DynamoDBMapperConfig dynamoDBMapperConfig) {

        if (environment == null) {
            environment = AppContextAware.bean(Environment.class);
        }
        String tableNameToReturn;
        DynamoDBTable dynamoDBTable = clazz.getAnnotation(DynamoDBTable.class);
        if (dynamoDBTable == null) {
            throw new RuntimeException("Not a dynamodb table to resolve name");
        }
        tableNameToReturn = dynamoDBTable.tableName();
        DynamodbTableEnv dynamodbTableEnv = clazz.getAnnotation(DynamodbTableEnv.class);
        if(dynamoDBTable.tableName().isEmpty() && (dynamodbTableEnv==null || dynamodbTableEnv.value().isEmpty())){
            throw new RuntimeException("Table name in DynamoDBTable or DynamodbTableEnv is mandatory");
        }
        //If the @DynamoDBTable has value and no annotation @DynamodbTableEnv is present then the region and profile
        // is considered to form the full name. Below is the example
        //Ex : users--ap-south-1--prod, users--ca-central-1--dev
        if (dynamodbTableEnv == null) {
            return tableNameToReturn + DELIMITER + region + DELIMITER + activeProfile;
        } else {
            //if a value is present annotation @DynamodbTableEnv then it is treated as the key from the environment
            return environment.getProperty(dynamodbTableEnv.value());
        }
    }

}