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方法是什么”的学习就结束了，希望能够解决大家的疑惑。理论与实践的搭配能更好的帮助大家学习，快去试试吧！若想继续学习更多相关知识，请继续关注亿速云网站，小编会继续努力为大家带来更多实用的文章！