Slick:autoInc如何在MultiDBCakeExample示例中工作?

时间:2012-12-06 07:53:24

标签: scala slick

我正在尝试了解Slick的工作原理以及如何使用它...并在GitHub中查看他们的示例我最终在MultiDBCakeExample.scala中使用了此代码段:

trait PictureComponent { this: Profile => //requires a Profile to be mixed in...
  import profile.simple._ //...to be able import profile.simple._

  object Pictures extends Table[(String, Option[Int])]("PICTURES") {
    ...

    def * = url ~ id

    val autoInc = url returning id into { case (url, id) => Picture(url, id) }

    def insert(picture: Picture)(implicit session: Session): Picture = {
      autoInc.insert(picture.url)
    }
  }
}

我认为*方法在表中返回一行,而autoInc应该以某种方式提供自动递增实体ID的功能......但说实话我在理解这一点时遇到了一些麻烦一段代码。 returning指的是什么? autoInc返回什么?

我查看了Slick文档但我无法找到有用的信息。任何帮助都会非常感激; - )

1 个答案:

答案 0 :(得分:6)

因为autoInc可能令人困惑,我将为您提供一个工作示例(请注意我的数据库是PostgreSQL所以我需要使用forInsert进行hack以使Postgresql驱动程序增加auto-inc值)。

case class GeoLocation(id: Option[Int], latitude: Double, longitude: Double, altitude: Double)

/**
 * Define table "geo_location".
 */
object GeoLocations extends RichTable[GeoLocation]("geo_location") {
  def latitude = column[Double]("latitude")
  def longitude = column[Double]("longitude")
  def altitude = column[Double]("altitude")

  def * = id.? ~ latitude ~ longitude ~ altitude <> (GeoLocation, GeoLocation.unapply _)
  def forInsert = latitude ~ longitude ~ altitude <> ({ (lat, long, alt) => GeoLocation(None, lat, long, alt) },
    { g: GeoLocation => Some((g.latitude, g.longitude, g.altitude)) })
}

我的RichTable是一个抽象类,为了不为每个表声明id,只是扩展它:

abstract class RichTable[T](name: String) extends Table[T](name) {
  def id = column[Int]("id", O.PrimaryKey, O.AutoInc)

  val byId = createFinderBy(_.id)

}

并使用它:

GeoLocations.forInsert.insert(GeoLocation(None, 22.23, 25.36, 22.22))

由于您为None传递id,当Slick插入此新实体时,它将由PostgreSql驱动程序自动生成。 自从我开始使用Slick之后的几个星期,我真的推荐它了!

更新:如果您不想使用forInsert投影,则会出现以下其他方法 - 在我的情况下,实体为Address

为架构创建的每个表创建序列:

session.withTransaction {
  DBSchema.tables.drop
  DBSchema.tables.create
  // Create schemas to generate ids too.
  Q.updateNA("create sequence address_seq")
}

定义一个使用序列生成id的方法(我在once类中定义了这个RichTable

  def getNextId(seqName: String) = Database { implicit db: Session =>
    Some((Q[Int] + "select nextval('" + seqName + "_seq') ").first)
  }

并在映射器中覆盖insert方法,如:

  def insert(model : Address) = Database { implicit db: Session =>
    *.insert(model.copy(id = getNextId(classOf[Address].getSimpleName())))
  }

现在,你可以在插入时传递None,这个方法可以为你做好工作......