org.hibernate.context.spi.CurrentTenantIdentifierResolver Java Examples

@Bean
public LocalContainerEntityManagerFactoryBean entityManagerFactory(DataSource dataSource,
    MultiTenantConnectionProvider multiTenantConnectionProviderImpl,
    CurrentTenantIdentifierResolver currentTenantIdentifierResolverImpl) {
  Map<String, Object> properties = new HashMap<>();
  properties.putAll(jpaProperties.getHibernateProperties(dataSource));
  properties.put(Environment.MULTI_TENANT, MultiTenancyStrategy.SCHEMA);
  properties.put(Environment.MULTI_TENANT_CONNECTION_PROVIDER, multiTenantConnectionProviderImpl);
  properties.put(Environment.MULTI_TENANT_IDENTIFIER_RESOLVER, currentTenantIdentifierResolverImpl);

  LocalContainerEntityManagerFactoryBean em = new LocalContainerEntityManagerFactoryBean();
  em.setDataSource(dataSource);
  em.setPackagesToScan("com.srai");
  em.setJpaVendorAdapter(jpaVendorAdapter());
  em.setJpaPropertyMap(properties);
  return em;
}
 

@Bean
public LocalContainerEntityManagerFactoryBean entityManagerFactory(DataSource dataSource, MultiTenantConnectionProvider multiTenantConnectionProvider,
                                                                   CurrentTenantIdentifierResolver tenantIdentifierResolver) {
    final LocalContainerEntityManagerFactoryBean em = new LocalContainerEntityManagerFactoryBean();
    em.setDataSource(dataSource);
    em.setPackagesToScan("com.mycompany.models");

    em.setJpaVendorAdapter(this.jpaVendorAdapter());

    final Map<String, Object> jpaProperties = new HashMap<>();
    jpaProperties.put(Environment.MULTI_TENANT, MultiTenancyStrategy.SCHEMA);
    jpaProperties.put(Environment.MULTI_TENANT_CONNECTION_PROVIDER, multiTenantConnectionProvider);
    jpaProperties.put(Environment.MULTI_TENANT_IDENTIFIER_RESOLVER, tenantIdentifierResolver);
    jpaProperties.put(Environment.FORMAT_SQL, true);

    em.setJpaPropertyMap(jpaProperties);
    return em;
}
 

/**
 * org.springframework.beans.factory.FactoryBean that creates a JPA
 * {@link javax.persistence.EntityManagerFactory} according to JPA's standard
 * container bootstrap contract. This is the most powerful way to set up a
 * shared JPA EntityManagerFactory in a Spring application context; the
 * EntityManagerFactory can then be passed to JPA-based DAOs via dependency
 * injection. Note that switching to a JNDI lookup or to a
 * {@link org.springframework.orm.jpa.LocalEntityManagerFactoryBean} definition
 * is just a matter of configuration!
 * 
 * @param multiTenantConnectionProvider
 * @param currentTenantIdentifierResolver
 * @return
 */
@Bean
public LocalContainerEntityManagerFactoryBean entityManagerFactoryBean(
        MultiTenantConnectionProvider multiTenantConnectionProvider,
        CurrentTenantIdentifierResolver currentTenantIdentifierResolver) {

    Map<String, Object> hibernateProps = new LinkedHashMap<>();
    hibernateProps.putAll(this.jpaProperties.getProperties());
    hibernateProps.put(Environment.MULTI_TENANT, MultiTenancyStrategy.DATABASE);
    hibernateProps.put(Environment.MULTI_TENANT_CONNECTION_PROVIDER, multiTenantConnectionProvider);
    hibernateProps.put(Environment.MULTI_TENANT_IDENTIFIER_RESOLVER, currentTenantIdentifierResolver);

    // No dataSource is set to resulting entityManagerFactoryBean
    LocalContainerEntityManagerFactoryBean result = new LocalContainerEntityManagerFactoryBean();
    result.setPackagesToScan(new String[] { Employee.class.getPackage().getName() });
    result.setJpaVendorAdapter(new HibernateJpaVendorAdapter());
    result.setJpaPropertyMap(hibernateProps);

    return result;
}
 

@Override
public CurrentTenantIdentifierResolver getCurrentTenantIdentifierResolver() {
	return getSessionFactoryOptions().getCurrentTenantIdentifierResolver();
}
 

public CurrentTenantIdentifierResolver getCurrentTenantIdentifierResolver() {
    return sessionFactoryImplementor.getCurrentTenantIdentifierResolver();
}
 

@Override
public SessionFactoryBuilder applyCurrentTenantIdentifierResolver(CurrentTenantIdentifierResolver resolver) {
	this.optionsBuilder.applyCurrentTenantIdentifierResolver( resolver );
	return this;
}
 

public void applyCurrentTenantIdentifierResolver(CurrentTenantIdentifierResolver resolver) {
	this.currentTenantIdentifierResolver = resolver;
}
 

@Override
public CurrentTenantIdentifierResolver getCurrentTenantIdentifierResolver() {
	return currentTenantIdentifierResolver;
}
 

@Override
public T applyCurrentTenantIdentifierResolver(CurrentTenantIdentifierResolver resolver) {
	delegate.applyCurrentTenantIdentifierResolver( resolver );
	return getThis();
}
 

