aboutsummaryrefslogtreecommitdiff
path: root/src/main/scala/Generators.scala
diff options
context:
space:
mode:
authorStewart Stewart <stewinsalot@gmail.com>2017-02-24 08:50:52 -0500
committerStewart Stewart <stewinsalot@gmail.com>2017-02-24 08:50:52 -0500
commitbb4be989a15f250741ab771e7ed78f512e74632d (patch)
tree9f7989c2ac182c77bcf7ed7ad21f7690c41fff0e /src/main/scala/Generators.scala
parent42f846add5a7b82f64a09f526d8817cb12b8ca86 (diff)
downloadslick-codegen-plugin-bb4be989a15f250741ab771e7ed78f512e74632d.tar.gz
slick-codegen-plugin-bb4be989a15f250741ab771e7ed78f512e74632d.tar.bz2
slick-codegen-plugin-bb4be989a15f250741ab771e7ed78f512e74632d.zip
scalafmt
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)
}
}