com._4dconcept.springframework.data.marklogic.MarklogicCollectionUtils Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of spring-data-marklogic Show documentation
Marklogic support for Spring Data
There is a newer version: 2.1.3.RELEASE
Show newest version
package com._4dconcept.springframework.data.marklogic;

import com._4dconcept.springframework.data.marklogic.core.mapping.Collection;
import com._4dconcept.springframework.data.marklogic.core.mapping.MarklogicPersistentEntity;
import com._4dconcept.springframework.data.marklogic.core.mapping.MarklogicPersistentProperty;
import org.springframework.core.annotation.AnnotationUtils;
import org.springframework.data.mapping.PersistentPropertyAccessor;
import org.springframework.data.mapping.PropertyHandler;
import org.springframework.data.mapping.context.MappingContext;
import org.springframework.util.StringUtils;

import java.lang.reflect.AnnotatedElement;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.Collections;
import java.util.List;
import java.util.Objects;
import java.util.Optional;
import java.util.stream.Collectors;

public interface MarklogicCollectionUtils {

    default  List extractCollections(T entity, MappingContext, MarklogicPersistentProperty> mappingContext) {
        ArrayList collections = new ArrayList<>();

        MarklogicPersistentEntity persistentEntity = mappingContext.getPersistentEntity(entity.getClass());
        PersistentPropertyAccessor propertyAccessor = persistentEntity.getPropertyAccessor(entity);

        persistentEntity.doWithProperties((PropertyHandler) property -> {
            Optional collectionAnnotation = getCollectionAnnotation(property);
            if (collectionAnnotation.isPresent()) {
                Object value = propertyAccessor.getProperty(property);
                if (value != null) {
                    collections.addAll(doWithCollectionValue(value, collectionAnnotation.get()));
                }
            }
        });

        for (Method method : entity.getClass().getMethods()) {
            if (method.getParameterCount() > 0) {
                continue;
            }

            try {
                Optional collectionAnnotation = getCollectionAnnotation(method);
                if (collectionAnnotation.isPresent()) {
                    Object value = method.invoke(entity);
                    if (value != null) {
                        collections.addAll(doWithCollectionValue(value, collectionAnnotation.get()));
                    }
                }
            } catch (IllegalAccessException | InvocationTargetException e) {
                throw new UnsupportedOperationException(String.format("Unable to read value from %s", method), e);
            }

        }

        return collections.stream().distinct().collect(Collectors.toList());
    }

    default Optional getCollectionAnnotation(MarklogicPersistentProperty property) {
        ArrayList annotatedElements = new ArrayList<>();

        if (property.getPropertyDescriptor() != null && property.getPropertyDescriptor().getReadMethod() != null) {
            annotatedElements.add(property.getPropertyDescriptor().getReadMethod());
        }

        if (property.getField() != null) {
            annotatedElements.add(property.getField());
        }

        return annotatedElements.stream().map(e -> AnnotationUtils.findAnnotation(e, Collection.class)).filter(Objects::nonNull).findFirst();
    }

    default Optional getCollectionAnnotation(Method method) {
        return Optional.ofNullable(AnnotationUtils.findAnnotation(method, Collection.class));
    }

    default List doWithCollectionValue(Object value, Collection collection) {
        if (value instanceof java.util.Collection) {
            ArrayList collections = new ArrayList<>();
            java.util.Collection values = (java.util.Collection) value;
            for (Object o : values) {
                if (StringUtils.hasText(collection.prefix())) {
                    collections.add(String.format("%s:%s", collection.prefix(), o.toString()));
                } else {
                    collections.add(o.toString());
                }
            }
            return collections;
        } else {
            if (StringUtils.hasText(collection.prefix())) {
                return Collections.singletonList(String.format("%s:%s", collection.prefix(), value.toString()));
            } else {
                return Collections.singletonList(value.toString());
            }

        }
    }


}