com.cybermkd.route.interceptor.InterceptorBuilder Maven / Gradle / Ivy

Go to download
package com.cybermkd.route.interceptor;

import com.cybermkd.route.core.Resource;
import com.cybermkd.route.interceptor.annotation.ClearInterceptors;
import com.cybermkd.route.interceptor.annotation.Interceptors;
import com.cybermkd.route.interceptor.exception.InterceptorException;

import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

/**
 * InterceptorBuilder
 */
public class InterceptorBuilder {
    private static final Interceptor[] NULL_INTERCEPTOR_ARRAY = new Interceptor[0];
    private Map, Interceptor> intersMap = new HashMap, Interceptor>();

    public void addToInterceptorsMap(Interceptor[] defaultInters) {
        for (Interceptor inter : defaultInters)
            intersMap.put((Class) inter.getClass(), inter);
    }

    /**
     * Build interceptors of Resource
     */
    public Interceptor[] buildResourceInterceptors(Class resourceClass) {
        Interceptors before = resourceClass.getAnnotation(Interceptors.class);
        return before != null ? createInterceptors(before) : NULL_INTERCEPTOR_ARRAY;
    }

    /**
     * Build interceptors of Method
     */
    public Interceptor[] buildMethodInterceptors(Method method) {
        Interceptors before = method.getAnnotation(Interceptors.class);
        return before != null ? createInterceptors(before) : NULL_INTERCEPTOR_ARRAY;
    }

    /**
     * Build interceptors of Action
     */
    public Interceptor[] buildRouteInterceptors(Interceptor[] defaultInters, Interceptor[] resourceInters, Class resourceClass, Interceptor[] methodInters, Method method) {

        List allInters = new ArrayList();

        for (Interceptor defaultInter : defaultInters) {
            allInters.add(defaultInter);
        }
        for (Interceptor resourceInter : resourceInters) {
            allInters.add(resourceInter);
        }
        for (Interceptor methodInter : methodInters) {
            allInters.add(methodInter);
        }
        //去除clean的aop

        Class[] resourceClears = getResourceClears(resourceClass);
        Class[] methodClears = getMethodClears(method);
        if ((resourceClears != null && resourceClears.length > 0) || (methodClears != null && methodClears.length > 0)) {
            for (int i = 0; i < allInters.size(); i++) {
                i = clearInterceptor(allInters, resourceClears, i);
                i = clearInterceptor(allInters, methodClears, i);
            }
        }

        return allInters.toArray(new Interceptor[allInters.size()]);
    }

    private int clearInterceptor(List allInters, Class[] clears, int i) {
        if (allInters.size() > 0 && clears != null && clears.length > 0) {
            for (Class ic : clears) {
                if (ic == allInters.get(i).getClass()) {
                    allInters.remove(i);
                    i--;
                }
            }
        }
        return i;
    }

    private Class[] getMethodClears(Method method) {
        ClearInterceptors clearInterceptor = method.getAnnotation(ClearInterceptors.class);
        return clearInterceptor != null ? clearInterceptor.value() : null;
    }

    private Class[] getResourceClears(Class resourceClass) {
        ClearInterceptors clearInterceptor = resourceClass.getAnnotation(ClearInterceptors.class);
        return clearInterceptor != null ? clearInterceptor.value() : null;
    }

    /**
     * Create interceptors with Annotation of Aspect. Singleton version.
     */
    private Interceptor[] createInterceptors(Interceptors aroundAnnotation) {
        Interceptor[] result = null;
        Class[] interceptorClasses = (Class[]) aroundAnnotation.value();
        if (interceptorClasses != null && interceptorClasses.length > 0) {
            result = new Interceptor[interceptorClasses.length];
            for (int i = 0; i < result.length; i++) {
                result[i] = intersMap.get(interceptorClasses[i]);
                if (result[i] != null)
                    continue;

                try {
                    result[i] = interceptorClasses[i].newInstance();
                    intersMap.put(interceptorClasses[i], result[i]);
                } catch (Exception e) {
                    throw new InterceptorException(e.getMessage(), e);
                }
            }
        }
        return result;
    }
}