Skip to content

Commit 4871357

Browse files
joviegasjoviegas
committed
Remove extra spaces in Json and make it same as Items as in V1
1 parent 0c074c0 commit 4871357

File tree

3 files changed

+166
-79
lines changed

3 files changed

+166
-79
lines changed

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

Lines changed: 76 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -29,6 +29,7 @@
2929
import java.util.Optional;
3030
import java.util.Set;
3131
import java.util.stream.Collectors;
32+
import java.util.stream.StreamSupport;
3233
import software.amazon.awssdk.annotations.Immutable;
3334
import software.amazon.awssdk.annotations.SdkInternalApi;
3435
import software.amazon.awssdk.core.SdkBytes;
@@ -46,7 +47,6 @@
4647
import software.amazon.awssdk.protocols.jsoncore.JsonNodeParser;
4748
import software.amazon.awssdk.services.dynamodb.model.AttributeValue;
4849
import software.amazon.awssdk.utils.Lazy;
49-
import software.amazon.awssdk.utils.StringUtils;
5050
import software.amazon.awssdk.utils.Validate;
5151

5252
/**
@@ -181,7 +181,6 @@ public String getJson(String attributeName) {
181181
return null;
182182
}
183183
return JSON_ATTRIBUTE_CONVERTER.transformTo(attributeValue).toString();
184-
// TODO: Does toString return valid JSON? will remove this after comparing V1 side by side.
185184
}
186185

187186
@Override
@@ -215,22 +214,16 @@ public Map<String, AttributeValue> getUnknownTypeMap(String attributeName) {
215214

216215
@Override
217216
public String toJson() {
218-
StringBuilder output = new StringBuilder();
219-
output.append('{');
220-
boolean isFirst = true;
221-
for (Map.Entry<String, AttributeValue> entry : attributeValueMap.getValue().entrySet()) {
222-
if (!isFirst) {
223-
output.append(", ");
224-
} else {
225-
isFirst = false;
226-
}
227-
output.append('"')
228-
.append(StringUtils.replace(entry.getKey(), "\"", "\\"))
229-
.append("\": ")
230-
.append(JSON_ATTRIBUTE_CONVERTER.transformTo(entry.getValue()));
217+
if (nonAttributeValueMap.isEmpty()) {
218+
return "{}";
231219
}
232-
output.append('}');
233-
return output.toString();
220+
221+
return attributeValueMap.getValue().entrySet().stream()
222+
.map(entry -> "\""
223+
+ addEscapeCharacters(entry.getKey())
224+
+ "\":"
225+
+ stringValue(JSON_ATTRIBUTE_CONVERTER.transformTo(entry.getValue())))
226+
.collect(Collectors.joining(",", "{", "}"));
234227
}
235228

236229
@Override
@@ -286,6 +279,7 @@ public DefaultBuilder(DefaultEnhancedDocument enhancedDocument) {
286279

287280
public Builder putObject(String attributeName, Object value) {
288281
Validate.paramNotNull(attributeName, "attributeName");
282+
Validate.paramNotBlank(attributeName.trim(), "attributeName");
289283
enhancedTypeMap.remove(attributeName);
290284
nonAttributeValueMap.remove(attributeName);
291285
nonAttributeValueMap.put(attributeName, value);
@@ -435,6 +429,7 @@ private static AttributeValue getAttributeValueFromJson(String json) {
435429

436430
private static void checkInvalidAttribute(String attributeName, Object value) {
437431
Validate.paramNotNull(attributeName, "attributeName");
432+
Validate.paramNotBlank(attributeName.trim(), "attributeName");
438433
Validate.notNull(value, "%s must not be null. Use putNull API to insert a Null value", value);
439434
}
440435
}
@@ -461,4 +456,68 @@ public int hashCode() {
461456
return result;
462457
}
463458

459+
private static String stringValue(JsonNode jsonNode) {
460+
if (jsonNode.isArray()) {
461+
return StreamSupport.stream(jsonNode.asArray().spliterator(), false)
462+
.map(DefaultEnhancedDocument::stringValue)
463+
.collect(Collectors.joining(",", "[", "]"));
464+
}
465+
if (jsonNode.isObject()) {
466+
return mapToString(jsonNode);
467+
}
468+
469+
return jsonNode.isString() ? "\"" + addEscapeCharacters(jsonNode.text()) + "\"" : jsonNode.toString();
470+
}
471+
472+
private static String addEscapeCharacters(String input) {
473+
StringBuilder output = new StringBuilder();
474+
475+
for (int i = 0; i < input.length(); i++) {
476+
char ch = input.charAt(i);
477+
478+
switch (ch) {
479+
case '\\':
480+
output.append("\\\\"); // escape backslash with a backslash
481+
break;
482+
case '\n':
483+
output.append("\\n"); // newline character
484+
break;
485+
case '\r':
486+
output.append("\\r"); // carriage return character
487+
break;
488+
case '\t':
489+
output.append("\\t"); // tab character
490+
break;
491+
case '\f':
492+
output.append("\\f"); // form feed
493+
break;
494+
case '\"':
495+
output.append("\\\""); // double-quote character
496+
break;
497+
case '\'':
498+
output.append("\\'"); // single-quote character
499+
break;
500+
default:
501+
output.append(ch);
502+
break;
503+
}
504+
}
505+
506+
return output.toString();
507+
}
508+
509+
private static String mapToString(JsonNode jsonNode) {
510+
Map<String, JsonNode> value = jsonNode.asObject();
511+
512+
if (value.isEmpty()) {
513+
return "{}";
514+
}
515+
516+
StringBuilder output = new StringBuilder();
517+
output.append("{");
518+
value.forEach((k, v) -> output.append("\"").append(k).append("\":")
519+
.append(stringValue(v)).append(","));
520+
output.setCharAt(output.length() - 1, '}');
521+
return output.toString();
522+
}
464523
}

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

Lines changed: 45 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -16,6 +16,7 @@
1616
package software.amazon.awssdk.enhanced.dynamodb.document;
1717

1818
import static org.assertj.core.api.Assertions.assertThatExceptionOfType;
19+
import static org.assertj.core.api.Assertions.assertThatIllegalArgumentException;
1920
import static org.assertj.core.api.Assertions.assertThatIllegalStateException;
2021
import static org.assertj.core.api.Assertions.assertThatNullPointerException;
2122
import static software.amazon.awssdk.enhanced.dynamodb.AttributeConverterProvider.defaultProvider;
@@ -34,6 +35,10 @@
3435
import java.util.stream.Collectors;
3536
import java.util.stream.Stream;
3637
import org.junit.jupiter.api.Test;
38+
import org.junit.jupiter.params.ParameterizedTest;
39+
import org.junit.jupiter.params.provider.Arguments;
40+
import org.junit.jupiter.params.provider.MethodSource;
41+
import org.junit.jupiter.params.provider.ValueSource;
3742
import software.amazon.awssdk.core.SdkBytes;
3843
import software.amazon.awssdk.core.SdkNumber;
3944
import software.amazon.awssdk.enhanced.dynamodb.EnhancedType;
@@ -75,6 +80,7 @@ void enhancedDocumentGetters() {
7580
assertThat(document.getMap("simpleMap", EnhancedType.of(UUID.class), EnhancedType.of(BigDecimal.class)))
7681
.containsExactlyEntriesOf(expectedUuidBigDecimalMap);
7782
}
83+
7884
@Test
7985
void testNullArgsInStaticConstructor() {
8086
assertThatNullPointerException()
@@ -114,8 +120,7 @@ void builder_ResetsTheOldValues_beforeJsonSetterIsCalled() {
114120
.putString("simpleKeyNew", "simpleValueNew")
115121
.build();
116122

117-
assertThat(enhancedDocument.toJson()).isEqualTo("{\"stringKey\": \"stringValue\", \"simpleKeyNew\": "
118-
+ "\"simpleValueNew\"}");
123+
assertThat(enhancedDocument.toJson()).isEqualTo("{\"stringKey\":\"stringValue\",\"simpleKeyNew\":\"simpleValueNew\"}");
119124
assertThat(enhancedDocument.getString("simpleKeyOriginal")).isNull();
120125

121126
}
@@ -227,8 +232,8 @@ void error_When_DefaultProviderIsPlacedCustomProvider() {
227232
EnhancedType.of(CustomClassForDocumentAPI.class))
228233
.build();
229234

230-
assertThat(afterCustomClass.toJson()).isEqualTo("{\"direct_attr\": \"sample_value\", \"customObject\": "
231-
+ "{\"longNumber\": 26,\"string\": \"str_one\"}}");
235+
assertThat(afterCustomClass.toJson()).isEqualTo("{\"direct_attr\":\"sample_value\",\"customObject\":{\"longNumber\":26,"
236+
+ "\"string\":\"str_one\"}}");
232237

233238
EnhancedDocument enhancedDocument = EnhancedDocument.builder()
234239
.putString("direct_attr", "sample_value")
@@ -242,4 +247,40 @@ void error_When_DefaultProviderIsPlacedCustomProvider() {
242247
).withMessage("Converter not found for "
243248
+ "EnhancedType(software.amazon.awssdk.enhanced.dynamodb.converters.document.CustomClassForDocumentAPI)");
244249
}
250+
251+
private static Stream<Arguments> escapeDocumentStrings() {
252+
char c = 0x0a;
253+
return Stream.of(
254+
Arguments.of(String.valueOf(c),"{\"key\":\"\\n\"}")
255+
, Arguments.of("","{\"key\":\"\"}")
256+
, Arguments.of(" ","{\"key\":\" \"}")
257+
, Arguments.of("\t","{\"key\":\"\\t\"}")
258+
, Arguments.of("\n","{\"key\":\"\\n\"}")
259+
, Arguments.of("\r","{\"key\":\"\\r\"}")
260+
, Arguments.of("\f", "{\"key\":\"\\f\"}")
261+
);
262+
}
263+
264+
@ParameterizedTest
265+
@ValueSource(strings = {"", " " , "\t", " ", "\n", "\r", "\f"})
266+
void invalidKeyNames(String escapingString){
267+
assertThatIllegalArgumentException().isThrownBy(() ->
268+
EnhancedDocument.builder()
269+
.attributeConverterProviders(defaultProvider())
270+
.putString(escapingString, "sample")
271+
.build())
272+
.withMessageContaining("attributeName must not be blank or empty.");
273+
274+
}
275+
276+
@ParameterizedTest
277+
@MethodSource("escapeDocumentStrings")
278+
void escapingTheValues(String escapingString, String expectedJson) {
279+
280+
EnhancedDocument document = EnhancedDocument.builder()
281+
.attributeConverterProviders(defaultProvider())
282+
.putString("key", escapingString)
283+
.build();
284+
assertThat(document.toJson()).isEqualTo(expectedJson);
285+
}
245286
}

0 commit comments

Comments
 (0)