1. 程式人生 > >SpringBoot配置多資料來源

SpringBoot配置多資料來源

SpringBoot下配置多資料來源,有的時候我們的專案可能需要從不同的資料來源獲取或者操作資料,這個時候就需要配置多資料來源;
環境 Spring-Boot 1.5.3.RELEASE 、JDK1.8、HikariDataSource
搭建SpringBoot專案可以參看我之前寫的部落格;
配置HikariDataSource資料庫連線池可以參看[SpringBoot配置HikarDataSource](http://blog.csdn.net/chen15369337607/article/details/78142751%20%E4%BC%A0%E9%80%81%E9%97%A8)  

專案結構如下:
這裡寫圖片描述

首先是com.b505包 這個包裡放置SpringBoot的啟動類:
程式碼如下:

package com.b505;

import org.springframework.boot.SpringApplication;
import org.springframework.boot.autoconfigure.SpringBootApplication;


/**
  * <p>Company: B505資訊科技研究所 </p> 
  * @Description: 專案的入口
  * @Create Date: 2017年5月11日下午9:18:50
  * @Version
: V1.00 * @Author: 追到烏雲的盡頭找太陽 */
@SpringBootApplication public class SpringBootForWeChatApplication { public static void main(String[] args) { } }

其次是在src/main/resource 檔案下新建一個datasource.properties檔案,這個檔案是我們留著配置資料來源的檔案;檔案內容如下:

#第一個資料來源
spring.datasource.primary.dataSourceClassName=com
.mysql.jdbc.jdbc2.optional.MysqlDataSource spring.datasource.primary.dataSourceProperties.serverName=localhost spring.datasource.primary.dataSourceProperties.portNumber=3306 spring.datasource.primary.dataSourceProperties.databaseName=dbgirl spring.datasource.primary.username=你的使用者名稱 spring.datasource.primary.password=你的密碼 #第二個資料來源 spring.datasource.secondary.dataSourceClassName=com.mysql.jdbc.jdbc2.optional.MysqlDataSource spring.datasource.secondary.dataSourceProperties.serverName=localhost spring.datasource.secondary.dataSourceProperties.portNumber=3306 spring.datasource.secondary.dataSourceProperties.databaseName=dbgirltest spring.datasource.secondary.username=你的使用者名稱 spring.datasource.secondary.password=你的密碼 # 下面為連線池的補充設定,應用到上面所有資料來源中 #自動提交 spring.datasource.default-auto-commit=true #指定updates是否自動提交 spring.datasource.auto-commit=true spring.datasource.maximum-pool-size=100 spring.datasource.max-idle=10 spring.datasource.max-wait=10000 spring.datasource.min-idle=5 spring.datasource.initial-size=5 spring.datasource.validation-query=SELECT 1 spring.datasource.test-on-borrow=false spring.datasource.test-while-idle=true spring.datasource.time-between-eviction-runs-millis=18800

配置檔案寫好以後,在com.b505.config包下采用java註解的形式配置這兩個資料來源到專案中首先在此包下新建一個類名為:DataSourceConfig
程式碼如下:

package com.b505.config;

import javax.sql.DataSource;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.autoconfigure.jdbc.DataSourceBuilder;
import org.springframework.boot.context.properties.ConfigurationProperties;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.context.annotation.PropertySource;

/**
  * <p>Company: B505資訊科技研究所 </p> 
  * @Description: 資料來源配置類
  * @Create Date: 2017年5月11日下午9:24:00
  * @Version: V1.00 
  * @Author: 追到烏雲的盡頭找太陽
  */
@Configuration
@PropertySource("classpath:datasource.properties")
public class DataSourceConfig {

    private Logger logger = LoggerFactory.getLogger(DataSourceConfig.class);


    /**
      * @Description: 建立第一個資料庫連線池
      * @Parameters: DataSourceConfig
      * @Return: 
      * @Create Date: 2017年5月11日下午9:39:09
      * @Version: V1.00 
      * @author:追到烏雲的盡頭找太陽
      */
    @Bean(name = "primaryDataSource")
    @Primary
    @Qualifier("primaryDataSource")
    @ConfigurationProperties(prefix="spring.datasource.primary" )
    public DataSource primaryDataSource() {
        logger.info("第一個資料庫連線池建立中.......");
        return DataSourceBuilder.create().build();
    }

    /**
      * @Description: 建立第二個資料庫連線池
      * @Parameters: DataSourceConfig
      * @Return: 
      * @Create Date: 2017年5月11日下午9:39:53
      * @Version: V1.00 
      * @author:追到烏雲的盡頭找太陽
      */
    @Bean(name = "secondaryDataSource")
    @Qualifier("secondaryDataSource")
    @ConfigurationProperties(prefix="spring.datasource.secondary")
    public DataSource secondaryDataSource() {
        logger.info("第二個資料庫連線池建立中......");
        return DataSourceBuilder.create().build();
    }

}

@Configuration 此註解標明這是一個Spring 的配置類,在專案啟動的時候,會自動掃描此配置;

@PropertySource(“classpath:datasource.properties”) 這個註解是讀取properties檔案 不需要我們自己寫讀取properties的檔案的程式碼,SpringBoot幫我們完成了,十分方便;classpath一定不能丟;

@ConfigurationProperties(prefix=”spring.datasource.primary” )
這個註解中的prefix 會獲取datasource.properties檔案中的spring.datasource.primary開頭的value值;
下面就是第一個資料來源的配置類了,新建PrimaryDataSouceConfig:

package com.b505.config;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.boot.autoconfigure.orm.jpa.JpaProperties;
import org.springframework.boot.orm.jpa.EntityManagerFactoryBuilder;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.context.annotation.Primary;
import org.springframework.data.jpa.repository.config.EnableJpaRepositories;
import org.springframework.orm.jpa.JpaTransactionManager;
import org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean;
import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.annotation.EnableTransactionManagement;

import javax.persistence.EntityManager;
import javax.sql.DataSource;

import java.util.Map;

/**
  * <p>Company: B505資訊科技研究所 </p> 
  * @Description: 第一個資料來源的配置類
  * @Create Date: 2017年5月11日下午9:22:12
  * @Version: V1.00 
  * @Author: 追到烏雲的盡頭找太陽
  */
@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(
        entityManagerFactoryRef="entityManagerFactoryPrimary",
        transactionManagerRef="transactionManagerPrimary",
        basePackages= { "com.b505.dao.girl" }) //設定Repository所在位置
public class PrimaryDataSouceConfig {

    @Autowired @Qualifier("primaryDataSource")
    private DataSource primaryDataSource;

    @Primary
    @Bean(name = "entityManagerPrimary")
    public EntityManager entityManager(EntityManagerFactoryBuilder builder) {
        return entityManagerFactoryPrimary(builder).getObject().createEntityManager();
    }

    @Primary
    @Bean(name = "entityManagerFactoryPrimary")
    public LocalContainerEntityManagerFactoryBean entityManagerFactoryPrimary (EntityManagerFactoryBuilder builder) {
        return builder
                .dataSource(primaryDataSource)
                .properties(getVendorProperties(primaryDataSource))
                .packages("com.b505.bean") //設定實體類所在位置
                .persistenceUnit("primaryPersistenceUnit")
                .build();
    }

    @Autowired
    private JpaProperties jpaProperties;

    private Map<String, String> getVendorProperties(DataSource dataSource) {
        return jpaProperties.getHibernateProperties(dataSource);
    }

    @Primary
    @Bean(name = "transactionManagerPrimary")
    public PlatformTransactionManager transactionManagerPrimary(EntityManagerFactoryBuilder builder) {
        return new JpaTransactionManager(entityManagerFactoryPrimary(builder).getObject());
    }

}

第二個資料來源的配置類與上面基本相同,如果需要操作不同的資料來源,將上面的@EnableJpaRepositories(
entityManagerFactoryRef=”entityManagerFactoryPrimary”,
transactionManagerRef=”transactionManagerPrimary”,
basePackages= { “com.b505.dao.girl” }) //設定Repository所在位置修改即可。
啟動專案後:
這裡寫圖片描述
這樣在SpringBoot中配置多資料來源就可以了。操作則可根據自己的業務需要進行。