Serialization optimization for DDB enhanced Client

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

@@ -17,8 +17,6 @@
 
 import static java.util.Collections.unmodifiableList;
 import static java.util.Collections.unmodifiableMap;
-import static software.amazon.awssdk.enhanced.dynamodb.internal.document.JsonStringFormatHelper.addEscapeCharacters;
-import static software.amazon.awssdk.enhanced.dynamodb.internal.document.JsonStringFormatHelper.stringValue;
 
 import java.util.ArrayList;
 import java.util.Arrays;
@@ -193,7 +191,7 @@ public String getJson(String attributeName) {
         if (attributeValue == null) {
             return null;
         }
-        return stringValue(JSON_ATTRIBUTE_CONVERTER.transformTo(attributeValue));
+        return StrategyJsonSerializer.serializeSingleAttributeValue(attributeValue);
     }
 
     @Override
@@ -230,12 +228,7 @@ public String toJson() {
         if (nonAttributeValueMap.isEmpty()) {
             return "{}";
         }
-        return attributeValueMap.getValue().entrySet().stream()
-                                .map(entry -> "\""
-                                              + addEscapeCharacters(entry.getKey())
-                                              + "\":"
-                                              + stringValue(JSON_ATTRIBUTE_CONVERTER.transformTo(entry.getValue())))
-                                .collect(Collectors.joining(",", "{", "}"));
+        return StrategyJsonSerializer.serializeAttributeValueMap(attributeValueMap.getValue());
     }
 
     @Override

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

@@ -0,0 +1,158 @@
+/*
+ * Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License").
+ * You may not use this file except in compliance with the License.
+ * A copy of the License is located at
+ *
+ *  http://aws.amazon.com/apache2.0
+ *
+ * or in the "license" file accompanying this file. This file is distributed
+ * on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either
+ * express or implied. See the License for the specific language governing
+ * permissions and limitations under the License.
+ */
+
+package software.amazon.awssdk.enhanced.dynamodb.internal.document;
+
+import java.nio.charset.StandardCharsets;
+import java.util.Map;
+import software.amazon.awssdk.annotations.SdkInternalApi;
+import software.amazon.awssdk.core.SdkBytes;
+import software.amazon.awssdk.protocols.json.SdkJsonGenerator;
+import software.amazon.awssdk.services.dynamodb.model.AttributeValue;
+import software.amazon.awssdk.thirdparty.jackson.core.JsonFactory;
+
+@SdkInternalApi
+public final class StrategyJsonSerializer {
+    private static final JsonFactory JSON_FACTORY = new JsonFactory();
+
+    private StrategyJsonSerializer() {
+    }
+
+    private enum JsonSerializationStrategy {
+        NULL,
+        STRING,
+        NUMBER,
+        BOOLEAN,
+        BYTES,
+        LIST,
+        MAP,
+        STRING_SET,
+        NUMBER_SET,
+        BYTES_SET
+    }
+
+    public static String serializeAttributeValueMap(Map<String, AttributeValue> map) {
+        SdkJsonGenerator jsonGen = new SdkJsonGenerator(JSON_FACTORY, "application/json");
+
+        jsonGen.writeStartObject();
+        map.forEach((key, value) -> {
+            jsonGen.writeFieldName(key);
+            serializeAttributeValue(jsonGen, value);
+        });
+        jsonGen.writeEndObject();
+
+        return new String(jsonGen.getBytes(), StandardCharsets.UTF_8);
+    }
+
+    public static String serializeSingleAttributeValue(AttributeValue av) {
+        SdkJsonGenerator jsonGen = new SdkJsonGenerator(JSON_FACTORY, "application/json");
+        serializeAttributeValue(jsonGen, av);
+        return new String(jsonGen.getBytes(), StandardCharsets.UTF_8);
+    }
+
+    public static void serializeAttributeValue(SdkJsonGenerator generator, AttributeValue av) {
+        JsonSerializationStrategy strategy = getStrategy(av);
+
+        switch (strategy) {
+            case NULL:
+                generator.writeNull();
+                break;
+            case STRING:
+                generator.writeValue(av.s());
+                break;
+            case NUMBER:
+                generator.writeNumber(av.n());
+                break;
+            case BOOLEAN:
+                generator.writeValue(av.bool());
+                break;
+            case BYTES:
+                generator.writeValue(av.b().asByteBuffer());
+                break;
+            case LIST:
+                generator.writeStartArray();
+                for (AttributeValue item : av.l()) {
+                    serializeAttributeValue(generator, item);
+                }
+                generator.writeEndArray();
+                break;
+            case MAP:
+                generator.writeStartObject();
+                for (Map.Entry<String, AttributeValue> entry : av.m().entrySet()) {
+                    generator.writeFieldName(entry.getKey());
+                    serializeAttributeValue(generator, entry.getValue());
+                }
+                generator.writeEndObject();
+                break;
+            case STRING_SET:
+                generator.writeStartArray();
+                for (String s : av.ss()) {
+                    generator.writeValue(s);
+                }
+                generator.writeEndArray();
+                break;
+            case NUMBER_SET:
+                generator.writeStartArray();
+                for (String n : av.ns()) {
+                    generator.writeNumber(n);
+                }
+                generator.writeEndArray();
+                break;
+            case BYTES_SET:
+                generator.writeStartArray();
+                for (SdkBytes b : av.bs()) {
+                    generator.writeValue(b.asByteBuffer());
+                }
+                generator.writeEndArray();
+                break;
+            default:
+                throw new IllegalStateException("Unsupported strategy: " + strategy);
+        }
+    }
+
+    private static JsonSerializationStrategy getStrategy(AttributeValue av) {
+        if (av.nul() != null && av.nul()) {
+            return JsonSerializationStrategy.NULL;
+        }
+        if (av.s() != null) {
+            return JsonSerializationStrategy.STRING;
+        }
+        if (av.n() != null) {
+            return JsonSerializationStrategy.NUMBER;
+        }
+        if (av.bool() != null) {
+            return JsonSerializationStrategy.BOOLEAN;
+        }
+        if (av.b() != null) {
+            return JsonSerializationStrategy.BYTES;
+        }
+        if (av.hasL()) {
+            return JsonSerializationStrategy.LIST;
+        }
+        if (av.hasM()) {
+            return JsonSerializationStrategy.MAP;
+        }
+        if (av.hasSs()) {
+            return JsonSerializationStrategy.STRING_SET;
+        }
+        if (av.hasNs()) {
+            return JsonSerializationStrategy.NUMBER_SET;
+        }
+        if (av.hasBs()) {
+            return JsonSerializationStrategy.BYTES_SET;
+        }
+        throw new IllegalStateException("Unknown AttributeValue type: " + av);
+    }
+}

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

