diff options
Diffstat (limited to 'src/main/scala/Generators.scala')
-rw-r--r-- | src/main/scala/Generators.scala | 14 |
1 files changed, 11 insertions, 3 deletions
diff --git a/src/main/scala/Generators.scala b/src/main/scala/Generators.scala index 345baa4..402f87f 100644 --- a/src/main/scala/Generators.scala +++ b/src/main/scala/Generators.scala @@ -5,10 +5,18 @@ class RowSourceCodeGenerator( model: m.Model, override val headerComment: String, override val imports: String, - override val schemaName: String -) extends SourceCodeGenerator(model) with RowOutputHelpers { + override val schemaName: String, + fullDatabaseModel: m.Model, + idType: Option[String], + manualForeignKeys: Map[(String, String), (String, String)] +) extends TypedIdSourceCodeGenerator( + fullDatabaseModel, + idType, + manualForeignKeys +) with RowOutputHelpers { - override def Table = new Table(_) { table => + override def Table = new TypedIdTable(_) { table => + override def Column = new TypedIdColumn(_) { } override def EntityType = new EntityType { override def code: String = (if (classEnabled) "final " else "") + super.code |