dubbo实现简析(5)--dubbo 的注解用法实现

除了将spring扩展schemas定义在xml中，dubbo也支持用现在流行的注解方法直接定义在java文件中

public class AnnotationBean extends AbstractConfig implements DisposableBean, BeanFactoryPostProcessor, BeanPostProcessor, ApplicationContextAware

AnnotationBean 实现了    BeanFactoryPostProcessor， BeanPostProcessor， 

postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory)方法可以修改和追加BeanFactory的行为，所以它很适合扫描包内java文件的注解定义。代码如下：

public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory)

            throws BeansException {

        if (annotationPackage == null || annotationPackage.length() == 0) {

            return;

        }

        if (beanFactory instanceof BeanDefinitionRegistry) {

            try {

                // init scanner

                Class<?> scannerClass = ReflectUtils.forName("org.springframework.context.annotation.ClassPathBeanDefinitionScanner");

                Object scanner = scannerClass.getConstructor(new Class<?>[] {BeanDefinitionRegistry.class, boolean.class}).newInstance(new Object[] {(BeanDefinitionRegistry) beanFactory, true});

                // add filter

                Class<?> filterClass = ReflectUtils.forName("org.springframework.core.type.filter.AnnotationTypeFilter");

                Object filter = filterClass.getConstructor(Class.class).newInstance(Service.class);

                Method addIncludeFilter = scannerClass.getMethod("addIncludeFilter", ReflectUtils.forName("org.springframework.core.type.filter.TypeFilter"));

                addIncludeFilter.invoke(scanner, filter);

                // scan packages

                String[] packages = Constants.COMMA_SPLIT_PATTERN.split(annotationPackage);

                Method scan = scannerClass.getMethod("scan", new Class<?>[]{String[].class});

                scan.invoke(scanner, new Object[] {packages});

            } catch (Throwable e) {

                // spring 2.0

            }

        }

    }  

这里扫描的是 @Service  这个注解，同时限定了扫描的包。从实现上说，这个 AnnotationBean  应该定义在applicationContext.xml中 ：

   <bean class="com.alibaba.dubbo.config.spring. AnnotationBean ">

       <property name=" annotationPackage " value=" com.alibaba.dubbo.demo.provider.userservice,  com.alibaba.dubbo. demo.provider.productservice ">

    </bean>

具体的类应该这样：

@Service (interfaceClass="xxx.UserService",timeout="2000")

public class UserServiceImpl implements UserService {

   ......

}

而在客户consumer调用程序，应该是这样的：

public class UserAction {

    

   @Reference( interface="xxx.UserService" )

    private UserService userService;

    ......

}

AnnotationBean  实例初始化后, 也会把 annotationPackage下面的class也扫描进来，而且将这些Bean也注册在IOC容器中，接着 AnnotationBean  就可以处理这些class中的@Service , @Reference 了 。

下面这个方法应该出现在consumer程序中, 它解析@Reference注解,当然这方法可以跟接口类打在同一个包里。就是说其实应该有俩个AnnotationBean，一个定义在Provider，用于暴露服务，只扫描@Service；另一个要提供给客户，用于引用服务,只扫描@Reference。

postProcessBeforeInitialization(Object bean, String beanName)方法将在afterPropertiesSet方法前被调用:

public Object postProcessBeforeInitialization(Object bean, String beanName)

            throws BeansException {

        if (! isMatchPackage(bean)) {

            return bean;

        }

        Method[] methods = bean.getClass().getMethods();

        for (Method method : methods) {

            String name = method.getName();

            if (name.length() > 3 && name.startsWith("set")

                    && method.getParameterTypes().length == 1

                    && Modifier.isPublic(method.getModifiers())

                    && ! Modifier.isStatic(method.getModifiers())) {

                try {

                  Reference reference = method.getAnnotation(Reference.class);

                  if (reference != null) {

                 Object value = refer(reference, method.getParameterTypes()[0]);

                 if (value != null) {

                 method.invoke(bean, new Object[] {  });

                 }

                  }

                } catch (Throwable e) {

                    logger.error("Failed to init remote service reference at method " + name + " in class " + bean.getClass().getName() + ", cause: " + e.getMessage(), e);

                }

            }

        }

        Field[] fields = bean.getClass().getDeclaredFields();

        for (Field field : fields) {

            try {

                if (! field.isAccessible()) {

                    field.setAccessible(true);

                }

                Reference reference = field.getAnnotation(Reference.class);

              if (reference != null) {

                Object value = refer(reference, field.getType());

                if (value != null) {

                 field.set(bean, value);

                }

              }

            } catch (Throwable e) {

              logger.error("Failed to init remote service reference at filed " + field.getName() + " in class " + bean.getClass().getName() + ", cause: " + e.getMessage(), e);

            }

        }

        return bean;

    }

