Slick 3.0针对Oracle的自定义代码生成

我正在尝试为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，但是签名是Seq[MTable]=>Boolean，所以我不知道如何处理它。过滤传递给driver.createModel的表是否是正确的方法？我查看并尝试了其他带有override（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/"

谢谢。

这样试试：

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

试着这样做：

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

---

更新：在我发布上述内容后，找到了解决Oracle代码生成问题的方法。它应该固定在光滑的3.2中。用此SourceCodeGenerator替换上述代码。看

---

更新：在我发布上述内容后，找到了解决Oracle代码生成问题的方法。它应该固定在光滑的3.2中。用此SourceCodeGenerator替换上述代码。看

---

谢谢@bhavya，这完全符合您发布的内容。生成的代码现在无法编译，但这是一个单独的问题。谢谢@bhavya，这与您发布的完全一样。生成的代码现在不编译，但这是一个单独的问题。