@@ -31,6 +31,7 @@
 import org.junit.jupiter.params.ParameterizedTest;
 import org.junit.jupiter.params.provider.ArgumentsSource;
 import software.amazon.awssdk.enhanced.dynamodb.AttributeValueType;
+import software.amazon.awssdk.enhanced.dynamodb.DefaultAttributeConverterProvider;
 import software.amazon.awssdk.enhanced.dynamodb.EnhancedType;
 import software.amazon.awssdk.enhanced.dynamodb.TableMetadata;
 import software.amazon.awssdk.enhanced.dynamodb.converters.document.CustomAttributeForDocumentConverterProvider;

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

@@ -496,4 +496,162 @@ void error_when_usingClassGetPut_for_CollectionValues(){
                                                 () -> EnhancedDocument.builder().build().get("listKey" , List.class))
                                             .withMessage("Values of type List are not supported by this API, please use the getList API instead");
     }
+
+    @Test
+    void toJson_numberFormatting_veryLargeNumbers() {
+        EnhancedDocument doc = EnhancedDocument.builder()
+            .attributeConverterProviders(defaultProvider())
+            .putNumber("longMax", Long.MAX_VALUE)
+            .putNumber("longMin", Long.MIN_VALUE)
+            .putNumber("doubleMax", Double.MAX_VALUE)
+            .putNumber("scientific", 1.23e+100)
+            .putNumber("manyDecimals", 1.123456789012345)
+            .build();
+
+        String json = doc.toJson();
+        assertThat(json).isEqualTo("{\"longMax\":9223372036854775807,\"longMin\":-9223372036854775808,"
+                                  + "\"doubleMax\":1.7976931348623157E308,\"scientific\":1.23E100,"
+                                  + "\"manyDecimals\":1.123456789012345}");
+    }
+
+    @Test
+    void toJson_numberFormatting_trailingZeros() {
+        EnhancedDocument doc = EnhancedDocument.builder()
+            .attributeConverterProviders(defaultProvider())
+            .putNumber("twoPointZero", 2.0)
+            .putNumber("tenPointZeroZero", 10.00)
+            .build();
+
+        String json = doc.toJson();
+        assertThat(json).isEqualTo("{\"twoPointZero\":2.0,\"tenPointZeroZero\":10.0}");
+    }
+
+    @Test
+    void toJson_stringEscaping_allControlCharacters() {
+        EnhancedDocument doc = EnhancedDocument.builder()
+            .attributeConverterProviders(defaultProvider())
+            .putString("allEscapes", "line1\nline2\ttab\"quote\\backslash\r\f")
+            .build();
+
+        String json = doc.toJson();
+        assertThat(json).isEqualTo("{\"allEscapes\":\"line1\\nline2\\ttab\\\"quote\\\\backslash\\r\\f\"}");
+    }
+
+    @Test
+    void toJson_stringEscaping_forwardSlash() {
+        EnhancedDocument doc = EnhancedDocument.builder()
+            .attributeConverterProviders(defaultProvider())
+            .putString("slash", "path/to/resource")
+            .build();
+
+        String json = doc.toJson();
+        assertThat(json).isEqualTo("{\"slash\":\"path/to/resource\"}");
+    }
+
+    @Test
+    void toJson_emptyString() {
+        EnhancedDocument doc = EnhancedDocument.builder()
+            .attributeConverterProviders(defaultProvider())
+            .putString("empty", "")
+            .build();
+
+        String json = doc.toJson();
+        assertThat(json).isEqualTo("{\"empty\":\"\"}");
+    }
+
+    @Test
+    void toJson_bytesEncoding_emptyBytes() {
+        EnhancedDocument doc = EnhancedDocument.builder()
+            .attributeConverterProviders(defaultProvider())
+            .putBytes("empty", SdkBytes.fromByteArray(new byte[0]))
+            .build();
+
+        String json = doc.toJson();
+        assertThat(json).isEqualTo("{\"empty\":\"\"}");
+    }
+
+    @Test
+    void toJson_listWithAllNulls() {
+        EnhancedDocument doc = EnhancedDocument.builder()
+            .attributeConverterProviders(defaultProvider())
+            .putJson("nullList", "[null,null,null]")
+            .build();
+
+        String json = doc.toJson();
+        assertThat(json).isEqualTo("{\"nullList\":[null,null,null]}");
+    }
+
+    @Test
+    void toJson_mapWithNullValues() {
+        EnhancedDocument doc = EnhancedDocument.builder()
+            .attributeConverterProviders(defaultProvider())
+            .putJson("nullValues", "{\"key1\":null,\"key2\":\"value\",\"key3\":null}")
+            .build();
+
+        String json = doc.toJson();
+        assertThat(json).isEqualTo("{\"nullValues\":{\"key1\":null,\"key2\":\"value\",\"key3\":null}}");
+    }
+
+    @Test
+    void toJson_deeplyNestedStructure() {
+        String deepJson = "{\"level1\":{\"level2\":{\"level3\":{\"level4\":{\"level5\":{\"level6\":{\"level7\":{\"value\":\"deep\"}}}}}}}}";
+
+        EnhancedDocument doc = EnhancedDocument.builder()
+            .attributeConverterProviders(defaultProvider())
+            .putJson("nested", deepJson)
+            .build();
+
+        String json = doc.toJson();
+        assertThat(json).isEqualTo("{\"nested\":" + deepJson + "}");
+    }
+
+    @Test
+    void toJson_deeplyNestedArrays() {
+        String deepArrayJson = "[[[[[[\"innermost\"]]]]]]";
+
+        EnhancedDocument doc = EnhancedDocument.builder()
+            .attributeConverterProviders(defaultProvider())
+            .putJson("nestedArrays", deepArrayJson)
+            .build();
+
+        String json = doc.toJson();
+        assertThat(json).isEqualTo("{\"nestedArrays\":" + deepArrayJson + "}");
+    }
+
+    @Test
+    void toJson_emoji() {
+
+    String emoji = "{\"smile\":\"Hello 😀 World\"}";
+
+    EnhancedDocument doc = EnhancedDocument.builder()
+                                           .attributeConverterProviders(defaultProvider())
+                                           .putJson("emoji", emoji)
+                                           .build();
+
+    String json = doc.toJson();
+    assertThat(json).isEqualTo("{\"emoji\":" + emoji + "}");
+    }
+
+    @Test
+    void getJson_returnsRawValue() {
+        EnhancedDocument doc = EnhancedDocument.builder()
+            .attributeConverterProviders(defaultProvider())
+            .putString("key", "value")
+            .build();
+
+        String result = doc.getJson("key");
+        assertThat(result).isEqualTo("\"value\"");
+    }
+
+    @Test
+    void getJson_nonExistentAttribute() {
+        EnhancedDocument doc = EnhancedDocument.builder()
+            .attributeConverterProviders(defaultProvider())
+            .putString("exists", "value")
+            .build();
+
+        String result = doc.getJson("doesNotExist");
+        assertThat(result).isNull();
+    }
+
 }