From 42f846add5a7b82f64a09f526d8817cb12b8ca86 Mon Sep 17 00:00:00 2001 From: Stewart Stewart Date: Fri, 24 Feb 2017 08:50:31 -0500 Subject: extend use table output helpers --- src/main/scala/Generators.scala | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) (limited to 'src/main') diff --git a/src/main/scala/Generators.scala b/src/main/scala/Generators.scala index 20314b5..741fe6b 100644 --- a/src/main/scala/Generators.scala +++ b/src/main/scala/Generators.scala @@ -29,14 +29,19 @@ class RowSourceCodeGenerator( } class TableSourceCodeGenerator( - pkg: String, - fullDatabaseModel: m.Model, schemaOnlyModel: m.Model, + override val headerComment: String, + override val imports: String, + override val schemaName: String, + fullDatabaseModel: m.Model, + pkg: String, manualForeignKeys: Map[(String, String), (String, String)], parentType: Option[String], idType: Option[String], typeReplacements: Map[String, String]) - extends TypedIdSourceCodeGenerator(fullDatabaseModel, idType, manualForeignKeys) { + extends TypedIdSourceCodeGenerator(fullDatabaseModel, idType, manualForeignKeys) + with TableOutputHelpers +{ val defaultIdImplementation = """|final case class Id[T](v: Int) -- cgit v1.2.3