units of work can now be chained

This commit is contained in:
Athou
2014-08-13 10:01:38 +02:00
parent 1319bf4a8c
commit 9320b6beb8

View File

@@ -16,24 +16,31 @@ public abstract class UnitOfWork<T> {
protected abstract T runInSession() throws Exception; protected abstract T runInSession() throws Exception;
public T run() { public T run() {
final Session session = sessionFactory.openSession(); boolean newSession = !ManagedSessionContext.hasBind(sessionFactory);
if (ManagedSessionContext.hasBind(sessionFactory)) {
throw new IllegalStateException("Already in a unit of work!"); final Session session = newSession ? sessionFactory.openSession() : sessionFactory.getCurrentSession();
}
T t = null; T t = null;
try { try {
ManagedSessionContext.bind(session); if (newSession) {
session.beginTransaction(); ManagedSessionContext.bind(session);
session.beginTransaction();
}
try { try {
t = runInSession(); t = runInSession();
commitTransaction(session); if (newSession) {
commitTransaction(session);
}
} catch (Exception e) { } catch (Exception e) {
rollbackTransaction(session); if (newSession) {
rollbackTransaction(session);
}
this.<RuntimeException> rethrow(e); this.<RuntimeException> rethrow(e);
} }
} finally { } finally {
session.close(); if (newSession) {
ManagedSessionContext.unbind(sessionFactory); session.close();
ManagedSessionContext.unbind(sessionFactory);
}
} }
return t; return t;
} }