aboutsummaryrefslogtreecommitdiff
path: root/src/main/scala/Generators.scala
diff options
context:
space:
mode:
Diffstat (limited to 'src/main/scala/Generators.scala')
-rw-r--r--src/main/scala/Generators.scala34
1 files changed, 18 insertions, 16 deletions
diff --git a/src/main/scala/Generators.scala b/src/main/scala/Generators.scala
index 741fe6b..51e02ce 100644
--- a/src/main/scala/Generators.scala
+++ b/src/main/scala/Generators.scala
@@ -2,21 +2,22 @@ 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,
- fullDatabaseModel: m.Model,
- idType: Option[String],
- manualForeignKeys: Map[(String, String), (String, String)]
+ model: m.Model,
+ override val headerComment: String,
+ override val imports: String,
+ override val schemaName: String,
+ fullDatabaseModel: m.Model,
+ idType: Option[String],
+ manualForeignKeys: Map[(String, String), (String, String)]
) extends TypedIdSourceCodeGenerator(
- fullDatabaseModel,
- idType,
- manualForeignKeys
-) with RowOutputHelpers {
+ fullDatabaseModel,
+ idType,
+ manualForeignKeys
+ )
+ with RowOutputHelpers {
override def Table = new TypedIdTable(_) { table =>
- override def Column = new TypedIdColumn(_) { }
+ override def Column = new TypedIdColumn(_) {}
override def EntityType = new EntityType {
override def code: String =
(if (classEnabled) "final " else "") + super.code
@@ -39,9 +40,10 @@ class TableSourceCodeGenerator(
parentType: Option[String],
idType: Option[String],
typeReplacements: Map[String, String])
- extends TypedIdSourceCodeGenerator(fullDatabaseModel, idType, manualForeignKeys)
- with TableOutputHelpers
-{
+ extends TypedIdSourceCodeGenerator(fullDatabaseModel,
+ idType,
+ manualForeignKeys)
+ with TableOutputHelpers {
val defaultIdImplementation =
"""|final case class Id[T](v: Int)
@@ -94,7 +96,7 @@ class TableSourceCodeGenerator(
}
override def Column = new TypedIdColumn(_) {
- override def rawType: String = {
+ override def rawType: String = {
typeReplacements.getOrElse(model.tpe, super.rawType)
}
}