com.mars.mvc.proxy.ExecuteRef Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mars-mvc Show documentation
There is a newer version: 3.3.3
package com.mars.mvc.proxy;

import com.mars.core.annotation.MarsApi;
import com.mars.core.annotation.MarsReference;
import com.mars.core.annotation.enums.RefType;
import com.mars.core.load.LoadHelper;
import com.mars.core.model.MarsBeanModel;
import com.mars.core.util.StringUtil;

import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.util.Map;

/**
 * 执行api引用的资源
 */
public class ExecuteRef {

    /**
     * 所有的bean对象
     */
    private static Map beanModelMap = LoadHelper.getBeanObjectMap();

    /**
     * 指定服务层的方法
     * @param controlCls 被代理的类
     * @param method 方法
     * @param args 参数
     * @return 返回值
     * @throws Exception 异常
     */
    protected static Object executeRef(Class controlCls, Method method, Object[] args) throws Exception {
        /* 根据注解获取到对应的bean对象实体 */
        MarsReference marsReference = method.getAnnotation(MarsReference.class);
        MarsApi marsApi = controlCls.getAnnotation(MarsApi.class);

        /* 获取bean对象的class和实例 */
        MarsBeanModel marsBeanModel = getMarsBeanModel(marsReference,marsApi);
        Class cls = marsBeanModel.getCls();
        Object obj = marsBeanModel.getObj();

        /* 获取引用的资源名称 */
        String refName = getRefName(method,marsReference);

        if(marsReference == null || marsReference.refType().equals(RefType.METHOD)){
            /* 如果引用的是一个方法则执行bean里面对应的方法 */
            Object result = executeRefMethod(cls,obj,args,refName);
            if(result != null && result.equals("errorRef")){
                throw new Exception("没有找到名称为["+refName+"]的方法");
            }
            return result;
        } else {
            /* 否则就将bean里面对应的属性的值返回 */
            Field field = cls.getDeclaredField(refName);
            if(field == null){
                throw new Exception("没有找到名称为["+refName+"]的属性");
            }
            field.setAccessible(true);
            return field.get(obj);
        }
    }

    /**
     * 获取要执行的bean对象
     * @param marsReference 引用注解
     * @param marsApi api注解
     * @return bean对象实体类
     * @throws Exception 异常
     */
    private static MarsBeanModel getMarsBeanModel(MarsReference marsReference, MarsApi marsApi) throws Exception {
        String beanName = null;
        if(marsReference != null && !StringUtil.isNull(marsReference.beanName())){
            beanName = marsReference.beanName();
        } else if(marsApi != null && !StringUtil.isNull(marsApi.refBean())){
            beanName = marsApi.refBean();
        }

        if(StringUtil.isNull(beanName)){
            throw new Exception("请检查您的MarsReference注解或者MarsApi注解是否有指定要引用的bean名称");
        }

        MarsBeanModel marsBeanModel = beanModelMap.get(beanName);
        if(marsBeanModel == null){
            throw new Exception("没有找到name为["+beanName+"]的MarsBean");
        }
        return marsBeanModel;
    }

    /**
     * 执行方法
     * @param cls 类
     * @param obj 对象
     * @param args 参数
     * @param refName 引用的资源
     * @return 返回值
     * @throws Exception 异常
     */
    private static Object executeRefMethod(Class cls,Object obj,Object[] args, String refName) throws Exception {
        Method[] methods = cls.getDeclaredMethods();
        for(Method methodItem : methods){
            if(methodItem.getName().equals(refName)){
                if(args == null || args.length < 1){
                    return methodItem.invoke(obj);
                }
                return methodItem.invoke(obj,args);
            }
        }
        return "errorRef";
    }

    /**
     * 获取引用的资源名称
     * @param method api的方法
     * @param marsReference 引用注解配置
     * @return
     */
    private static String getRefName(Method method,MarsReference marsReference){
        if(marsReference == null || StringUtil.isNull(marsReference.refName())){
            return method.getName();
        }
        return marsReference.refName();
    }
}