private void releaseUnclosedEntityManager(EntityManager em) {
if (em == null) {
return;
}
if (!em.isOpen()) {
em = null;
return;
}
if (JtaStatusHelper.isActive(TestingJtaPlatformImpl.INSTANCE.getTransactionManager())) {
log.warn("Cleaning up unfinished transaction");
try {
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().rollback();
} catch (SystemException ignored) {
}
}
try {
if (em.getTransaction().isActive()) {
em.getTransaction().rollback();
log.warn(
"You left an open transaction! Fix your test case. For now, we are closing it for you.");
}
} catch (IllegalStateException e) {
}
if (em.isOpen()) {
// as we open an EM beforeQuery the test runs, it will still be open if the test uses a custom
// EM.
// or, the person may have forgotten to close. So, do not raise a "fail", but log the fact.
em.close();
log.warn("The EntityManager is not closed. Closing it.");
}
}
@Test
public void testCurrentSession() throws Exception {
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
Session s = sessionFactory().getCurrentSession();
Session s2 = sessionFactory().getCurrentSession();
assertSame(s, s2);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
assertFalse(s.isOpen());
// TODO : would be nice to automate-test that the SF internal map actually gets cleaned up
// i verified that is does currently in my debugger...
}
@Test
public void testCMT() throws Exception {
sessionFactory().getStatistics().clear();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
Session s = openSession();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
assertFalse(s.isOpen());
assertEquals(sessionFactory().getStatistics().getFlushCount(), 0);
assertEquals(sessionFactory().getStatistics().getEntityInsertCount(), 0);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s = openSession();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().rollback();
assertFalse(s.isOpen());
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s = openSession();
Map item = new HashMap();
item.put("name", "The Item");
item.put("description", "The only item we have");
s.persist("Item", item);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
assertFalse(s.isOpen());
assertEquals(sessionFactory().getStatistics().getFlushCount(), 1);
assertEquals(sessionFactory().getStatistics().getEntityInsertCount(), 1);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s = openSession();
item = (Map) s.createQuery("from Item").uniqueResult();
assertNotNull(item);
s.delete(item);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
assertFalse(s.isOpen());
assertEquals(sessionFactory().getStatistics().getTransactionCount(), 4);
assertEquals(sessionFactory().getStatistics().getSuccessfulTransactionCount(), 3);
assertEquals(sessionFactory().getStatistics().getEntityDeleteCount(), 1);
assertEquals(sessionFactory().getStatistics().getEntityInsertCount(), 1);
assertEquals(sessionFactory().getStatistics().getSessionOpenCount(), 4);
assertEquals(sessionFactory().getStatistics().getSessionCloseCount(), 4);
assertEquals(sessionFactory().getStatistics().getQueryExecutionCount(), 1);
assertEquals(sessionFactory().getStatistics().getFlushCount(), 2);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s = openSession();
s.createQuery("delete from Item").executeUpdate();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
}
protected AuditReader getAuditReader() {
EntityManager entityManager = getOrCreateEntityManager();
SessionImplementor sessionImplementor = entityManager.unwrap(SessionImplementor.class);
if (sessionImplementor.getTransactionCoordinator().getTransactionCoordinatorBuilder().isJta()) {
if (!JtaStatusHelper.isActive(TestingJtaPlatformImpl.INSTANCE.getTransactionManager())) {
try {
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
} catch (Exception e) {
e.printStackTrace();
}
}
} else if (!entityManager.getTransaction().isActive()) {
entityManager.getTransaction().begin();
}
if (auditReader != null) {
return auditReader;
}
return auditReader = AuditReaderFactory.get(entityManager);
}
@Test
public void testCurrentSessionWithIterate() throws Exception {
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
Session s = openSession();
Map item1 = new HashMap();
item1.put("name", "Item - 1");
item1.put("description", "The first item");
s.persist("Item", item1);
Map item2 = new HashMap();
item2.put("name", "Item - 2");
item2.put("description", "The second item");
s.persist("Item", item2);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
// First, test iterating the partial iterator; iterate to past
// the first, but not the second, item
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s = sessionFactory().getCurrentSession();
Iterator itr = s.createQuery("from Item").iterate();
if (!itr.hasNext()) {
fail("No results in iterator");
}
itr.next();
if (!itr.hasNext()) {
fail("Only one result in iterator");
}
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
// Next, iterate the entire result
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s = sessionFactory().getCurrentSession();
itr = s.createQuery("from Item").iterate();
if (!itr.hasNext()) {
fail("No results in iterator");
}
while (itr.hasNext()) {
itr.next();
}
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s = openSession();
s.createQuery("delete from Item").executeUpdate();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
}
@Test
public void testConcurrent() throws Exception {
sessionFactory().getStatistics().clear();
assertEquals(0, sessionFactory().getStatistics().getUpdateTimestampsCacheHitCount());
assertEquals(0, sessionFactory().getStatistics().getUpdateTimestampsCachePutCount());
assertEquals(0, sessionFactory().getStatistics().getUpdateTimestampsCacheMissCount());
assertNotNull(sessionFactory().getEntityPersister("Item").getCacheAccessStrategy());
assertEquals(0, sessionFactory().getStatistics().getEntityLoadCount());
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
Session s = openSession();
Map foo = new HashMap();
foo.put("name", "Foo");
foo.put("description", "a big foo");
s.persist("Item", foo);
Map bar = new HashMap();
bar.put("name", "Bar");
bar.put("description", "a small bar");
s.persist("Item", bar);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
assertEquals(0, sessionFactory().getStatistics().getUpdateTimestampsCacheHitCount());
assertEquals(
2,
sessionFactory()
.getStatistics()
.getUpdateTimestampsCachePutCount()); // One preinvalidate & one invalidate
assertEquals(0, sessionFactory().getStatistics().getUpdateTimestampsCacheMissCount());
sessionFactory().getCache().evictEntityRegion("Item");
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
Session s1 = openSession();
foo = (Map) s1.get("Item", "Foo");
// foo.put("description", "a big red foo");
// s1.flush();
Transaction tx = TestingJtaPlatformImpl.INSTANCE.getTransactionManager().suspend();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
Session s2 = openSession();
foo = (Map) s2.get("Item", "Foo");
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().resume(tx);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
sessionFactory().getCache().evictEntityRegion("Item");
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s1 = openSession();
s1.createCriteria("Item").list();
// foo.put("description", "a big red foo");
// s1.flush();
tx = TestingJtaPlatformImpl.INSTANCE.getTransactionManager().suspend();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s2 = openSession();
s2.createCriteria("Item").list();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().resume(tx);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s2 = openSession();
s2.createCriteria("Item").list();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
assertEquals(7, sessionFactory().getStatistics().getEntityLoadCount());
assertEquals(0, sessionFactory().getStatistics().getEntityFetchCount());
assertEquals(3, sessionFactory().getStatistics().getQueryExecutionCount());
assertEquals(0, sessionFactory().getStatistics().getQueryCacheHitCount());
assertEquals(0, sessionFactory().getStatistics().getQueryCacheMissCount());
assertEquals(0, sessionFactory().getStatistics().getUpdateTimestampsCacheHitCount());
assertEquals(2, sessionFactory().getStatistics().getUpdateTimestampsCachePutCount());
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s = openSession();
s.createQuery("delete from Item").executeUpdate();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
}
@Test
public void testCurrentSessionWithScroll() throws Exception {
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
Session s = sessionFactory().getCurrentSession();
Map item1 = new HashMap();
item1.put("name", "Item - 1");
item1.put("description", "The first item");
s.persist("Item", item1);
Map item2 = new HashMap();
item2.put("name", "Item - 2");
item2.put("description", "The second item");
s.persist("Item", item2);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
// First, test partially scrolling the result with out closing
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s = sessionFactory().getCurrentSession();
ScrollableResults results = s.createQuery("from Item").scroll();
results.next();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
// Next, test partially scrolling the result with closing
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s = sessionFactory().getCurrentSession();
results = s.createQuery("from Item").scroll();
results.next();
results.close();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
// Next, scroll the entire result (w/o closing)
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s = sessionFactory().getCurrentSession();
results = s.createQuery("from Item").scroll();
while (results.next()) {
// do nothing
}
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
// Next, scroll the entire result (closing)
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s = sessionFactory().getCurrentSession();
results = s.createQuery("from Item").scroll();
while (results.next()) {
// do nothing
}
results.close();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s = sessionFactory().getCurrentSession();
s.createQuery("delete from Item").executeUpdate();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
}
@Test
@RequiresDialectFeature(
value = DialectChecks.DoesReadCommittedNotCauseWritersToBlockReadersCheck.class,
comment = "write locks block readers")
public void testConcurrentCachedDirtyQueries() throws Exception {
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
Session s = openSession();
Map foo = new HashMap();
foo.put("name", "Foo");
foo.put("description", "a big foo");
s.persist("Item", foo);
Map bar = new HashMap();
bar.put("name", "Bar");
bar.put("description", "a small bar");
s.persist("Item", bar);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
synchronized (this) {
wait(1000);
}
sessionFactory().getStatistics().clear();
cleanupCache(); // we need a clean 2L cache here.
// open a TX and suspend it
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
Session s4 = openSession();
Transaction tx4 = TestingJtaPlatformImpl.INSTANCE.getTransactionManager().suspend();
// open a new TX and execute a query, this would fill the query cache.
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
Session s1 = openSession();
List r1 =
s1.createCriteria("Item").addOrder(Order.asc("description")).setCacheable(true).list();
assertEquals(r1.size(), 2);
foo = (Map) r1.get(0);
// update data and make query cache stale, but TX is suspended
foo.put("description", "a big red foo");
s1.flush();
Transaction tx1 = TestingJtaPlatformImpl.INSTANCE.getTransactionManager().suspend();
// open a new TX and run query again
// this TX is committed afterQuery query
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
Session s2 = openSession();
List r2 =
s2.createCriteria("Item").addOrder(Order.asc("description")).setCacheable(true).list();
assertEquals(r2.size(), 2);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
assertEquals(0, sessionFactory().getStatistics().getSecondLevelCacheHitCount());
assertEquals(0, sessionFactory().getStatistics().getSecondLevelCacheMissCount());
assertEquals(4, sessionFactory().getStatistics().getEntityLoadCount());
assertEquals(0, sessionFactory().getStatistics().getEntityFetchCount());
assertEquals(2, sessionFactory().getStatistics().getQueryExecutionCount());
assertEquals(2, sessionFactory().getStatistics().getQueryCachePutCount());
assertEquals(0, sessionFactory().getStatistics().getQueryCacheHitCount());
assertEquals(2, sessionFactory().getStatistics().getQueryCacheMissCount());
// updateTimestampsCache put happens at two places
// 1. {@link org.hibernate.engine.spi.ActionQueue#registerCleanupActions} calls preinvalidate
// 2. {@link
// org.hibernate.engine.spi.ActionQueue.AfterTransactionCompletionProcessQueue#afterTransactionCompletion} calls invalidate
// but since the TX which the update action happened is not committed yet, so there should be
// only 1 updateTimestamps put.
assertEquals(1, sessionFactory().getStatistics().getUpdateTimestampsCachePutCount());
// updateTimestampsCache hit only happens when the query cache data's timestamp is newer
// than the timestamp of when update happens
// since there is only 1 update action
assertEquals(1, sessionFactory().getStatistics().getUpdateTimestampsCacheHitCount());
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().resume(tx1);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
// update action's TX committed, so, invalidate is called, put new timestamp into
// UpdateTimestampsCache
assertEquals(2, sessionFactory().getStatistics().getUpdateTimestampsCachePutCount());
// but no more query cache lookup here, so it should still 1
assertEquals(1, sessionFactory().getStatistics().getUpdateTimestampsCacheHitCount());
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
Session s3 = openSession();
s3.createCriteria("Item").addOrder(Order.asc("description")).setCacheable(true).list();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
assertEquals(0, sessionFactory().getStatistics().getSecondLevelCacheHitCount());
assertEquals(0, sessionFactory().getStatistics().getSecondLevelCacheMissCount());
assertEquals(6, sessionFactory().getStatistics().getEntityLoadCount());
assertEquals(0, sessionFactory().getStatistics().getEntityFetchCount());
assertEquals(3, sessionFactory().getStatistics().getQueryExecutionCount());
assertEquals(3, sessionFactory().getStatistics().getQueryCachePutCount());
assertEquals(0, sessionFactory().getStatistics().getQueryCacheHitCount());
assertEquals(3, sessionFactory().getStatistics().getQueryCacheMissCount());
// a new query cache hit and one more update timestamps cache hit, so should be 2
assertEquals(2, sessionFactory().getStatistics().getUpdateTimestampsCacheHitCount());
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().resume(tx4);
List r4 =
s4.createCriteria("Item").addOrder(Order.asc("description")).setCacheable(true).list();
assertEquals(r4.size(), 2);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
assertEquals(2, sessionFactory().getStatistics().getSecondLevelCacheHitCount());
assertEquals(0, sessionFactory().getStatistics().getSecondLevelCacheMissCount());
assertEquals(6, sessionFactory().getStatistics().getEntityLoadCount());
assertEquals(0, sessionFactory().getStatistics().getEntityFetchCount());
assertEquals(3, sessionFactory().getStatistics().getQueryExecutionCount());
assertEquals(3, sessionFactory().getStatistics().getQueryCachePutCount());
assertEquals(1, sessionFactory().getStatistics().getQueryCacheHitCount());
assertEquals(3, sessionFactory().getStatistics().getQueryCacheMissCount());
assertEquals(3, sessionFactory().getStatistics().getUpdateTimestampsCacheHitCount());
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s = openSession();
s.createQuery("delete from Item").executeUpdate();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
}
@Test
public void testConcurrentCachedQueries() throws Exception {
sessionFactory().getStatistics().clear();
cleanupCache();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
Session s = openSession();
Map foo = new HashMap();
foo.put("name", "Foo");
foo.put("description", "a big foo");
s.persist("Item", foo);
Map bar = new HashMap();
bar.put("name", "Bar");
bar.put("description", "a small bar");
s.persist("Item", bar);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
synchronized (this) {
wait(1000);
}
sessionFactory().getStatistics().clear();
sessionFactory().getCache().evictEntityRegion("Item");
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
Session s4 = openSession();
Transaction tx4 = TestingJtaPlatformImpl.INSTANCE.getTransactionManager().suspend();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
Session s1 = openSession();
List r1 =
s1.createCriteria("Item").addOrder(Order.asc("description")).setCacheable(true).list();
assertEquals(r1.size(), 2);
Transaction tx1 = TestingJtaPlatformImpl.INSTANCE.getTransactionManager().suspend();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
Session s2 = openSession();
List r2 =
s2.createCriteria("Item").addOrder(Order.asc("description")).setCacheable(true).list();
assertEquals(r2.size(), 2);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
assertEquals(sessionFactory().getStatistics().getSecondLevelCacheHitCount(), 2);
assertEquals(sessionFactory().getStatistics().getSecondLevelCacheMissCount(), 0);
assertEquals(sessionFactory().getStatistics().getEntityLoadCount(), 2);
assertEquals(sessionFactory().getStatistics().getEntityFetchCount(), 0);
assertEquals(sessionFactory().getStatistics().getQueryExecutionCount(), 1);
assertEquals(sessionFactory().getStatistics().getQueryCachePutCount(), 1);
assertEquals(sessionFactory().getStatistics().getQueryCacheHitCount(), 1);
assertEquals(sessionFactory().getStatistics().getQueryCacheMissCount(), 1);
assertEquals(sessionFactory().getStatistics().getUpdateTimestampsCacheHitCount(), 1);
assertEquals(sessionFactory().getStatistics().getUpdateTimestampsCachePutCount(), 0);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().resume(tx1);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
Session s3 = openSession();
s3.createCriteria("Item").addOrder(Order.asc("description")).setCacheable(true).list();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
assertEquals(sessionFactory().getStatistics().getSecondLevelCacheHitCount(), 4);
assertEquals(sessionFactory().getStatistics().getSecondLevelCacheMissCount(), 0);
assertEquals(sessionFactory().getStatistics().getEntityLoadCount(), 2);
assertEquals(sessionFactory().getStatistics().getEntityFetchCount(), 0);
assertEquals(sessionFactory().getStatistics().getQueryExecutionCount(), 1);
assertEquals(sessionFactory().getStatistics().getQueryCachePutCount(), 1);
assertEquals(sessionFactory().getStatistics().getQueryCacheHitCount(), 2);
assertEquals(sessionFactory().getStatistics().getQueryCacheMissCount(), 1);
assertEquals(sessionFactory().getStatistics().getUpdateTimestampsCacheHitCount(), 2);
assertEquals(sessionFactory().getStatistics().getUpdateTimestampsCachePutCount(), 0);
assertEquals(sessionFactory().getStatistics().getUpdateTimestampsCacheMissCount(), 0);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().resume(tx4);
List r4 =
s4.createCriteria("Item").addOrder(Order.asc("description")).setCacheable(true).list();
assertEquals(r4.size(), 2);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
assertEquals(sessionFactory().getStatistics().getSecondLevelCacheHitCount(), 6);
assertEquals(sessionFactory().getStatistics().getSecondLevelCacheMissCount(), 0);
assertEquals(sessionFactory().getStatistics().getEntityLoadCount(), 2);
assertEquals(sessionFactory().getStatistics().getEntityFetchCount(), 0);
assertEquals(sessionFactory().getStatistics().getQueryExecutionCount(), 1);
assertEquals(sessionFactory().getStatistics().getQueryCachePutCount(), 1);
assertEquals(sessionFactory().getStatistics().getQueryCacheHitCount(), 3);
assertEquals(sessionFactory().getStatistics().getQueryCacheMissCount(), 1);
assertEquals(sessionFactory().getStatistics().getUpdateTimestampsCacheHitCount(), 3);
assertEquals(sessionFactory().getStatistics().getUpdateTimestampsCachePutCount(), 0);
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
s = openSession();
s.createQuery("delete from Item").executeUpdate();
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
}
@Override
protected void done() throws Throwable {
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().commit();
}
@Override
protected void prepare() throws Throwable {
TestingJtaPlatformImpl.INSTANCE.getTransactionManager().begin();
}