HHH-11740 - Default MultiTableBulkIdStrategy for DB2 does not work with connection pools

- Made test run on H2, Oracle, SQL server, MySQL and PostgreSQL
This commit is contained in:
Vlad Mihalcea 2017-05-15 18:50:31 +03:00
parent 386b04872f
commit 733f55f362
2 changed files with 53 additions and 9 deletions

View File

@ -7,6 +7,7 @@
package org.hibernate.test.unionsubclass;
import java.util.List;
import java.util.concurrent.atomic.AtomicBoolean;
import org.junit.Test;
@ -15,12 +16,18 @@ import org.hibernate.Hibernate;
import org.hibernate.Query;
import org.hibernate.Session;
import org.hibernate.Transaction;
import org.hibernate.cfg.AvailableSettings;
import org.hibernate.cfg.Configuration;
import org.hibernate.criterion.Order;
import org.hibernate.dialect.H2Dialect;
import org.hibernate.dialect.SQLServerDialect;
import org.hibernate.testing.SkipForDialect;
import org.hibernate.testing.TestForIssue;
import org.hibernate.testing.jdbc.SQLServerSnapshotIsolationConnectionProvider;
import org.hibernate.testing.junit4.BaseCoreFunctionalTestCase;
import org.hibernate.testing.transaction.TransactionUtil;
import org.hibernate.testing.util.ExceptionUtil;
import static org.hibernate.testing.transaction.TransactionUtil.doInHibernate;
import static org.junit.Assert.assertEquals;
@ -28,12 +35,32 @@ import static org.junit.Assert.assertFalse;
import static org.junit.Assert.assertNotNull;
import static org.junit.Assert.assertSame;
import static org.junit.Assert.assertTrue;
import static org.junit.Assert.fail;
/**
* @author Gavin King
*/
@SuppressWarnings("unchecked")
public class UnionSubclassTest extends BaseCoreFunctionalTestCase {
private SQLServerSnapshotIsolationConnectionProvider connectionProvider =
new SQLServerSnapshotIsolationConnectionProvider();
@Override
public void configure(Configuration cfg) {
super.configure( cfg );
if( SQLServerDialect.class.isAssignableFrom( DIALECT.getClass() )) {
cfg.getProperties().put( AvailableSettings.CONNECTION_PROVIDER, connectionProvider );
}
}
@Override
protected void releaseSessionFactory() {
super.releaseSessionFactory();
connectionProvider.stop();
}
@Override
public String[] getMappings() {
return new String[] { "unionsubclass/Beings.hbm.xml" };
@ -410,10 +437,6 @@ public class UnionSubclassTest extends BaseCoreFunctionalTestCase {
@Test
@TestForIssue( jiraKey = "HHH-11740" )
@SkipForDialect(
value = H2Dialect.class,
comment = "skip dialects that throw PessimisticLockException"
)
public void testBulkOperationsInTwoConcurrentSessions() throws Exception {
doInHibernate( this::sessionFactory, s -> {
Location mars = new Location( "Mars" );
@ -443,21 +466,35 @@ public class UnionSubclassTest extends BaseCoreFunctionalTestCase {
s.persist( human );
} );
AtomicBoolean pessimisticLocking = new AtomicBoolean( false );
doInHibernate( this::sessionFactory, s1 -> {
TransactionUtil.setJdbcTimeout( s1 );
assertEquals(
1,
s1.createQuery( "update Being set identity = 'John Doe' where identity = 'Jane Doe'" )
.executeUpdate()
);
doInHibernate( this::sessionFactory, s2 -> {
assertEquals( 1, s2.createQuery( "delete from Being where species = 'Martian'" ).executeUpdate() );
} );
try {
doInHibernate( this::sessionFactory, s2 -> {
TransactionUtil.setJdbcTimeout( s2 );
assertEquals( 1, s2.createQuery( "delete from Being where species = 'Martian'" ).executeUpdate() );
} );
}
catch (Exception e) {
if ( !ExceptionUtil.isSqlLockTimeout(e) ) {
fail( e.getMessage() );
}
pessimisticLocking.set( true );
}
} );
doInHibernate( this::sessionFactory, s -> {
Human human = (Human) s.createQuery( "from Being" ).uniqueResult();
assertEquals( "John Doe", human.getIdentity() );
if ( !pessimisticLocking.get() ) {
Human human = (Human) s.createQuery( "from Being" ).uniqueResult();
assertEquals( "John Doe", human.getIdentity() );
}
s.createQuery( "delete from Being" ).executeUpdate();
s.createQuery( "delete from Hive" ).executeUpdate();
s.createQuery( "delete from Location" ).executeUpdate();

View File

@ -28,6 +28,7 @@ import org.hibernate.dialect.Dialect;
import org.hibernate.dialect.H2Dialect;
import org.hibernate.dialect.MySQLDialect;
import org.hibernate.dialect.PostgreSQL81Dialect;
import org.hibernate.dialect.SQLServerDialect;
import org.jboss.logging.Logger;
@ -481,6 +482,12 @@ public class TransactionUtil {
st.execute();
}
}
else if( Dialect.getDialect() instanceof SQLServerDialect ) {
try (Statement st = connection.createStatement()) {
//Prepared Statements fail for SET commands
st.execute(String.format( "SET LOCK_TIMEOUT %d", millis / 10));
}
}
else {
try {
connection.setNetworkTimeout( Executors.newSingleThreadExecutor(), (int) millis );