diff options
author | Stewart Stewart <stewinsalot@gmail.com> | 2017-02-15 14:38:35 -0800 |
---|---|---|
committer | Stewart Stewart <stewinsalot@gmail.com> | 2017-02-15 14:38:35 -0800 |
commit | f56b126119453e52d0434b186b6287bbe179501b (patch) | |
tree | 310704c4cbe64f2aacf7b5e1182955b03c4ee4e0 /src/main/scala | |
parent | 902be96c8d5b418ccbc2e620fef47f189bf64be2 (diff) | |
download | slick-codegen-plugin-f56b126119453e52d0434b186b6287bbe179501b.tar.gz slick-codegen-plugin-f56b126119453e52d0434b186b6287bbe179501b.tar.bz2 slick-codegen-plugin-f56b126119453e52d0434b186b6287bbe179501b.zip |
don't override generated code
Diffstat (limited to 'src/main/scala')
-rw-r--r-- | src/main/scala/NamespacedCodegen.scala | 58 |
1 files changed, 1 insertions, 57 deletions
diff --git a/src/main/scala/NamespacedCodegen.scala b/src/main/scala/NamespacedCodegen.scala index f7c9d7c..0bd4717 100644 --- a/src/main/scala/NamespacedCodegen.scala +++ b/src/main/scala/NamespacedCodegen.scala @@ -94,26 +94,8 @@ class PackageNameGenerator(pkg: String, dbModel: Model) class ImportGenerator(dbModel: Model, schemaImports: List[String]) extends SourceCodeGenerator(dbModel) { - - val baseImports: String = schemaImports - .map("import " + _) - .mkString("\n") + "\n" - - val hlistImports: String = - """|import slick.collection.heterogeneous._ - |import slick.collection.heterogeneous.syntax._ - |""".stripMargin - - val plainSqlMapperImports: String = - if (tables.exists(_.PlainSqlMapper.enabled)) - """|import slick.jdbc.{GetResult => GR} - |//NOTE: GetResult mappers for plain SQL are only generated for tables where Slick knows how to map the types of all columns.\n - |""".stripMargin - else "" - override def code: String = - baseImports + hlistImports + plainSqlMapperImports - // remove the latter two when go back to inherriting `def code` + schemaImports.map("import " + _).mkString("\n") + "\n" } class Generator(pkg: String, @@ -141,44 +123,6 @@ class Generator(pkg: String, |} |""".stripMargin - // override def tables = { - // dbModel.tables.map(Table).sortBy(_.TableClass.rawName.toLowerCase) - // .filter(_.model.name.schema.getOrElse("`public`") == schemaName) - // } - // Can't override with final - // Can't reference super with lazy - - override def code: String = { - val tableCode = tables.map(_.code.mkString("\n")).mkString("\n\n") - - val ddlCode = - (if (ddlEnabled) { - "\n/** DDL for all tables. Call .create to execute. */" + - ( - if (tables.length > 5) - "\nlazy val schema: profile.SchemaDescription = Array(" + tables - .map(_.TableValue.name + ".schema") - .mkString(", ") + ").reduceLeft(_ ++ _)" - else if (tables.nonEmpty) - "\nlazy val schema: profile.SchemaDescription = " + tables - .map(_.TableValue.name + ".schema") - .mkString(" ++ ") - else - "\nlazy val schema: profile.SchemaDescription = profile.DDL(Nil, Nil)" - ) + - "\n\n" - } else "") - - if (idType.isEmpty) { // TODO move out - writeStringToFile(packageName + defaultIdImplementation, - "", //outputPath, - pkg, - "Id.scala") - } - - tableCode + "\n " + ddlCode - } - override def packageCode(profile: String, pkg: String, container: String, |