42.2 Spring Boot多数据源

（5）编写多数据源注册文件；

       这个注入是最核心的部分，咱们先看代码：

com.kfit.config.datasource.multids.MultipleDataSourceBeanDefinitionRegistryPostProcessor：

package com.kfit.config.datasource.multids;

 

import java.util.HashMap;

import java.util.Map;

import java.util.Map.Entry;

 

import javax.sql.DataSource;

 

importorg.springframework.beans.BeansException;

importorg.springframework.beans.MutablePropertyValues;

importorg.springframework.beans.factory.annotation.AnnotatedGenericBeanDefinition;

importorg.springframework.beans.factory.config.BeanDefinition;

importorg.springframework.beans.factory.config.BeanDefinitionHolder;

import org.springframework.beans.factory.config.ConfigurableListableBeanFactory;

importorg.springframework.beans.factory.support.BeanDefinitionReaderUtils;

importorg.springframework.beans.factory.support.BeanDefinitionRegistry;

importorg.springframework.beans.factory.support.BeanDefinitionRegistryPostProcessor;

importorg.springframework.beans.factory.support.BeanNameGenerator;

importorg.springframework.boot.bind.RelaxedPropertyResolver;

importorg.springframework.context.EnvironmentAware;

importorg.springframework.context.annotation.AnnotationBeanNameGenerator;

importorg.springframework.context.annotation.AnnotationConfigUtils;

importorg.springframework.context.annotation.AnnotationScopeMetadataResolver;

importorg.springframework.context.annotation.Configuration;

import org.springframework.context.annotation.ScopeMetadata;

importorg.springframework.context.annotation.ScopeMetadataResolver;

importorg.springframework.core.env.Environment;

 

/**

 * 动态建立多数据源注册到Spring中

 *

 接口：BeanDefinitionRegistryPostProcessor只要是注入bean,

 在上一节介绍过使用方式；

 

 接口：接口 EnvironmentAware 重写方法 setEnvironment

 能够在工程启动时，获取到系统环境变量和application配置文件中的变量。 

这个第24节介绍过.

 

 方法的执行顺序是：

 

 setEnvironment()-->postProcessBeanDefinitionRegistry()-->postProcessBeanFactory()

 

 

 *

 * @author Angel(QQ:412887952)

 * @version v.0.1

 */

@Configuration

