com.tairanchina.csp.dew.jdbc.proxy.MethodConstruction Maven / Gradle / Ivy

Go to download
package com.tairanchina.csp.dew.jdbc.proxy;

import com.ecfront.dew.common.$;
import com.ecfront.dew.common.Page;
import com.tairanchina.csp.dew.jdbc.annotations.ModelParam;
import com.tairanchina.csp.dew.jdbc.annotations.Param;
import com.tairanchina.csp.dew.jdbc.entity.EntityContainer;

import java.lang.annotation.Annotation;
import java.lang.reflect.Method;
import java.lang.reflect.Parameter;
import java.util.HashMap;
import java.util.Map;

/**
 * Created by 迹_Jason on 2017/7/24.
 * 方法相关信息解析
 */
public class MethodConstruction {

    private static final String PAGE_NUMBER_FLAG = "pageNumber";
    private static final String PAGE_SIZE_FLAG = "pageSize";
    private static final int DEFAULT_PAGE_NUMBER = 1;
    private static final int DEFAULT_PAGE_SIZE = 10;

    private Method method;

    private Annotation[] methodAnnotations;

    private Annotation[][] paramAnnotations;

    private Map paramsMap;

    MethodConstruction(Method method, Object[] args) {
        setMethod(method);
        setMethodAnnotations(method.getAnnotations());
        setParamAnnotations(method.getParameterAnnotations());
        bindingParams(method.getParameters(), args);
    }

    private void bindingParams(Parameter[] paramNames, Object[] paramValues) {
        paramsMap = new HashMap<>();
        int i = 0;
        for (Object v : paramValues) {
            if (v == null) {
                i++;
                continue;
            }
            if (containAnnotation(paramAnnotations[i])) {
                transformToMap(v, paramsMap);
            } else {
                Annotation paramAnn = getParamOfName(paramNames[i].getAnnotations());
                if (paramAnn != null) {
                    paramsMap.put(((Param) paramAnn).value(), v);
                }
            }
            i++;
        }
    }

    private Annotation getParamOfName(Annotation[] paramAnnotations) {
        for (Annotation ann : paramAnnotations) {
            if (ann instanceof Param)
                return ann;
        }
        return null;
    }

    private boolean containAnnotation(Annotation[] paramAnnotations) {
        for (Annotation ann : paramAnnotations) {
            if (ann instanceof ModelParam)
                return true;
        }
        return false;
    }

    private void transformToMap(Object entities, Map paramsMap) {
        EntityContainer.EntityClassInfo entityClassInfo = EntityContainer.getEntityClassByClazz(entities.getClass());
        Map values = $.bean.findValues(entities, null, null, entityClassInfo.columns.keySet(), null);
        values.forEach((k, v) -> {
            if (v != null) {
                paramsMap.put(k, v);
            }
        });
    }


    Annotation[] getMethodAnnotations() {
        return methodAnnotations;
    }

    Class getReturnType() {
        return method.getReturnType();
    }

    boolean flagOfPaging() {
        return method.getReturnType().isAssignableFrom(Page.class);
    }

    public long getPageNumber() {
        return flagOfPaging() && paramsMap.get(PAGE_NUMBER_FLAG) != null ? (long) paramsMap.get(PAGE_NUMBER_FLAG) : DEFAULT_PAGE_NUMBER;
    }

    public int getPageSize() {
        return flagOfPaging() && paramsMap.get(PAGE_SIZE_FLAG) != null ? (int) paramsMap.get(PAGE_SIZE_FLAG) : DEFAULT_PAGE_SIZE;
    }

    private void setMethod(Method method) {
        this.method = method;
    }

    private void setMethodAnnotations(Annotation[] methodAnnotations) {
        this.methodAnnotations = methodAnnotations;
    }

    private void setParamAnnotations(Annotation[][] paramAnnotations) {
        this.paramAnnotations = paramAnnotations;
    }

    public Map getParamsMap() {
        return paramsMap;
    }
}