Fixed transaction not getting closed

Fixes #16
This commit is contained in:
Leon Hofmeister 2020-09-22 16:58:24 +02:00
parent 9419ba2ee8
commit f43b8ecfa6
Signed by: delvh
GPG Key ID: 3DECE05F6D9A647C
1 changed files with 39 additions and 14 deletions

View File

@ -3,9 +3,7 @@ package envoy.server.data;
import java.time.Instant;
import java.util.List;
import javax.persistence.EntityManager;
import javax.persistence.EntityTransaction;
import javax.persistence.Persistence;
import javax.persistence.*;
import envoy.data.User.UserStatus;
import envoy.server.net.ConnectionManager;
@ -235,8 +233,8 @@ public final class PersistenceManager {
public void addContactBidirectional(long contactID1, long contactID2) {
// Get users by ID
Contact c1 = getContactByID(contactID1);
Contact c2 = getContactByID(contactID2);
final var c1 = getContactByID(contactID1);
final var c2 = getContactByID(contactID2);
// Add users to each others contact lists
c1.getContacts().add(c2);
@ -259,20 +257,47 @@ public final class PersistenceManager {
}
private void persist(Object obj) {
transaction.begin();
entityManager.persist(obj);
transaction.commit();
try {
transaction.begin();
entityManager.persist(obj);
transaction.commit();
// Last transaction threw an error resulting in the transaction not being closed
} catch (final IllegalStateException e) {
if (transaction.isActive()) {
transaction.rollback();
persist(obj);
}
}
}
private void merge(Object obj) {
transaction.begin();
entityManager.merge(obj);
transaction.commit();
try {
transaction.begin();
entityManager.merge(obj);
transaction.commit();
// Last transaction threw an error resulting in the transaction not being closed
} catch (final IllegalStateException e) {
if (transaction.isActive()) {
transaction.rollback();
merge(obj);
}
}
}
private void remove(Object obj) {
transaction.begin();
entityManager.remove(obj);
transaction.commit();
try {
transaction.begin();
entityManager.remove(obj);
transaction.commit();
// Last transaction threw an error resulting in the transaction not being closed
} catch (final IllegalStateException e) {
if (transaction.isActive()) {
transaction.rollback();
remove(obj);
}
}
}
}