【Java】利用反射執行Spring容器Bean指定的方法,支援多種引數自動呼叫

yangwqonly發表於2020-06-13

使用情景

  1. 將定時任務錄入資料庫(這樣做的好處是定時任務視覺化,也可以動態修改各個任務的執行時間),通過反射執行對應的方法;
  2. 配合Netty實現簡單的HTTP請求處理框架
  3. 其他需要使用反射執行Spring方法的業務亦可

目的

      很多文章都提到了反射,但是對於方法引數處理這一塊都是明確了型別,不支援按照實際引數動態轉換,而本篇文章提供了一個思路怎麼做到方法引數的動態呼叫。
      大家也可以通過利用本文的方法結合自己的業務場景寫出複用性更高、可擴充套件性更好的程式碼。歡迎各位指出文章中的錯誤,如果有更好的思路可以在下方評論,我們一起討論。
      歡迎轉發,請註明出處。

實現方式

前提:

明確清楚需要執行的類和方法。

思路

  1. 通過Spring容器獲取需要執行的類,注意:從spring容器中獲取的類可能是被JDK或CGLIB代理的(取決於你的環境配置);
  2. 獲取執行的Mehod物件;
  3. 封裝方法實際引數List,僅支援基本型別包裝類, String,物件,Map等引數型別自動轉換
  4. 執行Mehod的invoke方法

核心類

@Service
public class ReflectionService {

    @Resource
    private ApplicationContext applicationContext;

    private static final List<Class> WRAP_CLASS = Arrays.asList(Integer.class, Boolean.class, Double.class,Byte.class,Short.class, Long.class, Float.class, Double.class, BigDecimal.class, String.class);


    /**
     * 反射呼叫spring bean方法的入口
     * @param classz 類名
     * @param methodName 方法名
     * @param paramMap 實際引數
     * @throws Exception
     */
    public void invokeService(String classz, String methodName, Map<String,Object> paramMap) throws Exception {
        if(!applicationContext.containsBean(classz)) {
            throw new RuntimeException("Spring找不到對應的Bean");
        }

        // 從Spring中獲取代理物件(可能被JDK或者CGLIB代理)
        Object proxyObject = applicationContext.getBean(classz);

        // 獲取代理物件執行的方法
        Method method = getMethod(proxyObject.getClass(), methodName);

        // 獲取代理物件中的目標物件
        Class target = AopUtils.getTargetClass(proxyObject);

        // 獲取目標物件的方法,為什麼獲取目標物件的方法:只有目標物件才能通過 DefaultParameterNameDiscoverer 獲取引數的方法名,代理物件由於可能被JDK或CGLIB代理導致獲取不到引數名
        Method targetMethod = getMethod(target, methodName);

        if(method == null) {
            throw new RuntimeException(String.format("沒有找到%s方法", methodName));
        }

        // 獲取方法執行的引數
        List<Object> objects = getMethodParamList(targetMethod, paramMap);

        // 執行方法
        method.invoke(proxyObject, objects.toArray());
    }

    /**
     * 獲取方法實際引數,不支援基本型別
     * @param method
     * @param paramMap
     * @return
     */
    private List<Object> getMethodParamList(Method method, Map<String, Object> paramMap) throws Exception {
        List<Object> objectList = new ArrayList<>();

        // 利用Spring提供的類獲取方法形參名
        DefaultParameterNameDiscoverer nameDiscoverer = new DefaultParameterNameDiscoverer();
        String[] param =  nameDiscoverer.getParameterNames(method);

        for (int i = 0; i < method.getParameterTypes().length; i++) {
            Class<?> parameterType = method.getParameterTypes()[i];

            Object object = null;
            // 基本型別不支援,支援包裝類
            if(WRAP_CLASS.contains(parameterType)) {
                if(param != null && paramMap.containsKey(param[i])){
                    object = paramMap.get(param[i]);

                    object = ConvertUtils.convert(object, parameterType);
                }

            }else if (!parameterType.isPrimitive() ) {
                object = getInstance(parameterType);

                // 賦值
                BeanUtils.populate(object, paramMap);
            }

            objectList.add(object);
        }

        return objectList;
    }

    /**
     * 獲取型別例項
     * @param parameterType
     * @return
     * @throws Exception
     */
    private Object getInstance(Class<?> parameterType) throws Exception {
        if(parameterType.isAssignableFrom(List.class)) {
            return  new ArrayList();

        }else if(parameterType.isAssignableFrom(Map.class)) {
            return new HashMap();
        }else if(parameterType.isAssignableFrom(Set.class)) {
            return  new HashSet();
        }
        return parameterType.newInstance();
    }

    /**
     * 獲取目標方法
     * @param proxyObject
     * @param methodStr
     * @return
     */
    private Method getMethod(Class proxyObject, String methodStr) {
        Method[] methods = proxyObject.getMethods();

        for(Method method : methods) {
            if(method.getName().equalsIgnoreCase(methodStr)) {
                return method;
            }
        }

        return null;
    }
}

測試方法

package com.ywqonly.springtest.reflection;

import com.ywqonly.springtest.reflection.service.impl.ReflectionService;
import com.ywqonly.springtest.reflection.vo.CarVO;
import org.junit.Test;
import org.junit.runner.RunWith;
import org.springframework.boot.test.context.SpringBootTest;
import org.springframework.test.context.junit4.SpringRunner;

import javax.annotation.Resource;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;

@RunWith(SpringRunner.class)
@SpringBootTest
public class SpringReflectionTest {

    @Resource
    private ReflectionService reflectionService;

    @Test
    public void paramTest() throws Exception {
        Map<String, Object>  paramMap = new HashMap<>();

        paramMap.put("carName", "寶馬");
        paramMap.put("speed", "1");
        reflectionService.invokeService("carServiceImpl", "start", paramMap);
    }

    @Test
    public void objectTest() throws Exception {
        Map<String, Object>  paramMap = new HashMap<>();

        paramMap.put("carName", "寶馬");
        paramMap.put("speed", "2");
        reflectionService.invokeService("carServiceImpl", "startByVO", paramMap);
    }

    @Test
    public void mapTest() throws Exception {
        Map<String, Object>  paramMap = new HashMap<>();

        paramMap.put("carName", "寶馬");
        paramMap.put("speed", "3");
        reflectionService.invokeService("carServiceImpl", "startByMap", paramMap);
    }

}

原始碼分享

GITHUB原始碼地址

相關文章