1. 程式人生 > >在Springboot2.0項目中使用Druid配置多數據源

在Springboot2.0項目中使用Druid配置多數據源

clas sin word npr PE ng- 多個 pap final

在Springboot出現之前配置數據源以及相關的事物,緩存等內容一直是個繁瑣的工作,但是Springboot出現後這些基本都可以靠默認配置搞定,就變得很輕松了。這就是現在推崇模板>配置的原因,不過話說回來,如果你想配和模板不同,該繁瑣的地方仍然一樣繁瑣,比如今天要講的,在Springboot項目中配多數據源。接下來逐漸講一下。

為項目添加Druid依賴

直接去Maven倉庫搜最新的Druid和log4j,阿裏的Druid強依賴log4j,卻沒加到Maven依賴中,實際上,不裝log4j的話會報運行錯誤。同理Spring-boot-web,不過這個基本所有項目都有不特意強調了。

<!-- https://mvnrepository.com/artifact/com.alibaba/druid -->
<dependency> <groupId>com.alibaba</groupId> <artifactId>druid</artifactId> <version>1.1.9</version> </dependency> <!-- https://mvnrepository.com/artifact/log4j/log4j --> <dependency>
<groupId>log4j</groupId> <artifactId>log4j</artifactId> <version>1.2.17</version> </dependency> <dependency> <groupId>org.springframework.boot</groupId> <artifactId>
spring-boot-starter-web</artifactId> </dependency>

生成不同數據源需要的Bean

加入依賴後,我們需要以下幾個步驟:

  1. 生成對應每個數據源的DruidDataSource Bean(由於對於多個數據庫,同類名Bean有多個,故給每個Bean顯性賦名,顯性註入防錯誤)
  2. 將不同數據源註入到JPA,需要配置EntityManager,JpaProperties,PlatformTransactionManager

代碼如下,以下代碼省略getter,setter,一些具體問題參看代碼中註釋:

DruidDBConfig.java

@Configuration
@ConfigurationProperties("druid")
public class DruidDBConfig {
    //此處替換成你自己的logger方案
    private final Logger logger = LogSingleton.getLogSingleton().getLogger();

    //這些變量在yml中配置

    private String dbUrl1;
    private String username1;
    private String password1;
    private String driverClassName1;
    private String validationQuery1;

    private String dbUrl2;
    private String username2;
    private String password2;
    private String driverClassName2;
    private String validationQuery2;

    @Value("5")
    private int initialSize;
    @Value("5")
    private int minIdle;
    @Value("30")
    private int maxActive;
    /**
     * 單位是毫秒
     */
    @Value("60000")
    private int maxWait;
    /**
     * 配置間隔多久才進行一次檢測,檢測需要關閉的空閑連接,單位是毫秒
     */
    @Value("60000")
    private int timeBetweenEvictionRunsMillis;
    /**
     * 配置一個連接在池中最小生存的時間,單位是毫秒
     */
    @Value("300000")
    private int minEvictableIdleTimeMillis;
    @Value("true")
    private boolean testWhileIdle;
    @Value("true")
    private boolean testOnBorrow;
    @Value("false")
    private boolean testOnReturn;

    /**
     * 打開PSCache,並且指定每個連接上PSCache的大小
     */
    @Value("true")
    private boolean poolPreparedStatements;
    @Value("50")
    private int maxPoolPreparedStatementPerConnectionSize;
    /**
     * 配置監控統計攔截的filters,去掉後監控界面sql無法統計,‘wall‘用於防火墻
     */
    @Value("stat,wall,log4j")
    private String filters;
    /**
     * 通過connectProperties屬性來打開mergeSql功能;慢SQL記錄
     */
    @Value("druid.stat.mergeSql=true;druid.stat.slowSqlMillis=500")
    private String connectionProperties;

    @Bean(name = "primaryDataSource")
    @Qualifier("primaryDataSource")
    @Primary
    public DataSource primaryDataSource() {
        return getDruidDataSource(username1, password1, dbUrl1, driverClassName1, validationQuery1);
    }

    @Bean(name = "secondaryDataSource")
    @Qualifier("secondaryDataSource")
    public DataSource secondaryDataSource() {
        return getDruidDataSource(username2, password2, dbUrl2, driverClassName2, validationQuery2);
    }

