From e57e93f1c36adca193f682107c1b465c32253312 Mon Sep 17 00:00:00 2001
From: vlad <vlad@drivergrp.com>
Date: Wed, 9 Nov 2016 13:51:44 -0800
Subject: Database standard table creation method

---
 src/main/scala/xyz/driver/core/database.scala | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/src/main/scala/xyz/driver/core/database.scala b/src/main/scala/xyz/driver/core/database.scala
index 45d1b6d..0ebcd0d 100644
--- a/src/main/scala/xyz/driver/core/database.scala
+++ b/src/main/scala/xyz/driver/core/database.scala
@@ -1,10 +1,12 @@
 package xyz.driver.core
 
 import slick.backend.DatabaseConfig
+import slick.dbio.{DBIOAction, NoStream}
 import slick.driver.JdbcProfile
 import xyz.driver.core.time.Time
 
-import scala.concurrent.Future
+import scala.concurrent.{ExecutionContext, Future}
+import scala.language.reflectiveCalls
 
 object database {
 
@@ -25,6 +27,11 @@ object database {
     }
   }
 
+  type Schema = {
+    def create: DBIOAction[Unit, NoStream, slick.dbio.Effect.Schema]
+    def drop: DBIOAction[Unit, NoStream, slick.dbio.Effect.Schema]
+  }
+
   trait IdColumnTypes {
     val database: Database
 
@@ -41,8 +48,20 @@ object database {
 
   trait DatabaseObject extends IdColumnTypes {
 
+    implicit val exec: ExecutionContext
+
     def createTables(): Future[Unit]
     def disconnect(): Unit
+
+    def ensureTableExist(schemas: Seq[Schema]): Future[Unit] =
+      for {
+        dropping <- Future.sequence(schemas.map { schema =>
+                     database.database.run(schema.drop).recover { case _: Throwable => () }
+                   })
+        creation <- Future.sequence(schemas.map { schema =>
+                     database.database.run(schema.create).recover { case _: Throwable => () }
+                   })
+      } yield ()
   }
 
   abstract class DatabaseObjectAdapter extends DatabaseObject {
-- 
cgit v1.2.3