diff options
Diffstat (limited to 'src/main')
-rw-r--r-- | src/main/scala/Generators.scala | 11 |
1 files changed, 8 insertions, 3 deletions
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) |