使用 Slick 3 验证架构

Question

我希望使用 Scala 应用程序的 Slick 3 框架来管理数据库交互。我已经能够使用 Slick 自动生成必要的 table 对象，但是 我还想要一个集成测试来验证数据库中的模式是否与对象中的模式相匹配。 这是因为有时 tables 在没有提醒我的团队的情况下被更改，因此我们更愿意在集成测试而不是生产应用程序中捕获更改。

一种方法是简单地运行对测试运行中的每个 table 进行 select 查询。但是，我觉得应该有更直接的方法。此外，我不清楚如何系统地运行通过文件中定义的所有 table，除了手动将 table 对象附加到测试 [=27= 的某个序列]纳尔穿过。我注意到有一个模式字段，但它只能生成创建和删除语句。

如有任何帮助，我们将不胜感激。谢谢！

编辑：

这是我的解决方案，但我希望有一个更好的解决方案：

class TablesIT extends FunSuite with BeforeAndAfter with ScalaFutures {
  var db: Database = _
  before{ db = Database.forURL( /* personal details */ )}

  object ResultMap extends GetResult[Map[String,Any]] { //Object borrowed from 
    def apply(pr: PositionedResult) = {
      val rs = pr.rs // <- jdbc result set
      val md = rs.getMetaData
      val res = (1 to pr.numColumns).map{ i=> md.getColumnName(i) -> rs.getObject(i) }.toMap
      pr.nextRow // <- use Slick's advance method to avoid endless loop
      res
    }
  }

  def testTableHasCols[A <: Table[_]](table: slick.lifted.TableQuery[A]): Unit = {
    whenReady(db.run(table.take(1).result.headOption.asTry)) { case Success(t) => t match {
      case Some(r) => logTrace(r.toString)
      case None => logTrace("Empty table")
    }
    case Failure(ex) => fail("Query exception: " + ex.toString)
    }
  }

  def plainSqlSelect[A](query: String)(implicit gr: GetResult[A]): Future[Seq[A]] = {
    val stmt = sql"""#$query""".as[A]
    db.run(stmt)
  }


  def compareNumOfCols[A <: Table[_]](table: slick.lifted.TableQuery[A]) = {
    val tableName = table.baseTableRow.tableName
    val selectStar = whenReady(db.run(sql"""select * from #$tableName limit 1""".as(ResultMap).headOption)) {
      case Some(m) => m.size
      case None => 0
    }
    val model = whenReady(db.run(sql"""#${table.take(1).result.statements.head}""".as(ResultMap).headOption)) {
      case Some(m) => m.size
      case None => 0
    }
    assert(selectStar === model, "The number of columns do not match")
  }

  test("Test table1") {
    testTableHasCols(Table1)
    compareNumOfCols(Table1)
  }

// And on for each table
}

Answer 1

我最终设计了一个使用以下想法的更好的解决方案。它或多或少是一样的，不幸的是我仍然必须为每个 table 手动创建一个测试，但我认为该方法更清晰。但是请注意，由于信息模式，这仅适用于 PostgreSQL，但其他数据库系统有其他方法。

class TablesIT extends FunSuite with BeforeAndAfter with ScalaFutures {
  var db: Database = _
  before{ db = Database.forURL( /* personal details */ )}

  def testTableHasCols[A <: Table[_]](table: slick.lifted.TableQuery[A]): Unit = {
    whenReady(db.run(table.take(1).result.headOption.asTry)) { case Success(t) => t match {
      case Some(r) => logTrace(r.toString)
      case None => logTrace("Empty table")
    }
    case Failure(ex) => fail("Query exception: " + ex.toString)
    }
  }

  def compareNumOfCols[A <: Table[_]](table: slick.lifted.TableQuery[A]) = {
    val tableName = table.baseTableRow.tableName
    val selectStar = whenReady(db.run(sql"""select column_name from information_schema.columns where table_name='#$tableName'""".as[String])) {
      case m: Seq[String] => m.size
      case _ => 0
    }
    val model = table.baseTableRow.create_*.map(_.name).toSeq.size

    assert(selectStar === model, "The number of columns do not match")
  }

  test("Test table1") {
    testTableHasCols(Table1)
    compareNumOfCols(Table1)
  }

// And on for each table
}

使用 Slick 3 验证架构

Verify schema using Slick 3

scala

slick

typesafe

slick-3.0