Slick 写一个简单的建表函数

Question

Emm...我正在用 Play 2 试用 Slick。表创建过程变得非常令人沮丧，因为与其他 ORM（如 ebean）不同，Slick 不会检测数据库是否已创建，如果已经存在表存在，会报异常。我只是不想每次重启服务器时都删除和创建，所以我决定写一个小函数来帮助我：

  def databaseCreate(tables: Table*) = {
     for (table <- tables) {
       if (MTable.getTables(table.getClass.getName).list.isEmpty) table.ddl.create
     }
  }

它的作用是接收一些像这样的对象：

  object Tag extends Table [(Option[Int], String)]("Tags") {
    def id = column[Int]("TAG_ID", O.PrimaryKey, O.AutoInc)
    def tag_name = column[String]("TAG_NAME")

    def * = id.? ~ tag_name
  }

并使用scala.slick.jdbc.meta.MTable 中的MTable 方法来了解该表是否存在。然后我遇到了一个简单的 Java 反射问题。如果databaseCreate 方法采用字符串，那么我可以调用.ddl.create。所以我决定传入对象并使用 relfection：table.getClass.getName。唯一的问题是类型不匹配：（来自我的 IDE）

预期：MySQLDriver.simple.type#Table，实际：BlogData.Tag.type

BlogData 是我用来存储所有较小表对象的大对象。我该如何解决这个不匹配问题？使用一大堆asInstanceOf？这会使命令变得难以忍受又长又丑……

---

更正：

类型不匹配是一个误报，它来自 IDE，而不是来自类型安全控制台编译器。真正的问题是：

type Table takes type parameters
def databaseCreate(tables: Table*) = {
^
one error found

然后我按照建议更改了代码：

def databaseCreate(tables: Table[_]*)(implicit session: Session) = {
         for (table <- tables) {
           if (MTable.getTables(table.tableName).list.isEmpty) table.ddl.create
         }
      }

然后我得到了这个错误：

ambiguous implicit values: both value session of type slick.driver.MySQLDriver.simple.Session and method threadLocalSession in object Database of type => scala.slick.session.Session match expected type scala.slick.session.Session
if (MTable.getTables(table.tableName).list.isEmpty) table.ddl.create
^
one error found

我的导入在这里：

import play.api.GlobalSettings
import play.api.Application
import models.BlogData._
import scala.slick.driver.MySQLDriver.simple._
import Database.threadLocalSession
import play.api.Play.current
import scala.slick.jdbc.meta.MTable

Answer 1

在 Slick 2 中：为了完整起见，请参阅 this 和 that 相关线程。

我使用以下方法：

def createIfNotExists(tables: TableQuery[_ <: Table[_]]*)(implicit session: Session) {
  tables foreach {table => if(MTable.getTables(table.baseTableRow.tableName).list.isEmpty) table.ddl.create}
}

然后您可以使用隐式会话创建表：

db withSession {
  implicit session =>
    createIfNotExists(table1, table2, ..., tablen)
}

Answer 2

我假设您使用的是 SLICK v1，因为您有 Table 对象。我有 v1 的工作代码。此版本将在重新创建之前删除所有存在的表（但不会删除您不希望重新创建的任何表）。它还会在创建之前将 DDL 合并为一个，以使创建顺序正确：

def create() = database withSession {
  import scala.slick.jdbc.{StaticQuery => Q}
  val tables = Seq(TableA, TableB, TableC)
  def existingTables = MTable.getTables.list().map(_.name.name)
  tables.filter(t => existingTables.contains(t.tableName)).foreach{t =>
    Q.updateNA(s"drop table ${t.tableName} cascade").execute
  }
  val tablesToCreate = tables.filterNot(t => existingTables.contains(t.tableName))
  val ddl: Option[DDL] = tablesToCreate.foldLeft(None: Option[DDL]){(ddl, table) =>
    ddl match {
      case Some(d) => Some(d ++ table.ddl)
      case _ => Some(table.ddl)
    }
  }
  ddl.foreach{_.create}
}

Answer 3

不知道为什么您会收到此错误消息。我需要查看您的导入以及您调用databaseCreate 的位置，但问题是def databaseCreate(tables: Table*) 应该是def databaseCreate(tables: Table[_]*)，并且可能还需要第二个参数列表def databaseCreate(tables: Table[_]*)(implicit session: Session) = ...。

另外，您可以使用 table.tableName 代替 table.getClass.getName。

Answer 4

您应该将 asTry 与表 create DBIOAction 一起使用。它将检查是否存在表......如果不存在，DBIOAction 将用于创建表。

在主页DAO中

   val createHomePageTableAction: DBIOAction[Int, NoStream, Effect.Schema with Effect.Write] = homePageTable.schema.create >>(homePageTable+=Homepage("No Data","No Data",0l))

在 ExplorePageDAO 中

  val SoftwareTableCreateAction: FixedSqlAction[Unit, NoStream, Effect.Schema] = softwareTable.schema.create

在 createTablesController 中

package controllers

import javax.inject.{Inject, Singleton}
import play.api.mvc.{AbstractController, Action, AnyContent, ControllerComponents}
import services.dbServices.{ExplorePageDAO, HomePageDAO}

import scala.concurrent.ExecutionContext

@Singleton
class CreateTablesController @Inject()(cc: ControllerComponents)(implicit assetsFinder: AssetsFinder, executionContext: ExecutionContext)
  extends AbstractController(cc) {

  import services.dbServices.dbSandBox._
  def createTable: Action[AnyContent] = Action.async {

    dbAccess.run(HomePageDAO.createHomePageTableAction .asTry >> ExplorePageDAO.SoftwareTableCreateAction.asTry).map(_=> Ok("Table Created"))
  }


}