publicclassMultipleDataSourceBeanDefinitionRegistryPostProcessor implementsBeanDefinitionRegistryPostProcessor,EnvironmentAware{

 

      

       //做用域对象.

       privateScopeMetadataResolver scopeMetadataResolver = new AnnotationScopeMetadataResolver();

       //bean名称生成器.

       privateBeanNameGenerator beanNameGenerator = new AnnotationBeanNameGenerator();

             

       //如配置文件中未指定数据源类型，使用该默认值

   privatestaticfinal Object DATASOURCE_TYPE_DEFAULT = "org.apache.tomcat.jdbc.pool.DataSource";

       //  private static final ObjectDATASOURCE_TYPE_DEFAULT = "com.zaxxer.hikari.HikariDataSource";

             

       // 存放DataSource配置的集合;

   private Map<String, Map<String, Object>> dataSourceMap = newHashMap<String, Map<String, Object>>();

      

      

       @Override

       publicvoidpostProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {

              System.out.println("MultipleDataSourceBeanDefinitionRegistryPostProcessor.postProcessBeanFactory()");

              //设置为主数据源;

              beanFactory.getBeanDefinition("dataSource").setPrimary(true);

             

              if(!dataSourceMap.isEmpty()){

                     //不为空的时候.

                     BeanDefinitionbd = null;

               Map<String, Object> dsMap = null;

               MutablePropertyValues mpv = null;

               for (Entry<String, Map<String,Object>> entry : dataSourceMap.entrySet()) {

                       bd = beanFactory.getBeanDefinition(entry.getKey());

                       mpv = bd.getPropertyValues();

                       dsMap = entry.getValue();

                       mpv.addPropertyValue("driverClassName", dsMap.get("driverClassName"));

                    mpv.addPropertyValue("url", dsMap.get("url"));

                    mpv.addPropertyValue("username", dsMap.get("username"));

                    mpv.addPropertyValue("password", dsMap.get("password"));

               }

              }

       }

      

 

       @SuppressWarnings("unchecked")

       @Override

       publicvoidpostProcessBeanDefinitionRegistry(BeanDefinitionRegistry registry) throws BeansException{

        System.out.println("MultipleDataSourceBeanDefinitionRegistryPostProcessor.postProcessBeanDefinitionRegistry()");

              try {

                     if(!dataSourceMap.isEmpty()){

                            //不为空的时候，进行注册bean.

                            for(Entry<String,Map<String,Object>>entry:dataSourceMap.entrySet()){

                                   Objecttype = entry.getValue().get("type");//获取数据源类型，没有设置为默认的数据源.

                                   if(type == null){

                                          type= DATASOURCE_TYPE_DEFAULT;

                                   }

                                   registerBean(registry, entry.getKey(),(Class<?extendsDataSource>)Class.forName(type.toString()));

                            }

                     }

              }catch(ClassNotFoundException  e) {

                     //异常捕捉.

                     e.printStackTrace();

              }

       }

      

      

       /**

        * 注意重写的方法 setEnvironment 是在系统启动的时候被执行。

        * 这个方法主要是：加载多数据源配置

        * 从application.properties文件中进行加载;

        */

       @Override

       publicvoidsetEnvironment(Environment environment) {

              System.out.println("MultipleDataSourceBeanDefinitionRegistryPostProcessor.setEnvironment()");

             

              /*

               * 获取application.properties配置的多数据源配置，添加到map中，以后在postProcessBeanDefinitionRegistry进行注册。

               */

              //获取到前缀是"custom.datasource."的属性列表值.

              RelaxedPropertyResolverpropertyResolver = newRelaxedPropertyResolver(environment,"custom.datasource.");

              //获取到全部数据源的名称.

              StringdsPrefixs = propertyResolver.getProperty("names");

              String[]dsPrefixsArr = dsPrefixs.split(",");

              for(String dsPrefix:dsPrefixsArr){

                     /*

                      * 获取到子属性，对应一个map;

                      * 也就是这个map的key就是

                      *

                      * type、driver-class-name等;

                      *

                      *

                      */

                     Map<String,Object> dsMap = propertyResolver.getSubProperties(dsPrefix + ".");

                     //存放到一个map集合中，以后在注入进行使用.

                     dataSourceMap.put(dsPrefix, dsMap);

              }

       }

 

      

      

       /**

    * 注册Bean到Spring

    *

    * @param registry

    * @param name

    * @param beanClass

    * @author SHANHY

    * @create 2016年1月22日

    */

   privatevoid registerBean(BeanDefinitionRegistry registry, String name, Class<?>beanClass) {

        AnnotatedGenericBeanDefinition abd = newAnnotatedGenericBeanDefinition(beanClass);

       

        //单例仍是原型等等...做用域对象.

        ScopeMetadata scopeMetadata = this.scopeMetadataResolver.resolveScopeMetadata(abd);

        abd.setScope(scopeMetadata.getScopeName());

        // 能够自动生成name

        String beanName = (name != null ? name : this.beanNameGenerator.generateBeanName(abd, registry));

 

        AnnotationConfigUtils.processCommonDefinitionAnnotations(abd);

 

        BeanDefinitionHolder definitionHolder = newBeanDefinitionHolder(abd, beanName);

        BeanDefinitionReaderUtils.registerBeanDefinition(definitionHolder, registry);

   }

}

       在代码中已经加入了，注释，你们能够好好看，在这里简单说明下。

以上代码的执行顺序是：

setEnvironment()-->postProcessBeanDefinitionRegistry()--> postProcessBeanFactory()

 

在setEnvironment（）方法中主要是读取了application.properties的配置；

在postProcessBeanDefinitionRegistry（）方法中主要注册为spring的bean对象；

在postProcessBeanFactory（）方法中主要是注入从setEnvironment方法中读取的application.properties配置信息。

须要注意的是这里并无读取其它相同的数据源公共配置，这里咱们不作介绍，在下节介绍，主要是由于这节在实际中咱们并不会这么使用，这里只是过渡下，方便下节进行讲解。