HHH-10964 : Fix missing currentTenantIdentifierResolver

1. Add missing sessionFactoryBuilder.applyCurrentTenantIdentifierResolver() .
2. Use getter instead of field for consistency .

(cherry picked from commit 5ea8997ca7)
This commit is contained in:
Yanming Zhou 2016-07-18 13:25:21 +08:00 committed by Gail Badner
parent 906341349b
commit ed2c353717
2 changed files with 58 additions and 4 deletions

View File

@ -698,11 +698,14 @@ public class Configuration {
if ( getSessionFactoryObserver() != null ) {
sessionFactoryBuilder.addSessionFactoryObservers( getSessionFactoryObserver() );
}
if ( entityNotFoundDelegate != null ) {
sessionFactoryBuilder.applyEntityNotFoundDelegate( entityNotFoundDelegate );
if ( getEntityNotFoundDelegate() != null ) {
sessionFactoryBuilder.applyEntityNotFoundDelegate( getEntityNotFoundDelegate() );
}
if ( entityTuplizerFactory != null ) {
sessionFactoryBuilder.applyEntityTuplizerFactory( entityTuplizerFactory );
if ( getEntityTuplizerFactory() != null ) {
sessionFactoryBuilder.applyEntityTuplizerFactory( getEntityTuplizerFactory() );
}
if ( getCurrentTenantIdentifierResolver() != null ) {
sessionFactoryBuilder.applyCurrentTenantIdentifierResolver( getCurrentTenantIdentifierResolver() );
}
return sessionFactoryBuilder.build();

View File

@ -0,0 +1,51 @@
/*
* Hibernate, Relational Persistence for Idiomatic Java
*
* License: GNU Lesser General Public License (LGPL), version 2.1 or later.
* See the lgpl.txt file in the root directory or <http://www.gnu.org/licenses/lgpl-2.1.html>.
*/
package org.hibernate.test.multitenancy.schema;
import org.hibernate.cfg.Configuration;
import org.hibernate.context.spi.CurrentTenantIdentifierResolver;
import org.hibernate.testing.TestForIssue;
import org.hibernate.testing.junit4.BaseCoreFunctionalTestCase;
import org.junit.Test;
import static org.junit.Assert.assertSame;
/**
* @author Vlad Mihalcea
*/
@TestForIssue(jiraKey = "HHH-10964")
public class TenantResolverConfigurationTest extends BaseCoreFunctionalTestCase {
private TestCurrentTenantIdentifierResolver currentTenantResolver = new TestCurrentTenantIdentifierResolver();
@Override
protected Configuration constructAndConfigureConfiguration() {
Configuration configuration = super.constructAndConfigureConfiguration();
configuration.setCurrentTenantIdentifierResolver( currentTenantResolver );
return configuration;
}
@Test
public void testConfiguration() throws Exception {
assertSame(currentTenantResolver, sessionFactory().getCurrentTenantIdentifierResolver());
}
private static class TestCurrentTenantIdentifierResolver implements CurrentTenantIdentifierResolver {
private String currentTenantIdentifier;
@Override
public boolean validateExistingCurrentSessions() {
return false;
}
@Override
public String resolveCurrentTenantIdentifier() {
return currentTenantIdentifier;
}
}
}