Custom Slick Code Generator 3.0.0

Question

Может скользкий кодоген генерировать все сопоставленные классы дел за пределами ${container} traitчтобы они не наследовали его тип? Может быть, в другом файле в целом, т.е. Models.scala?

// SuppliersRowsDAA.scala
import persistence.Tables

object SuppliersRowsDAA {
  case class Save(sup: Tables.SuppliersRow)
}   

Я получаю эту ошибку компиляции:

[error] /app/src/main/scala/persistence/dal/SuppliersDAA.scala:5: type mismatch;
[error]  found   : persistence.Tables.SuppliersRow
[error]  required: SuppliersDAA.this.SuppliersRow
[error]     case Save(sup) ⇒ sender ! db.run(Suppliers += sup)

С помощью Tables#SuppliersRow импорт выдает ту же ошибку.

Если я вручную вырезать и вставить SuppliersRow класс case вне автоматически сгенерированного trait Tables оно работает!

....
trait Tables {
....
}
case class SuppliersRow(id: Int, userId: Int, name: String)
//EOF

Есть идеи?

Answer 1

Добавление оригинала docWithCode из EntityType в super.packageCode():

import slick.codegen.SourceCodeGenerator
import slick.model.Model
import scala.collection.mutable

class CustomizedCodeGenerator(model: Model) extends SourceCodeGenerator(model) {
  val models = new mutable.MutableList[String]

  override def packageCode(profile: String, pkg: String, container: String, parentType: Option[String]): String = {
    super.packageCode(profile, pkg, container, parentType) + "\n" + outsideCode
  }

  def outsideCode = s"${indent(models.mkString("\n"))}"

  override def Table = new Table(_) {
    override def EntityType = new EntityTypeDef {
      override def docWithCode: String = {
        models += super.docWithCode.toString + "\n"
        ""
      }
    }
  }

}