diff --git a/server/src/main/java/envoy/server/data/PersistenceManager.java b/server/src/main/java/envoy/server/data/PersistenceManager.java index a9915e8..fbadbb9 100755 --- a/server/src/main/java/envoy/server/data/PersistenceManager.java +++ b/server/src/main/java/envoy/server/data/PersistenceManager.java @@ -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); + } + } } }