Skip to content

Commit d4e5d4a

Browse files
committed
Set tableSchema to chained extensions read context.
1 parent 2bd587b commit d4e5d4a

File tree

3 files changed

+8
-0
lines changed

3 files changed

+8
-0
lines changed
Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,6 @@
1+
{
2+
"category": "AWS SDK for Java v2",
3+
"contributor": "alvinsee",
4+
"type": "bugfix",
5+
"description": "sets tableSchema field when provisioning the extension context in ChainExtension#afterRead"
6+
}

services-custom/dynamodb-enhanced/src/main/java/software/amazon/awssdk/enhanced/dynamodb/internal/extensions/ChainExtension.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -152,6 +152,7 @@ public ReadModification afterRead(DynamoDbExtensionContext.AfterRead context) {
152152
DefaultDynamoDbExtensionContext.builder().items(itemToTransform)
153153
.operationContext(context.operationContext())
154154
.tableMetadata(context.tableMetadata())
155+
.tableSchema(context.tableSchema())
155156
.build();
156157

157158
ReadModification readModification = iterator.next().afterRead(afterRead);

services-custom/dynamodb-enhanced/src/test/java/software/amazon/awssdk/enhanced/dynamodb/extensions/ChainExtensionTest.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -276,6 +276,7 @@ private DefaultDynamoDbExtensionContext getExtensionContext(int i, OperationName
276276
DefaultDynamoDbExtensionContext.Builder context =
277277
DefaultDynamoDbExtensionContext.builder()
278278
.tableMetadata(FakeItem.getTableMetadata())
279+
.tableSchema(FakeItem.getTableSchema())
279280
.operationContext(PRIMARY_CONTEXT)
280281
.items(fakeItems.get(i));
281282
if (operationName != null) {

0 commit comments

Comments
 (0)