From 80a0363548bf9ab7be4126ee357d3bba47f13c1e Mon Sep 17 00:00:00 2001 From: Stewart Stewart Date: Fri, 24 Feb 2017 09:39:11 -0500 Subject: use type replacements in row generator --- src/main/scala/Generators.scala | 11 +++++++++-- src/main/scala/NamespacedCodegen.scala | 3 ++- 2 files changed, 11 insertions(+), 3 deletions(-) (limited to 'src') 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( -- cgit v1.2.3