Mock注入的Dubbo bean方法是什么

這篇文章主要介紹“Mock注入的Dubbo bean方法是什么”，在日常操作中，相信很多人在Mock注入的Dubbo bean方法是什么問題上存在疑惑，小編查閱了各式資料，整理出簡單好用的操作方法，希望對大家解答”Mock注入的Dubbo bean方法是什么”的疑惑有所幫助！接下來，請跟著小編一起來學習吧！

最近用Spring cloud Alibaba重構了后端服務，在用Mockit作單元測試的過程中發現，還是會調用dubbo的遠程接口。debug發現@Reference注解的接口和@MockBean的對象從ApplicationContext中取到了兩個實例，那這樣就自己再做一層，Spring啟動完就將其修改為Mock代理實例。

代碼如下

import org.apache.dubbo.config.annotation.Reference;
import org.springframework.beans.BeansException;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.config.BeanPostProcessor;
import org.springframework.context.ApplicationContext;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;

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

@Order
@Component
public class MockerFactory implements BeanPostProcessor {

    @Autowired
    private ApplicationContext applicationContext;

    @Override
    public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
        return bean;
    }

    @Override
    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {

        //只處理自己業務包下的類
        if (bean.getClass().getPackage() == null || !bean.getClass().getPackage().getName().contains("業務包名")) {
            return bean;
        }

        List<Field> fields = org.apache.commons.lang3.reflect.FieldUtils.getAllFieldsList(bean.getClass());
        for (Field field : fields) {

            Reference reference = field.getAnnotation(Reference.class);
            Map<String, ?> beans = applicationContext.getBeansOfType(field.getType());
            //將用到dubbo接口的地方替換為mock代理的實例
            if (reference != null && beans.size() > 1) {
                field.setAccessible(true);
                setMockFieldToBean(bean, field, beans);
            }
        }
        return bean;
    }

    private void setMockFieldToBean(Object bean, Field field, Map<String, ?> beans) {
        for (Object fieldBean : beans.values()){
            if(fieldBean.getClass().getName().contains("$MockitoMock$")){
                try {
                    field.set(bean, fieldBean);
                } catch (IllegalAccessException e) {
                    throw new RuntimeException(e);
                }
            }
        }
    }

}

到此，關于“Mock注入的Dubbo bean方法是什么”的學習就結束了，希望能夠解決大家的疑惑。理論與實踐的搭配能更好的幫助大家學習，快去試試吧！若想繼續學習更多相關知識，請繼續關注億速云網站，小編會繼續努力為大家帶來更多實用的文章！