Initial extraction of Driver non-specific utilities

1 files changed, 23 insertions, 0 deletions

diff --git a/src/main/scala/xyz/driver/common/db/Transactions.scala b/src/main/scala/xyz/driver/common/db/Transactions.scala
new file mode 100644
index 0000000..2f5a2cc
--- /dev/null
+++ b/src/main/scala/xyz/driver/common/db/Transactions.scala
@@ -0,0 +1,23 @@
+package xyz.driver.common.db
+
+import xyz.driver.common.logging.PhiLogging
+
+import scala.concurrent.Future
+import scala.util.{Failure, Success, Try}
+
+class Transactions()(implicit context: SqlContext) extends PhiLogging {
+  def run[T](f: SqlContext => T): Future[T] = {
+    import context.executionContext
+
+    Future(context.transaction(f(context))).andThen {
+      case Failure(e) => logger.error(phi"Can't run a transaction: $e")
+    }
+  }
+
+  def runSync[T](f: SqlContext => T): Unit = {
+    Try(context.transaction(f(context))) match {
+      case Success(_) =>
+      case Failure(e) => logger.error(phi"Can't run a transaction: $e")
+    }
+  }
+}