snicoll / demo-multi-entity-managers Goto Github PK

package xxxxxx;

///Imports omitted...

import com.zaxxer.hikari.HikariDataSource;

@Configuration
@EnableJpaRepositories(
        entityManagerFactoryRef = "defaultEntityManagerFactory",
        transactionManagerRef = "defaultTransactionManager",
        basePackages = {"br.org.pti.basi.domain.repository"})
public class DefaultPersistenceConfig {
	private final PersistenceUnitManager persistenceUnitManager;

	public DefaultPersistenceConfig(ObjectProvider<PersistenceUnitManager> persistenceUnitManager) {
		this.persistenceUnitManager = persistenceUnitManager.getIfAvailable();
	}

	
	@Bean
	@ConfigurationProperties("spring.jpa")
	public JpaProperties customJpaProperties() {
		return new JpaProperties();
	}

	@Bean
	@Primary
	@ConfigurationProperties("spring.datasource")
	public DataSourceProperties defaultDataSourceProperties() {
		return new DataSourceProperties();
	}

	@Bean
	@Primary
	//@ConfigurationProperties(prefix = "spring.datasource.properties")
	public HikariDataSource defaultDataSource() {
		return defaultDataSourceProperties().initializeDataSourceBuilder()
				.type(HikariDataSource.class).build();
	}

	@Bean
	public LocalContainerEntityManagerFactoryBean defaultEntityManagerFactory( JpaProperties customJpaProperties) {
		System.out.println(customJpaProperties.getProperties().toString());
		EntityManagerFactoryBuilder builder = this.createEntityManagerFactoryBuilder( customJpaProperties );
		return builder
				.dataSource(defaultDataSource())
				.packages("br.org.pti.basi.domain.entity")
				.persistenceUnit("default")
				.build();
	}

	@Bean
	@Primary
	public JpaTransactionManager defaultTransactionManager( EntityManagerFactory defaultEntityManager ) {
		return new JpaTransactionManager(defaultEntityManager);
	}

	private EntityManagerFactoryBuilder createEntityManagerFactoryBuilder( JpaProperties customJpaProperties ) {
		JpaVendorAdapter jpaVendorAdapter = this.createJpaVendorAdapter(customJpaProperties);
		return new EntityManagerFactoryBuilder(jpaVendorAdapter, customJpaProperties.getProperties(), this.persistenceUnitManager);
	}

	private JpaVendorAdapter createJpaVendorAdapter(JpaProperties jpaProperties) {
		AbstractJpaVendorAdapter adapter = new HibernateJpaVendorAdapter();
		adapter.setShowSql(jpaProperties.isShowSql());
		adapter.setDatabase(jpaProperties.getDatabase());
		adapter.setDatabasePlatform(jpaProperties.getDatabasePlatform());
		adapter.setGenerateDdl(jpaProperties.isGenerateDdl());
		adapter.setDatabase(Database.POSTGRESQL);
		return adapter;
	}
	
}