diff options
| author | Stewart Stewart <stewinsalot@gmail.com> | 2017-02-24 09:39:11 -0500 |
|---|---|---|
| committer | Stewart Stewart <stewinsalot@gmail.com> | 2017-02-24 10:06:10 -0500 |
| commit | 80a0363548bf9ab7be4126ee357d3bba47f13c1e (patch) | |
| tree | 91420b4aa5fca91682a27d0abbb0ed55e62a3d87 | |
| parent | 73eb214f539913acf8af8e06a37cb3c0328cf891 (diff) | |
| download | slick-codegen-plugin-80a0363548bf9ab7be4126ee357d3bba47f13c1e.tar.gz slick-codegen-plugin-80a0363548bf9ab7be4126ee357d3bba47f13c1e.tar.bz2 slick-codegen-plugin-80a0363548bf9ab7be4126ee357d3bba47f13c1e.zip | |
use type replacements in row generator
| -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( |