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

原创文章,作者:ItWorker,如若转载,请注明出处:https://blog.ytso.com/220405.html

(0)
上一篇 2022年1月2日
下一篇 2022年1月2日

相关推荐

发表回复

登录后才能评论