spring boot springjpa 支持多個數(shù)據(jù)源的實例代碼

更新時間：2018年04月18日 16:00:24 作者：ba5189tsl

這篇文章主要介紹了spring boot springjpa 支持多個數(shù)據(jù)源的實例代碼,需要的朋友可以參考下

1.SpringBoot的程序啟動類

import org.springframework.boot.SpringApplication; 
import org.springframework.boot.autoconfigure.EnableAutoConfiguration; 
import org.springframework.boot.builder.SpringApplicationBuilder; 
import org.springframework.boot.context.web.SpringBootServletInitializer; 
import org.springframework.context.annotation.ComponentScan; 
@ComponentScan 
@EnableAutoConfiguration 
//@EnableJpaRepositories(basePackages = "com.sonychina.backend.repository") 
public class Application extends SpringBootServletInitializer { 
  public static void main(String[] args) { 
    SpringApplication app = new SpringApplication(Application.class); 
    app.run(args); 
    //SpringApplication.run(Application.class, args); 
  } 
  @Override 
  protected SpringApplicationBuilder configure(SpringApplicationBuilder application) { 
    return application.sources(Application.class); 
  } 
}

2.雙數(shù)據(jù)源配置類

import java.util.Map; 
import javax.sql.DataSource; 
import org.springframework.beans.factory.annotation.Autowired; 
import org.springframework.boot.autoconfigure.jdbc.DataSourceBuilder; 
import org.springframework.boot.autoconfigure.orm.jpa.EntityManagerFactoryBuilder; 
import org.springframework.boot.autoconfigure.orm.jpa.JpaProperties; 
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.data.jpa.repository.config.EnableJpaRepositories; 
import org.springframework.orm.jpa.LocalContainerEntityManagerFactoryBean; 
import com.test.entity.statistic.SysUser; 
import com.test.repository.system.SystemRepository; 
@Configuration 
@EnableJpaRepositories(entityManagerFactoryRef="entityManagerFactoryPrimary", 
  basePackageClasses= {SystemRepository.class}) 
public class GlobalDataConfiguration { 
// @Autowired 
// private DBConfig dbConfig; 
  @Autowired 
  private JpaProperties jpaProperties; 
  @Bean(name="primaryDataSource") 
  @Primary 
  @ConfigurationProperties(prefix="datasource.primary") 
  public DataSource primaryDataSource() { 
    System.out.println("-------------------- primaryDataSource init ---------------------"); 
    return DataSourceBuilder.create().build(); 
  } 
  @Bean(name="secondaryDataSource") 
  @ConfigurationProperties(prefix="datasource.secondary") 
  public DataSource secondaryDataSource() { 
    System.out.println("-------------------- secondaryDataSource init ---------------------"); 
//   DataSourceBuilder factory = DataSourceBuilder  
//       .create(DBConfig.class.getClassLoader())  
//       .driverClassName(dbConfig.getDriver())  
//       .url(dbConfig.getUrl())  
//       .username(dbConfig.getUser())  
//       .password(dbConfig.getPassword());  
//   return factory.build(); 
    return DataSourceBuilder.create().build(); 
  } 
// @Bean(name = "entityManagerPrimary") 
// @Primary 
// public EntityManager entityManager(EntityManagerFactoryBuilder builder) { 
//   return customerEntityManagerFactory(builder).getObject().createEntityManager(); 
// } 
  @Bean(name="entityManagerFactoryPrimary") 
  @Primary 
  public LocalContainerEntityManagerFactoryBean customerEntityManagerFactory(EntityManagerFactoryBuilder builder) { 
    return builder.dataSource(primaryDataSource()) 
          .properties(getVendorProperties(primaryDataSource())) 
          .packages(SysUser.class) 
          .persistenceUnit("system") 
          .build(); 
  } 
  private Map<String, String> getVendorProperties(DataSource dataSource) { 
    return jpaProperties.getHibernateProperties(dataSource); 
  } 
}

3.第二個jpa實體管理器

import java.util.Map; 
import javax.sql.DataSource; 
import org.springframework.beans.factory.annotation.Autowired; 
import org.springframework.beans.factory.annotation.Qualifier; 
import org.springframework.boot.autoconfigure.orm.jpa.EntityManagerFactoryBuilder; 
import org.springframework.boot.autoconfigure.orm.jpa.JpaProperties; 
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 com.test.entity.manage.Banner; 
import com.test.repository.manage.BannerRepository; 
@Configuration 
@EnableTransactionManagement 
@EnableJpaRepositories(entityManagerFactoryRef="entityManagerFactorySecondary", 
  transactionManagerRef="transactionManagerSecondary", 
  basePackageClasses= {BannerRepository.class}) 
public class SecondEMFBConfig { 
  @Autowired 
  private JpaProperties jpaProperties; 
  @Autowired@Qualifier("secondaryDataSource") 
  private DataSource dataSource; 
// @Bean(name = "entityManagerPrimary") 
// @Primary 
// public EntityManager entityManager(EntityManagerFactoryBuilder builder) { 
//   return customerEntityManagerFactory(builder).getObject().createEntityManager(); 
// } 
  @Bean(name="entityManagerFactorySecondary") 
  public LocalContainerEntityManagerFactoryBean customerEntityManagerFactory(EntityManagerFactoryBuilder builder) { 
    return builder.dataSource(dataSource) 
          .properties(getVendorProperties(dataSource)) 
          .packages(Banner.class) 
          .persistenceUnit("customers") 
          .build(); 
  } 
  private Map<String, String> getVendorProperties(DataSource dataSource) { 
    return jpaProperties.getHibernateProperties(dataSource); 
  } 
  @Bean(name = "transactionManagerSecondary") 
  PlatformTransactionManager transactionManagerSecondary(EntityManagerFactoryBuilder builder) { 
    return new JpaTransactionManager(customerEntityManagerFactory(builder).getObject()); 
  } 
}

4.repository類舉例

import org.springframework.data.jpa.repository.JpaRepository; 
import org.springframework.data.jpa.repository.Modifying; 
import org.springframework.data.jpa.repository.Query; 
import com.test.entity.manage.Banner; 
public interface BannerRepository extends JpaRepository<Banner, Long> { 
  @Modifying 
  @Query("update Banner m set m.name=?1 where m.id=?2") 
  public void update(String bannerName, Long id); 
}

1.5.注意：對@Primary修飾的LocalContainerEntityManagerFactoryBean可以不用指定TransactionManager，spring上下文自動使用默認的JpaTransactionManager，但是對于第二個或第三個等等必須指定TransactionManager?？梢詤⒖約pringboot官方文檔中的相關章節(jié)。

總結

以上所述是小編給大家介紹的spring boot springjpa 支持多個數(shù)據(jù)源的實例代碼，希望對大家有所幫助，如果大家有任何疑問請給我留言，小編會及時回復大家的。在此也非常感謝大家對腳本之家網(wǎng)站的支持！

您可能感興趣的文章: