org.babyfish.jimmer.sql.DraftInterceptorManager Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jimmer-sql Show documentation
A revolutionary ORM framework for both java and kotlin
The newest version!
package org.babyfish.jimmer.sql;

import org.apache.commons.lang3.reflect.TypeUtils;
import org.babyfish.jimmer.Draft;
import org.babyfish.jimmer.impl.util.TypeCache;
import org.babyfish.jimmer.meta.ImmutableProp;
import org.babyfish.jimmer.meta.ImmutableType;
import org.babyfish.jimmer.meta.TypedProp;
import org.babyfish.jimmer.runtime.ImmutableSpi;
import org.jetbrains.annotations.NotNull;
import org.jetbrains.annotations.Nullable;

import java.lang.reflect.Type;
import java.lang.reflect.TypeVariable;
import java.util.*;

class DraftInterceptorManager {

    private final Map>> interceptorMap;

    private final TypeCache> cache =
            new TypeCache<>(this::create, true);

    DraftInterceptorManager(Collection> interceptors) {
        Map>> interceptorMap = new HashMap<>();
        for (DraftInterceptor interceptor : interceptors) {
            if (interceptor != null) {
                Map, Type> argTypeMap = TypeUtils
                        .getTypeArguments(
                                interceptor.getClass(),
                                DraftInterceptor.class
                        );
                if (argTypeMap.isEmpty()) {
                    throw new IllegalArgumentException(
                            "Illegal draft interceptor type \"" +
                                    interceptor.getClass().getName() +
                                    "\", it extends \"" +
                                   DraftInterceptor.class.getName() +
                                    "\" but the generic type arguments are not specified"
                    );
                }
                TypeVariable[] typeVariables = DraftInterceptor.class.getTypeParameters();
                Type entityType = argTypeMap.get(typeVariables[0]);
                Type draftType = argTypeMap.get(typeVariables[1]);
                if (!(entityType instanceof Class) || 
                        !((Class) entityType).isInterface() || (
                                ((Class) entityType).getAnnotation(Entity.class) == null &&
                                        ((Class) entityType).getAnnotation(MappedSuperclass.class) == null
                        )
                ) {
                    throw new IllegalArgumentException(
                            "Illegal draft interceptor type \"" +
                                    interceptor.getClass().getName() +
                                    "\", it extends \"" +
                                    DraftInterceptor.class.getName() +
                                    "\" but its entity type \"" +
                                    entityType +
                                    "\" is not interface type decorated by \"@" +
                                    Entity.class.getName() +
                                    "\" or \"@" +
                                    MappedSuperclass.class.getName() +
                                    "\""
                    );
                }
                if (!(draftType instanceof Class) || 
                        !((Class) draftType).isInterface() ||
                        !Draft.class.isAssignableFrom((Class) draftType) || 
                        !((Class)entityType).isAssignableFrom((Class) draftType)) {
                    throw new IllegalArgumentException(
                            "Illegal draft interceptor type \"" +
                                    interceptor.getClass().getName() +
                                    "\", it extends \"" +
                                    DraftInterceptor.class.getName() +
                                    "\" but its draft type \"" +
                                    ((Class)draftType).getName() +
                                    "\" is not interface type which is draft interface of the entity type \"" +
                                    ((Class)entityType).getName() +
                                    "\""
                    );
                }
                ImmutableType immutableType = ImmutableType.get((Class) draftType);
                interceptorMap
                        .computeIfAbsent(immutableType, it -> new ArrayList<>())
                        .add(interceptor);
            }
        }
        this.interceptorMap = interceptorMap;
    }

    public DraftInterceptor get(ImmutableType type) {
        return cache.get(type);
    }

    @SuppressWarnings("unchecked")
    private DraftInterceptor create(ImmutableType type) {
        List> interceptors = new ArrayList<>();
        Set allTypes = type.getAllTypes();
        for (ImmutableType t : allTypes) {
            List> list = interceptorMap.get(t);
            if (list != null) {
                interceptors.addAll(list);
            }
        }
        if (interceptors.isEmpty()) {
            return null;
        }
        Set> dependencies = new LinkedHashSet<>();
        for (DraftInterceptor interceptor : interceptors) {
            Collection> typeProps = interceptor.dependencies();
            if (typeProps != null) {
                for (TypedProp typedProp : typeProps) {
                    ImmutableProp prop = typedProp.unwrap();
                    if (!prop.isColumnDefinition()) {
                        throw new IllegalArgumentException(
                                "Illegal draft interceptor type \"" +
                                        interceptor.getClass().getName() +
                                        "\", its \"dependencies\" contains the property \"" +
                                        prop +
                                        "\" which is not column definition"
                        );
                    }
                    if (!allTypes.contains(prop.getDeclaringType())) {
                        throw new IllegalArgumentException(
                                "Illegal draft interceptor type \"" +
                                        interceptor.getClass().getName() +
                                        "\", its \"dependencies\" contains the property \"" +
                                        prop +
                                        "\" which is not belong to the type \"" +
                                        type +
                                        "\""
                        );
                    }
                    if (!prop.isId()) {
                        dependencies.add(typedProp);
                    }
                }
            }
        }

        return new DraftInterceptor() {

            @Override
            public void beforeSave(@NotNull Draft draft, @Nullable ImmutableSpi original) {
                for (DraftInterceptor interceptor : interceptors) {
                    ((DraftInterceptor)interceptor).beforeSave(draft, original);
                }
            }

            @SuppressWarnings("unchecked")
            @Override
            public Collection> dependencies() {
                return (Collection>) (Collection) dependencies;
            }
        };
    }
}