From f932919f6bc57f5d080dd42e17af83af0a6ea080 Mon Sep 17 00:00:00 2001 From: Stewart Stewart Date: Fri, 9 Sep 2016 15:01:29 -0400 Subject: take schemas as a parameter --- src/main/scala/NamespacedCodegen.scala | 12 ++++-------- 1 file changed, 4 insertions(+), 8 deletions(-) (limited to 'src/main/scala/NamespacedCodegen.scala') diff --git a/src/main/scala/NamespacedCodegen.scala b/src/main/scala/NamespacedCodegen.scala index ca9897e..b853e9a 100644 --- a/src/main/scala/NamespacedCodegen.scala +++ b/src/main/scala/NamespacedCodegen.scala @@ -22,11 +22,8 @@ import java.io.FileWriter // generator places the relevant generated classes into separate // objects--a "a" object, and a "b" object) object NamespacedCodegen { - def parseSchemaList(schemaList: String): Map[String, List[String]] = { - val sl = schemaList.split(",").filter(_.trim.nonEmpty) - val tables: List[String] = sl.filter(_.contains(".")).toList - val schemas: List[String] = sl.filter(s => !(s.contains("."))).toList - + def parseSchemaList(schemaTableNames: List[String]): Map[String, List[String]] = { + val (tables, schemas) = schemaTableNames.partition(_.contains(".")) val mappedSchemas = schemas.map(_ -> List()).toMap val mappedTables = tables.groupBy(_.split("\\.")(0)).map { case (key, value) => (key, value.map(_.split("\\.")(1)).asInstanceOf[List[String]]) @@ -83,12 +80,11 @@ object NamespacedCodegen { pkg: String, filename: String, typesFilename: String, - schemaList: String + schemaTableNames: List[String] ): Unit = { val dc = DatabaseConfig.forURI[JdbcProfile](uri) val slickDriver = if(dc.driverIsObject) dc.driverName else "new " + dc.driverName - - val mappedSchemas = parseSchemaList(schemaList) + val mappedSchemas = parseSchemaList(schemaTableNames) val dbModel = Await.result(dc.db.run(createFilteredModel(dc.driver, mappedSchemas)), Duration.Inf) //finally dc.db.close -- cgit v1.2.3