@Override
public CurrentTenantIdentifierResolver getCurrentTenantIdentifierResolver() {
	return delegate.getCurrentTenantIdentifierResolver();
}
 

@Override
public CurrentTenantIdentifierResolver getCurrentTenantIdentifierResolver() {
	return delegate.getCurrentTenantIdentifierResolver();
}
 

public void setCurrentTenantIdentifierResolver(CurrentTenantIdentifierResolver currentTenantIdentifierResolver) {
	this.currentTenantIdentifierResolver = currentTenantIdentifierResolver;
}
 

public CurrentTenantIdentifierResolver getCurrentTenantIdentifierResolver() {
	return currentTenantIdentifierResolver;
}
 

/**
 * Overridden to reliably pass a {@link CurrentTenantIdentifierResolver} to the SessionFactory.
 * @since 4.3.2
 * @see AvailableSettings#MULTI_TENANT_IDENTIFIER_RESOLVER
 */
@Override
public void setCurrentTenantIdentifierResolver(CurrentTenantIdentifierResolver currentTenantIdentifierResolver) {
	getProperties().put(AvailableSettings.MULTI_TENANT_IDENTIFIER_RESOLVER, currentTenantIdentifierResolver);
	super.setCurrentTenantIdentifierResolver(currentTenantIdentifierResolver);
}
 

/**
 * Overridden to reliably pass a {@link CurrentTenantIdentifierResolver} to the SessionFactory.
 * @since 4.3.2
 * @see AvailableSettings#MULTI_TENANT_IDENTIFIER_RESOLVER
 */
@Override
public void setCurrentTenantIdentifierResolver(CurrentTenantIdentifierResolver currentTenantIdentifierResolver) {
	getProperties().put(AvailableSettings.MULTI_TENANT_IDENTIFIER_RESOLVER, currentTenantIdentifierResolver);
	super.setCurrentTenantIdentifierResolver(currentTenantIdentifierResolver);
}
 

/**
 * Overridden to reliably pass a {@link CurrentTenantIdentifierResolver} to the SessionFactory.
 * @since 4.3.2
 * @see AvailableSettings#MULTI_TENANT_IDENTIFIER_RESOLVER
 */
@Override
public void setCurrentTenantIdentifierResolver(CurrentTenantIdentifierResolver currentTenantIdentifierResolver) {
	getProperties().put(AvailableSettings.MULTI_TENANT_IDENTIFIER_RESOLVER, currentTenantIdentifierResolver);
	super.setCurrentTenantIdentifierResolver(currentTenantIdentifierResolver);
}
 

/**
 * @todo make a Service ?
 */
CurrentTenantIdentifierResolver getCurrentTenantIdentifierResolver();
 

/**
 * Set a {@link CurrentTenantIdentifierResolver} to be passed on to the SessionFactory.
 * @see LocalSessionFactoryBuilder#setCurrentTenantIdentifierResolver
 */
public void setCurrentTenantIdentifierResolver(CurrentTenantIdentifierResolver currentTenantIdentifierResolver) {
	this.currentTenantIdentifierResolver = currentTenantIdentifierResolver;
}
 

/**
 * Specifies a strategy for resolving the notion of a "current" tenant-identifier when using multi-tenancy
 * together with current sessions
 *
 * @param resolver The resolution strategy to use.
 *
 * @return {@code this}, for method chaining
 *
 * @see org.hibernate.cfg.AvailableSettings#MULTI_TENANT_IDENTIFIER_RESOLVER
 */
SessionFactoryBuilder applyCurrentTenantIdentifierResolver(CurrentTenantIdentifierResolver resolver);
 

/**
 * Set a {@link CurrentTenantIdentifierResolver} to be passed on to the SessionFactory.
 * @see LocalSessionFactoryBuilder#setCurrentTenantIdentifierResolver
 */
public void setCurrentTenantIdentifierResolver(CurrentTenantIdentifierResolver currentTenantIdentifierResolver) {
	this.currentTenantIdentifierResolver = currentTenantIdentifierResolver;
}
 

/**
 * Since this is a multi-tenant application, Hibernate requires that the current
 * tenant identifier is resolved for use with
 * {@link org.hibernate.context.spi.CurrentSessionContext} and
 * {@link org.hibernate.SessionFactory#getCurrentSession()}
 * 
 * @return
 */
@Bean
public CurrentTenantIdentifierResolver currentTenantIdentifierResolver() {
    return new CurrentTenantIdentifierResolverImpl();
}
 

/**
 * Set a {@link CurrentTenantIdentifierResolver} to be passed on to the SessionFactory.
 * @see LocalSessionFactoryBuilder#setCurrentTenantIdentifierResolver
 */
public void setCurrentTenantIdentifierResolver(CurrentTenantIdentifierResolver currentTenantIdentifierResolver) {
	this.currentTenantIdentifierResolver = currentTenantIdentifierResolver;
}
 

/**
 * Set a {@link CurrentTenantIdentifierResolver} to be passed on to the SessionFactory.
 * @see LocalSessionFactoryBuilder#setCurrentTenantIdentifierResolver
 */
public void setCurrentTenantIdentifierResolver(CurrentTenantIdentifierResolver currentTenantIdentifierResolver) {
	this.currentTenantIdentifierResolver = currentTenantIdentifierResolver;
}
 

CurrentTenantIdentifierResolver getCurrentTenantIdentifierResolver();