Mock注入的Dubbo bean方法是什么

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

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

代碼如下

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 {

        //只處理自己業(yè)務(wù)包下的類
        if (bean.getClass().getPackage() == null || !bean.getClass().getPackage().getName().contains("業(yè)務(wù)包名")) {
            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代理的實(shí)例
            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);
                }
            }
        }
    }

}

到此，關(guān)于“Mock注入的Dubbo bean方法是什么”的學(xué)習(xí)就結(jié)束了，希望能夠解決大家的疑惑。理論與實(shí)踐的搭配能更好的幫助大家學(xué)習(xí)，快去試試吧！若想繼續(xù)學(xué)習(xí)更多相關(guān)知識，請繼續(xù)關(guān)注億速云網(wǎng)站，小編會繼續(xù)努力為大家?guī)砀鄬?shí)用的文章！