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