Skip to content

Implemented the configuration of projection for local secondary index. #20

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Mar 13, 2020
Merged

Implemented the configuration of projection for local secondary index. #20

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 12 additions & 3 deletions ...g/socialsignin/spring/data/dynamodb/repository/util/Entity2DynamoDBTableSynchronizer.java
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -46,10 +46,10 @@
/**
* This is the base class for all classes performing the validation or
* auto-creation of tables based on the entity classes.
*
*
* //TODO: It would be nice if the checks would run in parallel via a
* TaskScheduler (if available)
*
*
* @see Entity2DDL
*/
public class Entity2DynamoDBTableSynchronizer<T, ID> extends EntityInformationProxyPostProcessor<T, ID>
Expand All @@ -60,6 +60,7 @@ public class Entity2DynamoDBTableSynchronizer<T, ID> extends EntityInformationPr

private static final String CONFIGURATION_KEY_entity2ddl_auto = "${spring.data.dynamodb.entity2ddl.auto:none}";
private static final String CONFIGURATION_KEY_entity2ddl_gsiProjectionType = "${spring.data.dynamodb.entity2ddl.gsiProjectionType:ALL}";
private static final String CONFIGURATION_KEY_entity2ddl_lsiProjectionType = "${spring.data.dynamodb.entity2ddl.lsiProjectionType:ALL}";
private static final String CONFIGURATION_KEY_entity2ddl_readCapacity = "${spring.data.dynamodb.entity2ddl.readCapacity:10}";
private static final String CONFIGURATION_KEY_entity2ddl_writeCapacity = "${spring.data.dynamodb.entity2ddl.writeCapacity:1}";

Expand All @@ -68,18 +69,20 @@ public class Entity2DynamoDBTableSynchronizer<T, ID> extends EntityInformationPr

private final Entity2DDL mode;
private final ProjectionType gsiProjectionType;
private final ProjectionType lsiProjectionType;
private final ProvisionedThroughput pt;

private final Collection<DynamoDBEntityInformation<T, ID>> registeredEntities = new ArrayList<>();

public Entity2DynamoDBTableSynchronizer(AmazonDynamoDB amazonDynamoDB, DynamoDBMapper mapper, Entity2DDL mode) {
this(amazonDynamoDB, mapper, mode.getConfigurationValue(), ProjectionType.ALL.name(), 10L, 10L);
this(amazonDynamoDB, mapper, mode.getConfigurationValue(), ProjectionType.ALL.name(), ProjectionType.ALL.name(), 10L, 10L);
}

@Autowired
public Entity2DynamoDBTableSynchronizer(AmazonDynamoDB amazonDynamoDB, DynamoDBMapper mapper,
@Value(CONFIGURATION_KEY_entity2ddl_auto) String mode,
@Value(CONFIGURATION_KEY_entity2ddl_gsiProjectionType) String gsiProjectionType,
@Value(CONFIGURATION_KEY_entity2ddl_lsiProjectionType) String lsiProjectionType,
@Value(CONFIGURATION_KEY_entity2ddl_readCapacity) long readCapacity,
@Value(CONFIGURATION_KEY_entity2ddl_writeCapacity) long writeCapacity) {
this.amazonDynamoDB = amazonDynamoDB;
Expand All @@ -88,6 +91,7 @@ public Entity2DynamoDBTableSynchronizer(AmazonDynamoDB amazonDynamoDB, DynamoDBM
this.mode = Entity2DDL.fromValue(mode);
this.pt = new ProvisionedThroughput(readCapacity, writeCapacity);
this.gsiProjectionType = ProjectionType.fromValue(gsiProjectionType);
this.lsiProjectionType = ProjectionType.fromValue(lsiProjectionType);
}

@Override
Expand Down Expand Up @@ -169,6 +173,11 @@ private boolean performCreate(DynamoDBEntityInformation<T, ID> entityInformation
});
}

if(ctr.getLocalSecondaryIndexes() != null) {
ctr.getLocalSecondaryIndexes().forEach(lsi ->
lsi.setProjection(new Projection().withProjectionType(lsiProjectionType)));
}

boolean result = TableUtils.createTableIfNotExists(amazonDynamoDB, ctr);
if (result) {
TableUtils.waitUntilActive(amazonDynamoDB, ctr.getTableName());
Expand Down