Extract getTables.

lastland · Feb 28, 2016 · 84e32c1 · 84e32c1
1 parent 72adb24
commit 84e32c1
Showing 1 changed file with 9 additions and 6 deletions.
diff --git a/migrations/slick/src/main/scala/Codegen.scala b/migrations/slick/src/main/scala/Codegen.scala
@@ -7,6 +7,7 @@ import scala.concurrent.Future
 import scala.concurrent.Await
 import scala.concurrent.ExecutionContext.Implicits.global
 import slick.model.Model
+import slick.driver.JdbcProfile
 import slick.codegen.SourceCodeGenerator
 
 trait SlickCodegen {
@@ -22,6 +23,13 @@ trait SlickCodegen {
 
   def tableNames: Seq[String] = List()
 
+  def getTables(driver: JdbcProfile) = driver.createModel(Some(
+    driver.defaultTables.map { s =>
+      s.filter { t =>
+        tableNames.contains(t.name.name)
+      }
+    }))
+
   def getGenerator(m: Model, version: Int) = {
     new SourceCodeGenerator(m) {
       override def packageCode(
@@ -42,12 +50,7 @@ object Version{
       return
     }
     val driver = mm.dbConfig.driver
-    val action = driver.createModel(Some(
-      driver.defaultTables.map { s =>
-        s.filter { t =>
-          tableNames.contains(t.name.name)
-        }
-      })) flatMap { case m =>
+    val action = getTables(driver).flatMap { case m =>
         DBIO.from {
           Future {
             val latest = mm.latest