Code: Select all
@Component
public class HibernateListenerConfig {
@PersistenceUnit(unitName = "master")
private EntityManagerFactory entityManagerFactory;
@Autowired
private AbcDomainListener abcDomainListener;
@PostConstruct
protected void init() {
SessionFactoryImpl sessionFactory = entityManagerFactory.unwrap(SessionFactoryImpl.class);
EventListenerRegistry registry = sessionFactory.getServiceRegistry().getService(EventListenerRegistry.class);
registry.getEventListenerGroup(EventType.PRE_LOAD).appendListener(abcDomainListener);
}
}
public class AbcDomainListener extends Serializable {
public void onPreLoad(PreLoadEvent preLoadEvent) {
//DO SOMETHING
}
}
@Configuration
@EnableJpaRepositories(basePackages = "com.projectName.module",
includeFilters = {@ComponentScan.Filter(type = FilterType.ANNOTATION, value = Repository.class)}
)
@EnableTransactionManagement
public class DatabaseConfiguration implements EnvironmentAware {
.
.
.
public DataSource dataSourceMultiTenant() {
Map targetDataSourceMap = new HashMap();
HikariDataSource masterTenantDatasource = masterTenantDatasource();
targetDataSourceMap.put("DB_MASTER_TENANT",
masterTenantDatasource);
HikariDataSource dbTenant1Datasource = dbTenant1Datasource();
targetDataSourceMap.put("DB_TENANT_1",
dbTenant1Datasource);
DataSourceRouter clientRoutingDatasource
= new DataSourceRouter();
clientRoutingDatasource.setTargetDataSources(targetDataSourceMap);
clientRoutingDatasource.setDefaultTargetDataSource(masterTenantDatasource);
return clientRoutingDatasource;
}
@Primary
@Bean(name = "entityManagerFactory")
@Qualifier(value = "entityManagerFactory")
public LocalContainerEntityManagerFactoryBean entityManagerFactory(
EntityManagerFactoryBuilder builder, @Qualifier("dataSource") DataSource dataSource) {
return builder.dataSource(dataSource).packages("com.projectName.module").persistenceUnit("master")
.build();
}
.
.
.
}
public class DataSourceRouter extends AbstractRoutingDataSource {
@Override
protected Object determineCurrentLookupKey() {
return TenantService.getClientDatabase();
}
}
Mobile version