用于Oracle的Slick 3.0自定义代码生成

Question

我正试图为Oracle生成流畅的3.0代码。DB用户指向两个具有同名表的架构，因此生成的代码具有重复的类。我想从以"STAGE“结尾的模式中筛选出表。

以下是代码：

object CodeGen2 extends App {
  //https://stackoverflow.com/questions/28285129/slick-code-generation-for-only-a-single-schema

  val slickDriver = "com.typesafe.slick.driver.oracle.OracleDriver"
  val jdbcDriver = "oracle.jdbc.OracleDriver"
  val url = "jdbc:oracle:thin:@dbhost:1521:dbsid"
  val user = "dbuser"
  val password = "dbpassword"
  val destDir = "src/main/scala"
  val destPackage = "com.mycompany.mypackage"

  import scala.concurrent.{ExecutionContext, Await, Future}
  import scala.concurrent.duration.Duration
  import slick.codegen.SourceCodeGenerator
  import scala.concurrent.ExecutionContext.Implicits.global
  import slick.jdbc.JdbcModelBuilder
  import slick.jdbc.meta.MTable
  import com.typesafe.slick.driver.oracle.OracleDriver
  import slick.jdbc.JdbcBackend.DatabaseFactoryDef


  println("Starting codegen...")
  val db = OracleDriver.simple.Database.forURL(url, user=user, password=password, driver=jdbcDriver)
  val filteredTables = OracleDriver.defaultTables.filter(
        (t: MTable) => !t.name.schema.get.endsWith("STAGE")
      )

  val modelAction = OracleDriver.createModel(filteredTables, true)
  println("Generating model...")
  val model = Await.result(db.run(modelAction), Duration.Inf)
  val codegen = new SourceCodeGenerator(model) {
    // for illustration
    val noStage = model.tables.filter { table => !table.name.schema.get.endsWith("STAGE") } 
    noStage.foreach { table => println(table.name.schema.get) }

  }
  println("Generating files...")
  codegen.writeToFile(
    slickDriver, destDir, destPackage, "Tables", "Tables.scala"
  )
//  slick.codegen.SourceCodeGenerator.main(
//    Array(slickDriver, jdbcDriver, url, destDir, destPackage, user, password)
//  )
  println("Finished codegen.")
}

我试着过滤defaultTables，但是签名是SeqMTable =>布尔值，所以我不知道如何处理。过滤传递给driver.createModel的表是正确的方法吗？我查看并尝试了其他带有重写(N)方法的代码，但似乎没有任何可行的方法。

name := "slick-test"
version := "1.0"
scalaVersion := "2.11.6"

libraryDependencies ++= Seq(
  "com.typesafe.slick" %% "slick" % "3.0.0",
  "org.slf4j" % "slf4j-nop" % "1.6.4",
  "com.typesafe.slick" %% "slick-extensions" % "3.0.0",
  "com.typesafe.slick" %% "slick-codegen" % "3.0.0"
)

resolvers += "Typesafe Releases" at "http://repo.typesafe.com/typesafe/maven-releases/"

谢谢。

Answer 1

这样试一试：

 val filteredTables = OracleDriver.defaultTables.map(seq => seq.filter(t => !t.name.schema.get.endsWith("STAGE")))
  val modelAction = OracleDriver.createModel(Option(filteredTables), true)

Answer 2

更新:在我上面发布之后，找到了一个解决Oracle代码生成问题的方法。它应该用光滑的3.2来固定。用这个SourceCodeGenerator替换上面的代码。请参阅https://github.com/slick/slick/issues/1000

val codegen = new SourceCodeGenerator(model) {
  // slated for fix in 3.2
  override def Table = new Table(_) { table =>
    // override generator responsible for columns
    override def Column = new Column(_) {
      override def defaultCode = v => {
        def raw(v: Any) = rawType match {
          case "String" => "\"" + v + "\""
          case "Long" => v + "L"
          case "Float" => v + "F"
          case "Char" => "'" + v + "'"
          case "scala.math.BigDecimal" => v.toString.trim + "d"
          case "Byte" | "Short" | "Int" | "Double" | "Boolean" => v.toString
        }
        v match {
          case Some(x) => s"Some(${raw(x)})"
          case None    => "None"
          case x       => raw(x)
        }
      }
    }
  }
}