    /*
    /根據配置文件生成DruidDataSource
     */
    private DruidDataSource getDruidDataSource(String username, String password, String url, String driverClassName, String validationQuery) {
        DruidDataSource datasource = new DruidDataSource();

        datasource.setUrl(url);
        datasource.setUsername(username);
        datasource.setPassword(password);
        datasource.setDriverClassName(driverClassName);

        //configuration
        datasource.setInitialSize(initialSize);
        datasource.setMinIdle(minIdle);
        datasource.setMaxActive(maxActive);
        datasource.setMaxWait(maxWait);
        datasource.setTimeBetweenEvictionRunsMillis(timeBetweenEvictionRunsMillis);
        datasource.setMinEvictableIdleTimeMillis(minEvictableIdleTimeMillis);
        datasource.setValidationQuery(validationQuery);
        datasource.setTestWhileIdle(testWhileIdle);
        datasource.setTestOnBorrow(testOnBorrow);
        datasource.setTestOnReturn(testOnReturn);
        datasource.setPoolPreparedStatements(poolPreparedStatements);
        datasource.setMaxPoolPreparedStatementPerConnectionSize(maxPoolPreparedStatementPerConnectionSize);
        try {
            datasource.setFilters(filters);
        } catch (SQLException e) {
            logger.severe("druid configuration initialization filter : " + e);
        }
        datasource.setConnectionProperties(connectionProperties);

        return datasource;
    }
}

PrimaryConfig.java

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(
        entityManagerFactoryRef = "entityManagerFactoryPrimary",
        transactionManagerRef = "transactionManagerPrimary",
        basePackages = {"com.XX.XX請替換為你的數據源1Repository目錄"})
public class PrimaryConfig {

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

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

    @Resource
    private JpaProperties jpaProperties;

    private Map<String, Object> getVendorProperties() {
        //在yml中配置Hibernate,因此此處僅需一個空的Hibernate設置。
        return jpaProperties.getHibernateProperties(new HibernateSettings());
    }

    /**
     * 設置實體類所在位置
     */
    @Primary
    @Bean(name = "entityManagerFactoryPrimary")
    public LocalContainerEntityManagerFactoryBean entityManagerFactoryPrimary(EntityManagerFactoryBuilder builder) {
        return builder
                .dataSource(primaryDataSource)
                .packages("com.XX.XX請替換為你的數據源1Entity目錄")
                .persistenceUnit("primaryPersistenceUnit")
                .properties(getVendorProperties())
                .build();
    }

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

SecondaryConfig.java

@Configuration
@EnableTransactionManagement
@EnableJpaRepositories(
        entityManagerFactoryRef = "entityManagerFactorySecondary",
        transactionManagerRef = "transactionManagerSecondary",
        basePackages = {"com.XX.XX請替換為你的數據源2Repository目錄"})
public class SecondaryConfig {

    @Resource
    @Qualifier("secondaryDataSource")
    private DataSource secondaryDataSource;

    @Bean(name = "entityManagerSecondary")
    public EntityManager entityManager(EntityManagerFactoryBuilder builder) {
        return entityManagerFactorySecondary(builder).getObject().createEntityManager();
    }

    @Resource
    private JpaProperties jpaProperties;

    private Map<String, Object> getVendorProperties() {
        //在yml中配置Hibernate,因此此處僅需一個空的Hibernate設置。
        return jpaProperties.getHibernateProperties(new HibernateSettings());
    }

    @Bean(name = "entityManagerFactorySecondary")
    public LocalContainerEntityManagerFactoryBean entityManagerFactorySecondary(EntityManagerFactoryBuilder builder) {
        return builder
                .dataSource(secondaryDataSource)
                .packages("com.XX.XX請替換為你的數據源2Entity目錄")
                .persistenceUnit("secondaryPersistenceUnit")
                .properties(getVendorProperties())
                .build();
    }

    @Bean(name = "transactionManagerSecondary")
    PlatformTransactionManager transactionManagerSecondary(EntityManagerFactoryBuilder builder) {
        return new JpaTransactionManager(entityManagerFactorySecondary(builder).getObject());
    }

}

之後配置yml,註入在DruidDBConfig中未設置的值:

druid:
    # 第一個數據源的配置
    driverClassName1: com.microsoft.sqlserver.jdbc.SQLServerDriver
    dbUrl1: jdbc:sqlserver://XXX;databaseName=XXX
    username1: sa
    password1: 123456
    # 測試連通,必須是單條的可順利執行的SQL語句。
    validationQuery1: select 1
    # 第二個數據源的配置
    driverClassName2: com.microsoft.sqlserver.jdbc.SQLServerDriver
    dbUrl2: jdbc:sqlserver://YYY;databaseName=YYY
    username2: sa
    password2: 123456
    # 測試連通,必須是單條的可順利執行的SQL語句。
    validationQuery2: select 1
spring:
    # hibernate配置
    jpa:
        database: sql_server
        generate-ddl: true
        show-sql: true
        hibernate:
            ddl-auto: update
            naming:
                physical-strategy: org.springframework.boot.orm.jpa.hibernate.SpringPhysicalNamingStrategy

一些坑

一定要加log4j依賴

Entity的主鍵自增策略要從@GeneratedValue改為@GeneratedValue(strategy=GeneratedType.IDENTITY),因為默認方法在Druid下無法工作。

在Springboot2.0項目中使用Druid配置多數據源