配置多个数据源

src/main/java/cn/quantgroup/customer/config/data/DbConfig.java

 package cn.quantgroup.customer.config.data;
 
-import com.zaxxer.hikari.HikariConfig;
-import com.zaxxer.hikari.HikariDataSource;
+import javax.sql.DataSource;
 import org.springframework.beans.factory.annotation.Value;
+import org.springframework.boot.autoconfigure.jdbc.DataSourceBuilder;
 import org.springframework.context.annotation.Bean;
 import org.springframework.context.annotation.Configuration;
-import org.springframework.context.annotation.DependsOn;
 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.JpaVendorAdapter;
-import org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean;
-import org.springframework.orm.jpa.vendor.Database;
-import org.springframework.orm.jpa.vendor.HibernateJpaVendorAdapter;
-import org.springframework.transaction.PlatformTransactionManager;
-import org.springframework.transaction.annotation.EnableTransactionManagement;
-
-import javax.persistence.EntityManagerFactory;
-import javax.sql.DataSource;
-import java.util.Properties;
 
 @Configuration
-@EnableJpaRepositories(basePackages = "cn.quantgroup.customer.repo")
-@EnableTransactionManagement
+//@EnableJpaRepositories(basePackages = "cn.quantgroup.customer.repo")
+//@EnableTransactionManagement
 public class DbConfig {
 
     @Value("${data.mysql.jdbc-url}")
@@ -34,56 +21,83 @@ public class DbConfig {
     @Value("${data.mysql.user}")
     private String user;
 
+
+    @Value("${data.mysql.secondary.jdbc-url}")
+    private String jdbcUrlSecondary;
+
+    @Value("${data.mysql.secondary.password}")
+    private String passwordSecondary;
+
+    @Value("${data.mysql.secondary.user}")
+    private String userSecondary;
+
+
     @Value("${data.mysql.max-pool-size}")
     private Integer maxPoolSize;
 
-    @Bean
-    @DependsOn(value = "dataSource")
-    public EntityManagerFactory entityManagerFactory(DataSource dataSource) {
-        LocalContainerEntityManagerFactoryBean entityManager = new LocalContainerEntityManagerFactoryBean();
-        entityManager.setDataSource(dataSource);
-        entityManager.setPackagesToScan("cn.quantgroup.customer");
-        entityManager.setPersistenceUnitName("dataSource");
-        Properties properties = new Properties();
-        properties.put("hibernate.jdbc.batch_size", 30);
-        properties.put("hibernate.order_inserts", true);
-        properties.put("hibernate.order_updates", true);
-        entityManager.setJpaProperties(properties);
-        entityManager.setJpaVendorAdapter(jpaVendorAdapter());
-        entityManager.afterPropertiesSet();
-        return entityManager.getObject();
-    }
+    private String driverClassName ="com.mysql.jdbc.Driver";
+
 
-    @Bean
+    @Bean(name = "primaryDataSource")
     @Primary
-    public DataSource dataSource() {
-        HikariConfig config = new HikariConfig();
-        config.setJdbcUrl(jdbcUrl);
-        config.setPassword(password);
-        config.setUsername(user);
-        config.setMaximumPoolSize(maxPoolSize);
-        config.setMinimumIdle(20);
-        config.addDataSourceProperty("cachePrepStmts", "true");
-        config.addDataSourceProperty("prepStmtCacheSize", "250");
-        config.addDataSourceProperty("prepStmtCacheSqlLimit", "2048");
-        return new HikariDataSource(config);
+    public DataSource primaryDataSource() {
+        return DataSourceBuilder.create().url(jdbcUrl).username(user).password(password)
+            .driverClassName(driverClassName).build();
+    }
 
+    @Bean(name = "secondaryDataSource")
+    public DataSource secondaryDataSource() {
+        return DataSourceBuilder.create().url(jdbcUrlSecondary).username(userSecondary).
+            password(passwordSecondary).driverClassName(driverClassName).build();
     }
 
+//    @Bean
+//    @DependsOn(value = "dataSource")
+//    public EntityManagerFactory entityManagerFactory(DataSource dataSource) {
+//        LocalContainerEntityManagerFactoryBean entityManager = new LocalContainerEntityManagerFactoryBean();
+//        entityManager.setDataSource(dataSource);
+//        entityManager.setPackagesToScan("cn.quantgroup.customer");
+//        entityManager.setPersistenceUnitName("dataSource");
+//        Properties properties = new Properties();
+//        properties.put("hibernate.jdbc.batch_size", 30);
+//        properties.put("hibernate.order_inserts", true);
+//        properties.put("hibernate.order_updates", true);
+//        entityManager.setJpaProperties(properties);
+//        entityManager.setJpaVendorAdapter(jpaVendorAdapter());
+//        entityManager.afterPropertiesSet();
+//        return entityManager.getObject();
+//    }
 
-    private JpaVendorAdapter jpaVendorAdapter() {
-        HibernateJpaVendorAdapter hibernateJpaVendorAdapter = new HibernateJpaVendorAdapter();
-        hibernateJpaVendorAdapter.setShowSql(false);
-        hibernateJpaVendorAdapter.setGenerateDdl(false);
-        hibernateJpaVendorAdapter.setDatabase(Database.MYSQL);
-        return hibernateJpaVendorAdapter;
-    }
+//    @Bean
+//    @Primary
+//    public DataSource dataSource() {
+//        HikariConfig config = new HikariConfig();
+//        config.setJdbcUrl(jdbcUrl);
+//        config.setPassword(password);
+//        config.setUsername(user);
+//        config.setMaximumPoolSize(maxPoolSize);
+//        config.setMinimumIdle(20);
+//        config.addDataSourceProperty("cachePrepStmts", "true");
+//        config.addDataSourceProperty("prepStmtCacheSize", "250");
+//        config.addDataSourceProperty("prepStmtCacheSqlLimit", "2048");
+//        return new HikariDataSource(config);
+//
+//    }
 
-    @Bean
-    public PlatformTransactionManager transactionManager(EntityManagerFactory entityManagerFactory) {
-        JpaTransactionManager transactionManager = new JpaTransactionManager();
-        transactionManager.setEntityManagerFactory(entityManagerFactory);
-        return transactionManager;
-    }
+
+//    private JpaVendorAdapter jpaVendorAdapter() {
+//        HibernateJpaVendorAdapter hibernateJpaVendorAdapter = new HibernateJpaVendorAdapter();
+//        hibernateJpaVendorAdapter.setShowSql(false);
+//        hibernateJpaVendorAdapter.setGenerateDdl(false);
+//        hibernateJpaVendorAdapter.setDatabase(Database.MYSQL);
+//        return hibernateJpaVendorAdapter;
+//    }
+//
+//    @Bean
+//    public PlatformTransactionManager transactionManager(EntityManagerFactory entityManagerFactory) {
+//        JpaTransactionManager transactionManager = new JpaTransactionManager();
+//        transactionManager.setEntityManagerFactory(entityManagerFactory);
+//        return transactionManager;
+//    }
 
 }

src/main/java/cn/quantgroup/customer/config/data/PrimaryConfig.java

+package cn.quantgroup.customer.config.data;
+
+import org.springframework.beans.factory.annotation.Qualifier;
+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.annotation.Resource;
+import javax.persistence.EntityManager;
+import javax.sql.DataSource;
+import java.util.Properties;
+
+
+@Configuration
+@EnableTransactionManagement
+@EnableJpaRepositories(
+        entityManagerFactoryRef = "entityManagerFactoryPrimary",
+        transactionManagerRef = "transactionManagerPrimary",
+        basePackages = {"cn.quantgroup.customer.repo"}) //设置Repository所在位置
+
+public class PrimaryConfig {
+
+    @Resource
+    @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) {
+        LocalContainerEntityManagerFactoryBean entityManagerFactory = builder
+                .dataSource(primaryDataSource)
+                .packages("cn.quantgroup.customer.entity") //设置实体类所在位置
+                .persistenceUnit("primaryDataSource")
+                .build();
+        Properties properties = new Properties();
+        properties.put("hibernate.jdbc.batch_size", 30);
+        properties.put("hibernate.order_inserts", true);
+        properties.put("hibernate.order_updates", true);
+        entityManagerFactory.setJpaProperties(properties);
+        return entityManagerFactory;
+    }
+
+    @Primary
+    @Bean(name = "transactionManagerPrimary")
+    public PlatformTransactionManager transactionManagerPrimary(EntityManagerFactoryBuilder builder) {
+        return new JpaTransactionManager(entityManagerFactoryPrimary(builder).getObject());
+    }
+
+}
\ No newline at end of file

src/main/java/cn/quantgroup/customer/config/data/SecondaryConfig.java

+package cn.quantgroup.customer.config.data;
+
+import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.boot.orm.jpa.EntityManagerFactoryBuilder;
+import org.springframework.context.annotation.Bean;
+import org.springframework.context.annotation.Configuration;
+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.annotation.Resource;
+import javax.persistence.EntityManager;
+import javax.sql.DataSource;
+import java.util.Properties;
+
+
+@Configuration
+@EnableTransactionManagement
+@EnableJpaRepositories(
+        entityManagerFactoryRef = "entityManagerFactorySecondary",
+        transactionManagerRef = "transactionManagerSecondary",
+        basePackages = {"cn.quantgroup.third.customer.repo"}) //设置Repository所在位置
+public class SecondaryConfig {
+
+    @Resource
+    @Qualifier("secondaryDataSource")
+    private DataSource secondaryDataSource;
+
+
+    @Bean(name = "entityManagerSecondary")
+    public EntityManager entityManager(EntityManagerFactoryBuilder builder) {
+        return entityManagerFactorySecondary(builder).getObject().createEntityManager();
+    }
+
+    @Bean(name = "entityManagerFactorySecondary")
+    public LocalContainerEntityManagerFactoryBean entityManagerFactorySecondary(EntityManagerFactoryBuilder builder) {
+        LocalContainerEntityManagerFactoryBean entityManagerFactory
+                = builder
+                .dataSource(secondaryDataSource)
+                .packages("cn.quantgroup.third.customer.entity") //设置实体类所在位置
+                .persistenceUnit("secondaryDataSource")
+                .build();
+        Properties properties = new Properties();
+        properties.put("hibernate.jdbc.batch_size", 30);
+        properties.put("hibernate.order_inserts", true);
+        properties.put("hibernate.order_updates", true);
+        entityManagerFactory.setJpaProperties(properties);
+        entityManagerFactory.setJpaProperties(properties);
+        return entityManagerFactory;
+    }
+
+    @Bean(name = "transactionManagerSecondary")
+    public PlatformTransactionManager transactionManagerPrimary(EntityManagerFactoryBuilder builder) {
+        return new JpaTransactionManager(entityManagerFactorySecondary(builder).getObject());
+    }
+}
\ No newline at end of file