DynamoDBIdIsHashAndRangeKeyEntityInformationImpl.java

/*
 * Copyright © 2018 spring-data-dynamodb (https://github.com/prasanna0586/spring-data-dynamodb)
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License 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 org.socialsignin.spring.data.dynamodb.repository.support;

import org.springframework.data.annotation.Id;
import org.springframework.lang.NonNull;
import org.springframework.lang.Nullable;
import software.amazon.awssdk.enhanced.dynamodb.AttributeConverter;

import java.util.Map;
import java.util.Optional;
import java.util.Set;

/**
 * Encapsulates minimal information needed to load DynamoDB entities that have both hash and range key, and have a
 * composite id attribute annotated with {@link Id}.
 *
 * Delegates to metadata and hashKeyExtractor components for all operations.
 * @param <T> the entity type
 * @param <ID> the ID type
 * @author Prasanna Kumar Ramachandran
 */
public class DynamoDBIdIsHashAndRangeKeyEntityInformationImpl<T, ID>
        extends FieldAndGetterReflectionEntityInformation<T, ID>
        implements DynamoDBIdIsHashAndRangeKeyEntityInformation<T, ID> {

    @NonNull
    private final DynamoDBHashAndRangeKeyExtractingEntityMetadata<T, ID> metadata;
    private final HashAndRangeKeyExtractor<ID, ?> hashAndRangeKeyExtractor;
    @Nullable
    private final String projection = null;
    @Nullable
    private final Integer limit = null;

    /**
     * Creates a new DynamoDBIdIsHashAndRangeKeyEntityInformationImpl.
     *
     * @param domainClass the entity class
     * @param metadata the hash and range key extracting entity metadata
     */
    public DynamoDBIdIsHashAndRangeKeyEntityInformationImpl(@NonNull Class<T> domainClass,
                                                            @NonNull DynamoDBHashAndRangeKeyExtractingEntityMetadata<T, ID> metadata) {
        super(domainClass, Id.class);
        this.metadata = metadata;
        this.hashAndRangeKeyExtractor = metadata.getHashAndRangeKeyExtractor(getIdType());
    }

    @NonNull
    @Override
    public Optional<String> getProjection() {
        return Optional.empty();
    }

    @NonNull
    @Override
    public Optional<Integer> getLimit() {
        return Optional.empty();
    }

    @Override
    public boolean isRangeKeyAware() {
        return true;
    }

    @Nullable
    @Override
    public Object getHashKey(final ID id) {
        return hashAndRangeKeyExtractor.getHashKey(id);
    }

    @Override
    public Object getRangeKey(final ID id) {
        return hashAndRangeKeyExtractor.getRangeKey(id);
    }

    @Override
    public Optional<String> getOverriddenAttributeName(String attributeName) {
        return metadata.getOverriddenAttributeName(attributeName);
    }

    @Override
    public boolean isHashKeyProperty(String propertyName) {
        return metadata.isHashKeyProperty(propertyName);
    }

    @Override
    public boolean isCompositeHashAndRangeKeyProperty(String propertyName) {
        return metadata.isCompositeHashAndRangeKeyProperty(propertyName);
    }

    @Override
    public String getRangeKeyPropertyName() {
        return metadata.getRangeKeyPropertyName();
    }

    @Override
    public AttributeConverter<?> getAttributeConverterForProperty(String propertyName) {
        return metadata.getAttributeConverterForProperty(propertyName);
    }

    @Override
    public Set<String> getIndexRangeKeyPropertyNames() {
        return metadata.getIndexRangeKeyPropertyNames();
    }

    @Override
    public String getHashKeyPropertyName() {
        return metadata.getHashKeyPropertyName();
    }

    @Override
    public <H> HashAndRangeKeyExtractor<ID, H> getHashAndRangeKeyExtractor(Class<ID> idClass) {
        return metadata.getHashAndRangeKeyExtractor(idClass);
    }

    @Override
    public String getDynamoDBTableName() {
        return metadata.getDynamoDBTableName();
    }

    @Override
    public Map<String, String[]> getGlobalSecondaryIndexNamesByPropertyName() {
        return metadata.getGlobalSecondaryIndexNamesByPropertyName();
    }

    @Override
    public <H> T getHashKeyPropotypeEntityForHashKey(H hashKey) {
        return metadata.getHashKeyPropotypeEntityForHashKey(hashKey);
    }

    @Override
    public boolean isGlobalIndexHashKeyProperty(String propertyName) {
        return metadata.isGlobalIndexHashKeyProperty(propertyName);
    }

    @Override
    public boolean isGlobalIndexRangeKeyProperty(String propertyName) {
        return metadata.isGlobalIndexRangeKeyProperty(propertyName);
    }

}