7

我使用 Slick 3.0 创建了一个多对多集合,但我正在努力以我想要的方式检索数据。

事件和兴趣之间存在多对多的关系。这是我的表:

case class EventDao(title: String,
                    id: Option[Int] = None)


class EventsTable(tag: Tag)
  extends Table[EventDao](tag, "events") {

  def id = column[Int]("event_id", O.PrimaryKey, O.AutoInc)
  def title = column[String]("title")

  def * = (
    title,
    id.?) <> (EventDao.tupled, EventDao.unapply)

  def interests = EventInterestQueries.query.filter(_.eventId === id)
    .flatMap(_.interestFk)
}


object EventQueries {

  lazy val query = TableQuery[EventsTable]

  val findById = Compiled { k: Rep[Int] =>
    query.filter(_.id === k)
  }
}

这是EventsInterests:

case class EventInterestDao(event: Int, interest: Int)


class EventsInterestsTable(tag: Tag)
  extends Table[EventInterestDao](tag, "events_interests") {

  def eventId = column[Int]("event_id")
  def interestId = column[Int]("interest_id")

  def * = (
    eventId,
    interestId) <> (EventInterestDao.tupled, EventInterestDao.unapply)

  def eventFk = foreignKey("event_fk", eventId, EventQueries.query)(e => e.id)
  def interestFk = foreignKey("interest_fk", interestId, InterestQueries.query)(i => i.id)
}


object EventInterestQueries {
  lazy val query = TableQuery[EventsInterestsTable]
}

最后是兴趣:

case class InterestDao(name: String,
                       id: Option[Int] = None)

class InterestsTable(tag: Tag)
  extends Table[InterestDao](tag, "interests") {

  def id = column[Int]("interest_id", O.PrimaryKey, O.AutoInc)
  def name = column[String]("name")
  def name_idx = index("idx_name", name, unique = true)

  def * = (
    name,
    id.?) <> (InterestDao.tupled, InterestDao.unapply)

  def events = EventInterestQueries.query.filter(_.interestId === id)
    .flatMap(_.eventFk)
}


object InterestQueries {

  lazy val query = TableQuery[InterestsTable]

  val findById = Compiled { k: Rep[Int] =>
    query.filter(_.id === k)
  }
}

我可以使用以下内容查询和检索 (event.name,interest) 的元组:

val eventInterestQuery = for {
  event <- EventQueries.query
  interest <- event.interests
} yield (event.title, interest.name)

Await.result(db.run(eventInterestQuery.result).map(println), Duration.Inf)

所以这就是我目前所拥有的。

我想要的是能够填充一个案例类,如:

case class EventDao(title: String,
                interests: Seq[InterestDao],
                id: Option[Int] = None)

问题是,如果我像这样更新我的案例类,它会打乱我def *EventsTable. 另外,我必须将过滤器重命名为有点难看的EventsTable.interests东西,但如果有必要我可以忍受。EventsTable.interestIds

此外,我找不到编写for产生(event.name, Seq(interest.name)). 无论如何,这只是我能够产生一个(EventDao, Seq(InterestDao))我真正想要返回的元组的垫脚石。

有谁知道我怎样才能实现这些目标?我还希望能够“获取”一定数量的兴趣,因此对于某些查询,所有查询都会返回,但对于其他查询,只会返回前 3 个。

4

2 回答 2

7

所以在阅读了这个页面并在邮件列表上聊天之后,我终于让它工作了:

val eventInterestQuery = for {
  event <- EventQueries.query
  interest <- event.interests
} yield (event, interest)

Await.result(db.run(eventInterestQuery.result
  // convert the interests to a sequence.
  .map {
  _.groupBy(_._1)
    .map {
      case (k,v) => (k, v.map(_._2))
  }.toSeq
}
), Duration.Inf)
于 2015-03-28T23:10:57.153 回答
2

groupBy 的唯一问题是您失去了订单。你可以折叠结果。我已经为我当前的项目编写了这个助手:

def foldOneToMany[A, B](in: Seq[(A, Option[B])], eq: (A, B) => Boolean)
                        (f: (A, B) => A): Seq[A] = 
  in.foldLeft(List.empty[A]) {
    case (head :: tail, (_, Some(rel))) if eq(head, rel) =>
      f(head, rel) :: tail
    case (r, (el, Some(rel))) => f(el, rel) :: r
    case (r, (el, None)) => el :: r
  }.reverse

它可以与一些爱有关。现在它接受一个函数 A,B => Boolean 来确定 B 是否属于 A 和一个函数 A,B => A 将 B 添加到 A。

Virtualeyes也有一点。在 Postgres 中,您可以使用array_agg函数从数据库中使用更少的带宽。

于 2015-05-12T18:03:13.553 回答