优化代码结构

jackyzheng123 · Mar 5, 2018 · 27dd35c · 27dd35c
1 parent c750377
commit 27dd35c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/main/java/com/alibaba/dubbo/spring/boot/DubboConsumerAutoConfiguration.java b/src/main/java/com/alibaba/dubbo/spring/boot/DubboConsumerAutoConfiguration.java
@@ -68,9 +68,11 @@ public BeanPostProcessor beanPostProcessor() {
       @Override
       public Object postProcessBeforeInitialization(Object bean, String beanName)
           throws BeansException {
-        Class<?> objClz = bean.getClass();
+        Class<?> objClz;
         if (AopUtils.isAopProxy(bean)) {
           objClz = AopUtils.getTargetClass(bean);
+        } else {
+          objClz = bean.getClass();
         }
 
         try {