1 files changed, 53 insertions, 0 deletions
diff --git a/vid-app-common/src/main/java/org/onap/vid/utils/DaoUtils.java b/vid-app-common/src/main/java/org/onap/vid/utils/DaoUtils.java
new file mode 100644
index 000000000..c8a91626b
--- /dev/null
+++ b/vid-app-common/src/main/java/org/onap/vid/utils/DaoUtils.java
@@ -0,0 +1,53 @@
+package org.onap.vid.utils;
+
+import org.hibernate.Session;
+import org.hibernate.SessionFactory;
+import org.hibernate.Transaction;
+import org.onap.vid.exceptions.GenericUncheckedException;
+import org.onap.portalsdk.core.FusionObject;
+
+import java.util.HashMap;
+import java.util.function.Function;
+
+public class DaoUtils {
+
+    //all credit for this wonderful method go to is9613
+    public static<T> T tryWithSessionAndTransaction(SessionFactory sessionFactory, Function<Session, T> update) {
+        // opens a session and transactions, executes the input query.
+        // gracefully close session and transaction once error occurres.
+        Session session = null;
+        Transaction tx = null;
+        try {
+            session = sessionFactory.openSession();
+            tx = session.beginTransaction();
+
+            T res = update.apply(session);
+
+            tx.commit();
+
+            return res;
+        } catch (RuntimeException e) {
+            try {
+                if (tx != null) {
+                    tx.rollback();
+                }
+            } catch (RuntimeException e2) {
+                // e2 is ingnored; we would like to know the
+                // original failure reason, not only the reason
+                // for rollback's failure
+                throw new GenericUncheckedException("Failed rolling back transaction", e);
+            }
+            throw new GenericUncheckedException("Rolled back transaction", e);
+        } finally {
+            if (session != null) {
+                session.close();
+            }
+        }
+    }
+
+    public static HashMap<String, Object> getPropsMap() {
+        HashMap<String, Object> props = new HashMap<>();
+        props.put(FusionObject.Parameters.PARAM_USERID, 0);
+        return props;
+    }
+}