Scala 扩展 SLICK 表格的干燥方式

在本文中，我们将介绍如何在 Scala 中以干燥（Don’t Repeat Yourself，DRY）的方式扩展 SLICK 数据库框架中的表格。SLICK 是一个强大的异步数据库访问库，但是在创建和扩展表格时，重复的代码可能会让我们感到繁琐和冗余。为了解决这个问题，我们将使用 Scala 的继承机制来减少重复的代码，并提高代码的可读性和可维护性。

阅读更多：Scala 教程

使用继承扩展 SLICK 表格

在扩展 SLICK 表格时，我们可以定义一个基础表格类，并在需要的时候继承它来创建特定的表格。这样可以避免重复定义相同的属性和方法，实现代码复用。

首先，我们需要创建一个基础表格类，例如 BaseTable。在这个类中，我们定义了一些通用的属性和方法，例如表格名称、字段、主键等。以下是一个示例：

import slick.jdbc.JdbcProfile

abstract class BaseTable[T](tag: Tag, tableName: String) extends Table[T](tag, tableName) {
  // 定义 id 字段
  def id = column[Long]("id", O.PrimaryKey, O.AutoInc)
}

然后，我们可以通过继承 BaseTable 类来创建特定的表格。例如，我们希望创建一个名为 UserTable 的用户表格，可以这样做：

import slick.lifted.ProvenShape

class UserTable(tag: Tag) extends BaseTable[(Long, String, Int)](tag, "users") {
  // 定义其他字段
  def name = column[String]("name")
  def age = column[Int]("age")

  // 定义投影
  def * : ProvenShape[(Long, String, Int)] = (id, name, age)
}

通过这种方式，我们可以在不重复定义 id 字段的情况下，轻松地创建一个 UserTable 表格。

使用 Traits 扩展 SLICK 表格

除了继承基础类外，我们还可以使用 Scala 的 Traits 来扩展 SLICK 表格。Traits 提供了一种代码复用的方式，可以在多个类之间共享方法和属性。

首先，我们定义一个 BaseTable Trait，并在其中实现通用的属性和方法。以下是一个示例：

import slick.jdbc.JdbcProfile

trait BaseTable {
  val profile: JdbcProfile
  import profile.api._

  // 定义 id 字段
  def id = column[Long]("id", O.PrimaryKey, O.AutoInc)
}

然后，我们可以通过混入该 Trait 来扩展其他的表格类。例如，我们可以定义一个 UserTable 表格，并混入 BaseTable Trait，如下所示：

import slick.lifted.ProvenShape

class UserTable(tag: Tag, val profile: JdbcProfile) extends Table[(Long, String, Int)](tag, "users") with BaseTable {
  // 定义其他字段
  def name = column[String]("name")
  def age = column[Int]("age")

  // 定义投影
  def * : ProvenShape[(Long, String, Int)] = (id, name, age)
}

通过这种方式，我们可以共享 BaseTable Trait 中定义的 id 字段，而不需要重复定义。

通过扩展和混入创建更复杂的表格

除了基础的扩展和混入之外，我们还可以使用继承和 Traits 的组合来创建更复杂的表格。通过灵活使用这些特性，我们可以根据需要创建各种不同的表格。

以下是一个示例，展示了如何使用继承和混入来创建一个包含时间戳的表格：

import slick.lifted.ProvenShape
import java.sql.Timestamp

trait TimestampTable extends BaseTable {
  val profile: JdbcProfile
  import profile.api._

  def createdAt = column[Timestamp]("created_at")
  def updatedAt = column[Timestamp]("updated_at")
}

class UserTable(tag: Tag, val profile: JdbcProfile) extends Table[(Long, String, Int, Timestamp, Timestamp)](tag, "users")
  with BaseTable
  with TimestampTable {

  // 定义其他字段
  def name = column[String]("name")
  def age = column[Int]("age")

  // 定义投影
  def * : ProvenShape[(Long, String, Int, Timestamp, Timestamp)] = (id, name, age, createdAt, updatedAt)
}