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
converterFor(EnhancedType<T> enhancedType)36     public <T> AttributeConverter<T> converterFor(EnhancedType<T> enhancedType) {
37         return (AttributeConverter<T>) converterCache.get(enhancedType);
38     }
39 
create()40     public static CustomAttributeForDocumentConverterProvider create(){
41         return new CustomAttributeForDocumentConverterProvider();
42     }
43 }