[TOC] # 多數據源的支持 ## 同源數據庫的多源支持 日常項目中因為使用的分布式開發模式，不同的服務有不同的數據源，常常需要在一個項目中使用多個數據源，因此需要配置sping data jpa對多數據源的使用，一般分一下為三步： 1. 創建數據庫 test1 和 test2 2. 配置多數據源 3. 不同源的 repository 放?不同包路徑 4. 聲明不同的包路徑下使?不同的數據源、事務?持 不同的包路徑下創建對應的 repository 5. 測試使? **配置兩個數據源** ~~~ spring.datasource.primary.jdbc-url=jdbc:mysql://localhost:3306/test1?serverTimezone=UTC&useUnicode=true&characterEncoding=utf-8&useSSL=true spring.datasource.primary.username=root spring.datasource.primary.password=root spring.datasource.primary.driver-class-name=com.mysql.cj.jdbc.Driver spring.datasource.secondary.jdbc-url=jdbc:mysql://localhost:3306/test2?serverTimezone=UTC&useUnicode=true&characterEncoding=utf-8&useSSL=true spring.datasource.secondary.username=root spring.datasource.secondary.password=root spring.datasource.secondary.driver-class-name=com.mysql.cj.jdbc.Driver #spring.jpa.show-sql=true #SQL 輸出 spring.jpa.properties.hibernate.show_sql=true spring.jpa.properties.hibernate.hbm2ddl.auto=create spring.jpa.properties.hibernate.dialect=org.hibernate.dialect.MySQL5InnoDBDialect #format ?下 SQL 進?輸出 spring.jpa.properties.hibernate.format_sql=true ~~~ **配置數據源** 創建 DataSourceCon?g 添加 @Con?guration 注解，在項?啟動時運?初始化數據庫資源 ~~~ import org.springframework.beans.factory.annotation.Autowired; import org.springframework.boot.autoconfigure.orm.jpa.HibernateProperties; import org.springframework.boot.autoconfigure.orm.jpa.HibernateSettings; import org.springframework.boot.autoconfigure.orm.jpa.JpaProperties; import org.springframework.boot.context.properties.ConfigurationProperties; import org.springframework.boot.jdbc.DataSourceBuilder; import org.springframework.context.annotation.Bean; import org.springframework.context.annotation.Configuration; import org.springframework.context.annotation.Primary; import javax.sql.DataSource; import java.util.Map; @Configuration public class DataSourceConfig { @Autowired private JpaProperties jpaProperties; @Autowired private HibernateProperties hibernateProperties; //加載第一個數據源 @Bean(name = "primaryDataSource") @Primary @ConfigurationProperties("spring.datasource.primary") public DataSource firstDataSource() { return DataSourceBuilder.create().build(); } //加載第二個數據源 @Bean(name = "secondaryDataSource") @ConfigurationProperties("spring.datasource.secondary") public DataSource secondDataSource() { return DataSourceBuilder.create().build(); } //加載 JPA 的相關配置信息，JpaProperties 是 JPA 的?些屬性配置信息，構建 LocalEntityManagerFactoryBean 需要參數信息注?到?法中。 @Bean(name = "vendorProperties") public Map<String, Object> getVendorProperties() { return hibernateProperties.determineHibernateProperties(jpaProperties.getProperties(), new HibernateSettings()); } } ~~~ **第?個數據源的加載配置過程** ?先來看第?個數據源的加載配置過程，創建 PrimaryCon?g 類，將上?創建好的第?個數據源注?到類中，添加 @Con?guration 和 @EnableTransactionManagement 注解，第?個代表啟動時加載，第?個注解 表示啟?事務，同時將第?個數據源和 JPA 配置信息注?到類中。 ~~~ import java.util.Map; import javax.persistence.EntityManager; import javax.sql.DataSource; import org.springframework.beans.factory.annotation.Autowired; 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; //將我們在類中配置好的 EntityManager 和事務信息注?到對應數據源的 repository ?錄 下，這樣此?錄下的 repository 就會擁有對應數據源和事務的信息 @Configuration //啟動時加載 @EnableTransactionManagement //啟用事務 @EnableJpaRepositories( entityManagerFactoryRef="entityManagerFactoryPrimary", transactionManagerRef="transactionManagerPrimary", basePackages= { "com.neo.repository.test1" }) //設置dao（repo）所在位置, basePackages ?持設置多個包路徑 public class PrimaryConfig { @Autowired @Qualifier("primaryDataSource") private DataSource primaryDataSource; @Autowired @Qualifier("vendorProperties") private Map<String, Object> vendorProperties; //LocalEntityManagerFactoryBean 負責創建?個適合于僅使? JPA 進?數據訪問的環境的 EntityManager， 構建的時候需要指明提示實體類的包路徑、數據源和 JPA 配置信息 @Bean(name = "entityManagerFactoryPrimary") @Primary public LocalContainerEntityManagerFactoryBean entityManagerFactoryPrimary (EntityManagerFactoryBuilder builder) { return builder .dataSource(primaryDataSource) .properties(vendorProperties) .packages("com.neo.model") //設置實體類所在位置 .persistenceUnit("primaryPersistenceUnit") .build(); } //利?上?的 entityManagerFactoryPrimary() ?法構建好最終的 EntityManager //EntityManager 是 JPA 中?于增、刪、改、查的接?，它的作?相當于?座橋梁，連接內存中的 Java 對象 和數據庫的數據存儲。使? EntityManager 中的相關接?對數據庫實體進?操作的時候， EntityManager 會 跟蹤實體對象的狀態，并決定在特定時刻將對實體的操作映射到數據庫操作上?。 @Bean(name = "entityManagerPrimary") @Primary public EntityManager entityManager(EntityManagerFactoryBuilder builder) { return entityManagerFactoryPrimary(builder).getObject().createEntityManager(); } //給數據源添加上 JPA 事務 @Bean(name = "transactionManagerPrimary") @Primary PlatformTransactionManager transactionManagerPrimary(EntityManagerFactoryBuilder builder) { return new JpaTransactionManager(entityManagerFactoryPrimary(builder).getObject()); } } ~~~ **第二個數據源的加載配置過程** 第?個數據源配置和第?個數據源配置類似，只是?法上去掉了注解：@Primary，第?個數據源數據源加載 配置類 SecondaryCon?g 完整代碼如下： ~~~ import org.springframework.beans.factory.annotation.Autowired; 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.persistence.EntityManager; import javax.sql.DataSource; import java.util.Map; @Configuration @EnableTransactionManagement @EnableJpaRepositories( entityManagerFactoryRef="entityManagerFactorySecondary", transactionManagerRef="transactionManagerSecondary", basePackages= { "com.neo.repository.test2" }) public class SecondaryConfig { @Autowired @Qualifier("secondaryDataSource") private DataSource secondaryDataSource; @Autowired @Qualifier("vendorProperties") private Map<String, Object> vendorProperties; @Bean(name = "entityManagerFactorySecondary") public LocalContainerEntityManagerFactoryBean entityManagerFactorySecondary (EntityManagerFactoryBuilder builder) { return builder .dataSource(secondaryDataSource) .properties(vendorProperties) .packages("com.neo.model") .persistenceUnit("secondaryPersistenceUnit") .build(); } @Bean(name = "entityManagerSecondary") public EntityManager entityManager(EntityManagerFactoryBuilder builder) { return entityManagerFactorySecondary(builder).getObject().createEntityManager(); } @Bean(name = "transactionManagerSecondary") PlatformTransactionManager transactionManagerSecondary(EntityManagerFactoryBuilder builder) { return new JpaTransactionManager(entityManagerFactorySecondary(builder).getObject()); } } ~~~ ## 異構數據庫多源支持 比如我們的項目中，即需要對mysql的支持，也需要對mongodb的查詢等。 實體類聲明`@Entity`關系型數據庫支持類型、聲明`@Document`為mongodb支持類型，不同的數據源使用不同的實體就可以了 ~~~ interface PersonRepository extends Repository<Person, Long> { … } @Entity public class Person { … } interface UserRepository extends Repository<User, Long> { … } @Document public class User { … } ~~~ 但是，如果User用戶既使用mysql也使用mongodb呢，也可以做混合使用 ~~~ interface JpaPersonRepository extends Repository<Person, Long> { … } interface MongoDBPersonRepository extends Repository<Person, Long> { … } @Entity @Document public class Person { … } ~~~ 也可以通過對不同的包路徑進行聲明，比如A包路徑下使用mysql,B包路徑下使用mongoDB ~~~ @EnableJpaRepositories(basePackages = "com.neo.repositories.jpa") @EnableMongoRepositories(basePackages = "com.neo.repositories.mongo") interface Configuration { } ~~~