diff options
Diffstat (limited to 'src/main/scala/Generators.scala')
-rw-r--r-- | src/main/scala/Generators.scala | 21 |
1 files changed, 21 insertions, 0 deletions
diff --git a/src/main/scala/Generators.scala b/src/main/scala/Generators.scala new file mode 100644 index 0000000..345baa4 --- /dev/null +++ b/src/main/scala/Generators.scala @@ -0,0 +1,21 @@ +import slick.codegen.SourceCodeGenerator +import slick.{model => m} + +class RowSourceCodeGenerator( + model: m.Model, + override val headerComment: String, + override val imports: String, + override val schemaName: String +) extends SourceCodeGenerator(model) with RowOutputHelpers { + + override def Table = new Table(_) { table => + override def EntityType = new EntityType { + override def code: String = + (if (classEnabled) "final " else "") + super.code + } + + override def code = Seq[Def](EntityType).map(_.docWithCode) + } + + override def code = tables.map(_.code.mkString("\n")).mkString("\n\n") +} |