diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/main/scala/Generators.scala | 11 | ||||
-rw-r--r-- | src/main/scala/NamespacedCodegen.scala | 3 |
2 files changed, 11 insertions, 3 deletions
diff --git a/src/main/scala/Generators.scala b/src/main/scala/Generators.scala index 59c9d9f..b3bfeab 100644 --- a/src/main/scala/Generators.scala +++ b/src/main/scala/Generators.scala @@ -8,7 +8,8 @@ class RowSourceCodeGenerator( override val schemaName: String, fullDatabaseModel: m.Model, idType: Option[String], - manualForeignKeys: Map[(String, String), (String, String)] + manualForeignKeys: Map[(String, String), (String, String)], + typeReplacements: Map[String, String] ) extends TypedIdSourceCodeGenerator( model, fullDatabaseModel, @@ -18,7 +19,13 @@ class RowSourceCodeGenerator( with RowOutputHelpers { override def Table = new TypedIdTable(_) { table => - override def Column = new TypedIdColumn(_) {} + + override def Column = new TypedIdColumn(_) { + override def rawType: String = { + typeReplacements.getOrElse(model.tpe, super.rawType) + } + } + override def EntityType = new EntityType { override def code: String = (if (classEnabled) "final " else "") + super.code diff --git a/src/main/scala/NamespacedCodegen.scala b/src/main/scala/NamespacedCodegen.scala index c3bf89c..7ae4ac0 100644 --- a/src/main/scala/NamespacedCodegen.scala +++ b/src/main/scala/NamespacedCodegen.scala @@ -69,7 +69,8 @@ object Generator { schemaName = schemaName, dbModel, idType, - manualForeignKeys + manualForeignKeys, + typeReplacements ) val tableGenerator = new TableSourceCodeGenerator( |