Skip to content

Commit 2731251

Browse files
joviegasjoviegas
committed
The builder for EnhancedDocument should not rely on the order in which attribute converters are added to it
1 parent a1f7443 commit 2731251

File tree

8 files changed

+671
-61
lines changed

8 files changed

+671
-61
lines changed

‎services-custom/dynamodb-enhanced/src/main/java/software/amazon/awssdk/enhanced/dynamodb/internal/document/DefaultEnhancedDocument.java

Lines changed: 33 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -71,10 +71,16 @@ private DefaultEnhancedDocument(Map<String, AttributeValue> attributeValueMap) {
7171
}
7272

7373
public DefaultEnhancedDocument(DefaultBuilder builder) {
74-
attributeValueMap = Collections.unmodifiableMap(builder.getAttributeValueMap());
75-
attributeConverterProviders = ChainConverterProvider.create(builder.attributeConverterProviders);
74+
List<AttributeConverterProvider> providers = builder.attributeConverterProviders;
75+
attributeConverterProviders =
76+
ChainConverterProvider.create(providers != null && !providers.isEmpty()
77+
? providers
78+
: Collections.singletonList(AttributeConverterProvider.defaultProvider()));
79+
attributeValueMap = Collections.unmodifiableMap(objectMapToAttributeMap(builder.attributeValueMap,
80+
attributeConverterProviders));
7681
}
7782