refer方法就是从远程拿到暴露服务的Invoker代理，这个和ReferenceBean前文描述的相同，就不细说了。

public Object postProcessAfterInitialization(Object bean, String beanName)

            throws BeansException {

        if (! isMatchPackage(bean)) {

            return bean;

        }

        Service service = bean.getClass().getAnnotation(Service.class);

        if (service != null) {

            ServiceBean<Object> serviceConfig = new ServiceBean<Object>(service);

            if (void.class.equals(service.interfaceClass())

                    && "".equals(service.interfaceName())) {

                if (bean.getClass().getInterfaces().length > 0) {

                    serviceConfig.setInterface(bean.getClass().getInterfaces()[0]);

                } else {

                    throw new IllegalStateException("Failed to export remote service class " + bean.getClass().getName() + ", cause: The @Service undefined interfaceClass or interfaceName, and the service class unimplemented any interfaces.");

                }

            }

            if (applicationContext != null) {

                serviceConfig.setApplicationContext(applicationContext);

                if (service.registry() != null && service.registry().length > 0) {

                    List<RegistryConfig> registryConfigs = new ArrayList<RegistryConfig>();

                    for (String registryId : service.registry()) {

                        if (registryId != null && registryId.length() > 0) {

                            registryConfigs.add((RegistryConfig)applicationContext.getBean(registryId, RegistryConfig.class));

                        }

                    }

                    serviceConfig.setRegistries(registryConfigs);

                }

                if (service.provider() != null && service.provider().length() > 0) {

                    serviceConfig.setProvider((ProviderConfig)applicationContext.getBean(service.provider(),ProviderConfig.class));

                }

                if (service.monitor() != null && service.monitor().length() > 0) {

                    serviceConfig.setMonitor((MonitorConfig)applicationContext.getBean(service.monitor(), MonitorConfig.class));

                }

                if (service.application() != null && service.application().length() > 0) {

                    serviceConfig.setApplication((ApplicationConfig)applicationContext.getBean(service.application(), ApplicationConfig.class));

                }

                if (service.module() != null && service.module().length() > 0) {

                    serviceConfig.setModule((ModuleConfig)applicationContext.getBean(service.module(), ModuleConfig.class));

                }

                if (service.provider() != null && service.provider().length() > 0) {

                    serviceConfig.setProvider((ProviderConfig)applicationContext.getBean(service.provider(), ProviderConfig.class));

                } else {

                    

                }

                if (service.protocol() != null && service.protocol().length > 0) {

                    List<ProtocolConfig> protocolConfigs = new ArrayList<ProtocolConfig>();

                    for (String protocolId : service.registry()) {

                        if (protocolId != null && protocolId.length() > 0) {

                            protocolConfigs.add((ProtocolConfig)applicationContext.getBean(protocolId, ProtocolConfig.class));

                        }

                    }

                    serviceConfig.setProtocols(protocolConfigs);

                }

                try {

                    serviceConfig.afterPropertiesSet();

                } catch (RuntimeException e) {

                    throw (RuntimeException) e;

                } catch (Exception e) {

                    throw new IllegalStateException(e.getMessage(), e);

                }

            }

            serviceConfig.setRef(bean);

            serviceConfigs.add(serviceConfig);

            serviceConfig.export();

        }

        return bean;

    }

可以看到这里也是调用export方法，与我们前文描述一致。

所以，使用注解和使用schema 扩展，只是形式不同，最终都是调用相同的Bean方法来暴露和引用服务。

可以说是 殊途同归。 有空可以再看看 API framework的代码， 它是使用的注解，不过目前提供的基本功能还不够多。

转载于:https://my.oschina.net/u/603187/blog/552070