ren.yale.java.MethodsProcessor Maven / Gradle / Ivy

Go to download
package ren.yale.java;

import org.apache.logging.log4j.LogManager;
import org.apache.logging.log4j.Logger;
import ren.yale.java.annotation.Blocking;
import ren.yale.java.aop.Before;
import ren.yale.java.interceptor.Interceptor;
import ren.yale.java.method.ArgInfo;
import ren.yale.java.method.ClassInfo;
import ren.yale.java.method.MethodInfo;

import javax.ws.rs.*;
import javax.ws.rs.core.Context;
import java.lang.annotation.Annotation;
import java.lang.reflect.Constructor;
import java.lang.reflect.Method;
import java.lang.reflect.Parameter;
import java.util.Arrays;
import java.util.List;

/**
 * Yale
 *
 * create at:  2018-01-31 17:25
 **/
class MethodsProcessor {
    private final static Logger LOGGER = LogManager.getLogger(MethodsProcessor.class.getName());

    private MethodsProcessor() {
    }

    private static Object newClass(Class clazz){

        try {
            for (Constructor c : clazz.getDeclaredConstructors()) {
                c.setAccessible(true);
                if (c.getParameterCount() == 0) {
                    return c.newInstance();
                }
            }
        }catch (Exception e){
            LOGGER.error(e.getMessage());
        }
        return null;

    }

    private static String getPathValue(Path path){
        if (path==null||path.value()==null){
            return "";
        }
        return path.value();

    }
    private static String getProducesValue(Produces produces){
        if (produces==null||produces.value()==null||produces.value().length ==0){
            return MethodInfo.PRODUCES_TYPE_ALL;
        }

        StringBuilder sb = new StringBuilder();
        for (String str:produces.value()) {
            if (sb.length()==0){
                sb.append(str);
            }else{
                sb.append(";");
                sb.append(str);
            }
        }

        return sb.toString();

    }

    private static Interceptor[] getIntercepter(Class[] inter){
        try {
            Interceptor[] interceptors = new Interceptor[inter.length];
            int i=0;
            for (Class cls:inter) {
                Interceptor interceptor= cls.newInstance();
                interceptors[i] = interceptor;
                i++;
            }
            return interceptors;
        } catch (Exception e) {
            e.printStackTrace();
        }
        return null;
    }
    private static Interceptor[] getBefores(Before before){
        if (before!=null&&before.value()!=null&&before.value().length>0){
            return getIntercepter(before.value());
        }
        return null;
    }
    public static void get(List classInfos, Class clazz) {


        Path path = (Path) clazz.getAnnotation(Path.class);

        if (path==null||path.value()==null){
            return;
        }

        ClassInfo classInfo = new ClassInfo();
        classInfo.setClassPath(path.value());
        classInfo.setClazzObj(newClass(clazz));
        classInfo.setClazz(clazz);

        Interceptor[] interceptorsClazzBefore =
                getBefores((Before) clazz.getAnnotation(Before.class));
        if (interceptorsClazzBefore!=null){
            classInfo.setBefores(interceptorsClazzBefore);
        }


        for (Method method : clazz.getMethods()) {
            Class mt = method.getDeclaringClass();
            if ( mt ==  Object.class){
                continue;
            }
            MethodInfo methodInfo = new MethodInfo();


            Interceptor[] interceptorsMethodBefore =
                    getBefores((Before) method.getAnnotation(Before.class));
            if (interceptorsMethodBefore!=null){
                methodInfo.setBefores(interceptorsMethodBefore);
            }


            Blocking blocking = method.getAnnotation(Blocking.class);
            if (blocking!=null){
                methodInfo.setBlocking(true);
            }


            Path pathMthod = (Path) method.getAnnotation(Path.class);
            Produces produces = (Produces) method.getAnnotation(Produces.class);

            methodInfo.setMethodPath(getPathValue(pathMthod));
            methodInfo.setProducesType(getProducesValue(produces));

            methodInfo.setHttpMethod(getHttpMethod(method));
            methodInfo.setMethod(method);

            Parameter[] parameters = method.getParameters();
            Class[] parameterTypes = method.getParameterTypes();
            Annotation[][] annotations = method.getParameterAnnotations();


            int i=0;
            for (Annotation[] an:annotations) {

                ArgInfo argInfo = new ArgInfo();

                argInfo.setAnnotation(an);
                argInfo.setClazz(parameterTypes[i]);
                argInfo.setParameter(parameters[i]);

                for (Annotation ant:an) {
                    if (ant instanceof Context){
                        argInfo.setContext(true);
                    }else
                    if (ant instanceof DefaultValue){
                        argInfo.setDefaultValue(((DefaultValue) ant).value());
                    }else if (ant instanceof PathParam){
                        argInfo.setPathParam(true);
                        argInfo.setPathParam(((PathParam) ant).value());
                    }else if (ant instanceof QueryParam){
                        argInfo.setQueryParam(true);
                        argInfo.setQueryParam(((QueryParam) ant).value());
                    }else if (ant instanceof FormParam){
                        argInfo.setFormParam(true);
                        argInfo.setFormParam(((FormParam) ant).value());
                    }

                }

                i++;
                methodInfo.addArgInfo(argInfo);
            }

            classInfo.addMethodInfo(methodInfo);

        }
        classInfos.add(classInfo);
    }
    private static boolean isRestClass(Class cls) {

        List> search = Arrays.asList(Path.class);

        for (Class item: search) {
            if (cls.getAnnotation(item) != null) {
                return true;
            }
        }

        return false;
    }

    private static Class getHttpMethod(Method method) {

        List> search = Arrays.asList(
                GET.class,
                POST.class,
                PUT.class,
                DELETE.class,
                OPTIONS.class,
                HEAD.class);

        for (Class item: search) {
            if (method.getAnnotation(item) != null) {
                return item;
            }
        }

        return null;
    }
}