Skip to content

Commit 1cefd27

Browse files
Merge branch 'rishav/ddbec-v2/add-examples' of https://github.com/aws/aws-database-encryption-sdk-dynamodb into rishav/ddbec-v2/add-examples
2 parents 5214abe + 3d8321e commit 1cefd27

1 file changed

Lines changed: 14 additions & 5 deletions

File tree

  • DynamoDbEncryption/runtimes/java/src/main/java/software/amazon/cryptography/dbencryptionsdk/dynamodb/itemencryptor/internaldafny/legacy

DynamoDbEncryption/runtimes/java/src/main/java/software/amazon/cryptography/dbencryptionsdk/dynamodb/itemencryptor/internaldafny/legacy/InternalLegacyOverride.java

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -293,15 +293,21 @@ > convertActionsV1ToV2(Map<String, Set<EncryptionFlags>> v1Actions) {
293293
private static software.amazon.cryptools.dynamodbencryptionclientsdk2.encryption.EncryptionContext convertEncryptionContextV1ToV2(
294294
EncryptionContext v1Context
295295
) {
296-
return software.amazon.cryptools.dynamodbencryptionclientsdk2.encryption.EncryptionContext
296+
final Map<String, software.amazon.awssdk.services.dynamodb.model.AttributeValue> v2AttributeValues = V1MapToV2Map(v1Context.getAttributeValues());
297+
final software.amazon.cryptools.dynamodbencryptionclientsdk2.encryption.EncryptionContext.Builder builder = software.amazon.cryptools.dynamodbencryptionclientsdk2.encryption.EncryptionContext
297298
.builder()
298299
.tableName(v1Context.getTableName())
299300
.hashKeyName(v1Context.getHashKeyName())
300301
.rangeKeyName(v1Context.getRangeKeyName())
301-
.attributeValues(V1MapToV2Map(v1Context.getAttributeValues()))
302-
.developerContext(v1Context.getDeveloperContext())
303-
.materialDescription(v1Context.getMaterialDescription())
304-
.build();
302+
.developerContext(v1Context.getDeveloperContext());
303+
304+
if (v1Context.getMaterialDescription() != null) {
305+
builder.materialDescription(v1Context.getMaterialDescription());
306+
}
307+
if (v2AttributeValues != null) {
308+
builder.attributeValues(v2AttributeValues);
309+
}
310+
return builder.build();
305311
}
306312

307313
public static String ToNativeString(DafnySequence<? extends Character> s) {
@@ -510,6 +516,9 @@ public static software.amazon.awssdk.services.dynamodb.model.AttributeValue V1At
510516
> V1MapToV2Map(
511517
Map<String, com.amazonaws.services.dynamodbv2.model.AttributeValue> input
512518
) {
519+
if (input == null) {
520+
return null;
521+
}
513522
return input
514523
.entrySet()
515524
.stream()

0 commit comments

Comments
 (0)