83+
7884
public static DefaultBuilder builder() {
7985
return new DefaultBuilder();
8086

@@ -339,31 +345,19 @@ public String toJsonPretty() {
339345

340346
public static class DefaultBuilder implements EnhancedDocument.Builder {
341347

342-
Map<String, AttributeValue> attributeValueMap = new LinkedHashMap<>();
348+
Map<String, Object> attributeValueMap = new LinkedHashMap<>();
343349

344350
List<AttributeConverterProvider> attributeConverterProviders = new ArrayList<>();
345351

346-
public DefaultBuilder() {
347-
}
348-
349-
public Map<String, AttributeValue> getAttributeValueMap() {
350-
return attributeValueMap;
352+
private DefaultBuilder() {
351353
}
352354

353355
@Override
354356
public Builder add(String attributeName, Object value) {
355-
ChainConverterProvider attributeConverterProvider = providerFromBuildAndAppendDefault();
356-
attributeValueMap.put(attributeName, convert(value, attributeConverterProvider));
357+
attributeValueMap.put(attributeName, value);
357358
return this;
358359
}
359360

360-
private ChainConverterProvider providerFromBuildAndAppendDefault() {
361-
List<AttributeConverterProvider> converterProviders = new ArrayList<>(attributeConverterProviders);
362-
converterProviders.add(DefaultAttributeConverterProvider.create());
363-
ChainConverterProvider attributeConverterProvider = ChainConverterProvider.create(converterProviders);
364-
return attributeConverterProvider;
365-
}
366-
367361
@Override
368362
public Builder addString(String attributeName, String value) {
369363
Validate.isTrue(!StringUtils.isEmpty(attributeName), "attributeName cannot empty or null");
@@ -440,7 +434,7 @@ public Builder addSdkBytesSet(String attributeName, Set<SdkBytes> values) {
440434
public Builder addList(String attributeName, List<?> value) {
441435
Validate.isTrue(!StringUtils.isEmpty(attributeName), "attributeName cannot empty or null");
442436
if (!isNullValueAdded(attributeName, value)) {
443-
attributeValueMap.put(attributeName, convert(value, providerFromBuildAndAppendDefault()));
437+
attributeValueMap.put(attributeName, value);
444438
}
445439
return this;
446440
}
@@ -449,7 +443,7 @@ public Builder addList(String attributeName, List<?> value) {
449443
public Builder addMap(String attributeName, Map<String, ?> value) {
450444
Validate.isTrue(!StringUtils.isEmpty(attributeName), "attributeName cannot empty or null");
451445
if (!isNullValueAdded(attributeName, value)) {
452-
attributeValueMap.put(attributeName, convert(value, providerFromBuildAndAppendDefault()));
446+
attributeValueMap.put(attributeName, value);
453447
}
454448
return this;
455449
}
@@ -507,7 +501,9 @@ public Builder json(String json) {
507501
throw new IllegalArgumentException("Could not parse argument json " + json);
508502
}
509503
AttributeValue attributeValue = JSON_ITEM_ATTRIBUTE_CONVERTER.transformFrom(jsonNode);
510-
this.attributeValueMap = attributeValue.m();
504+
if (attributeValue != null && attributeValue.hasM()) {
505+
attributeValueMap = new LinkedHashMap<>(attributeValue.m());
506+
}
511507
return this;
512508
}
513509

@@ -550,4 +546,21 @@ public int hashCode() {
550546
result = 31 * result + (attributeConverterProviders != null ? attributeConverterProviders.hashCode() : 0);
551547
return result;
552548
}
549+
550+
private static Map<String, AttributeValue> objectMapToAttributeMap(Map<String, Object> objectMap,
551+
AttributeConverterProvider attributeConverterProvider) {
552+
if (objectMap == null) {
553+
return null;
554+
}
555+
Map<String, AttributeValue> result = new LinkedHashMap<>(objectMap.size());
556+
objectMap.forEach((key, value) -> {
557+
if (value instanceof AttributeValue) {
558+
result.put(key, (AttributeValue) value);
559+
} else {
560+
result.put(key, convert(value, attributeConverterProvider));
561+
}
562+
});
563+
return result;
564+
}
565+
553566
}

‎services-custom/dynamodb-enhanced/src/main/java/software/amazon/awssdk/enhanced/dynamodb/internal/document/DocumentUtils.java

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -146,10 +146,12 @@ public static AttributeValue convert(Object sourceObject, AttributeConverterProv
146146
return convertMapToAttributeValue((Map<?, ?>) sourceObject, attributeConverterProvider);
147147
}
148148
AttributeConverter attributeConverter = attributeConverterProvider.converterFor(EnhancedType.of(sourceObject.getClass()));
149+
if (attributeConverter == null) {
150+
throw new IllegalStateException("Converter not found for Class " + sourceObject.getClass().getSimpleName());
151+
}
149152
return attributeConverter.transformFrom(sourceObject);
150153
}
151154

152-
153155
/**
154156
* Coverts AttributeValue to simple java objects like String, SdkNumber, Boolean, List, Set, SdkBytes or Maps.
155157
*/

‎services-custom/dynamodb-enhanced/src/test/java/software/amazon/awssdk/enhanced/dynamodb/converters/document/CustomAttributeForDocumentConverterProvider.java

Lines changed: 95 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,95 @@
1+
/*
2+
* Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
3+
*
4+
* Licensed under the Apache License, Version 2.0 (the "License").
5+
* You may not use this file except in compliance with the License.
6+
* A copy of the License is located at
7+
*
8+
* http://aws.amazon.com/apache2.0
9+
*
10+
* or in the "license" file accompanying this file. This file is distributed
11+
* on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
12+
* express or implied. See the License for the specific language governing
13+
* permissions and limitations under the License.
14+
*/
15+
16+
package software.amazon.awssdk.enhanced.dynamodb.converters.document;
17+
18+
import java.util.Map;
19+
import software.amazon.awssdk.enhanced.dynamodb.AttributeConverter;
20+
import software.amazon.awssdk.enhanced.dynamodb.AttributeConverterProvider;
21+
import software.amazon.awssdk.enhanced.dynamodb.AttributeValueType;
22+
import software.amazon.awssdk.enhanced.dynamodb.EnhancedType;
23+
import software.amazon.awssdk.enhanced.dynamodb.internal.converter.attribute.EnhancedAttributeValue;
24+
import software.amazon.awssdk.enhanced.dynamodb.internal.converter.string.IntegerStringConverter;
25+
import software.amazon.awssdk.services.dynamodb.model.AttributeValue;
26+
import software.amazon.awssdk.utils.ImmutableMap;
27+
28+
public class CustomAttributeForDocumentConverterProvider implements AttributeConverterProvider {
29+
private final Map<EnhancedType<?>, AttributeConverter<?>> converterCache = ImmutableMap.of(
30+
EnhancedType.of(CustomClassForDocumentAPI.class), new CustomClassForDocumentAttributeConverter(),
31+
EnhancedType.of(Integer.class), new CustomIntegerAttributeConverter()
32+
);
33+
34+
@SuppressWarnings("unchecked")
35+
@Override
36+
public <T> AttributeConverter<T> converterFor(EnhancedType<T> enhancedType) {
37+
return (AttributeConverter<T>) converterCache.get(enhancedType);
38+
}
39+
40+
public static CustomAttributeForDocumentConverterProvider create(){
41+
return new CustomAttributeForDocumentConverterProvider();
42+
}
43+
44+
45+
private static class CustomStringAttributeConverter implements AttributeConverter<String> {
46+
47+
final static String DEFAULT_SUFFIX = "-custom";
48+
49+
@Override
50+
public AttributeValue transformFrom(String input) {
51+
return EnhancedAttributeValue.fromString(input + DEFAULT_SUFFIX).toAttributeValue();
52+
}
53+
54+
@Override
55+
public String transformTo(AttributeValue input) {
56+
return input.s();
57+
}
58+
59+
@Override
60+
public EnhancedType<String> type() {
61+
return EnhancedType.of(String.class);
62+
}
63+
64+
@Override
65+
public AttributeValueType attributeValueType() {
66+
return AttributeValueType.S;
67+
}
68+
}
69+
70+
private static class CustomIntegerAttributeCo implements AttributeConverter<Integer> {
71+
72+
final static Integer DEFAULT_INCREMENT = 10;
73+
74+
@Override
75+
public AttributeValue transformFrom(Integer input) {
76+
return EnhancedAttributeValue.fromNumber(IntegerStringConverter.create().toString(input + DEFAULT_INCREMENT))
77+
.toAttributeValue();
78+
}
79+
80+
@Override
81+
public Integer transformTo(AttributeValue input) {
82+
return Integer.valueOf(input.n());
83+
}
84+
85+
@Override
86+
public EnhancedType<Integer> type() {
87+
return EnhancedType.of(Integer.class);
88+
}
89+
90+
@Override
91+
public AttributeValueType attributeValueType() {
92+
return AttributeValueType.N;
93+
}
94+
}
95+
}

0 commit comments

Comments
 (0)