使用 Scala 和 Slick 将多对一映射

Question

我有一个用户可以再拥有一个 LoginInfos。我希望通过 id 找到这个用户，然后将每个关联的 LoginInfos 打包到 User 对象中。

我的期望是，我将获得与每个具有相同用户的 LoginInfo 的数量相等的行数。因此，我首先将用户映射到对象，然后再映射到每个 LoginInfos。有没有更好的办法？

这是我所拥有的：

def find(userID: UUID) = {
val query = for {
  dbUser <- slickUsers.filter(_.id === userID.toString)
  dbUserLoginInfo <- slickUserLoginInfos.filter(_.userID === dbUser.id)
  dbLoginInfo <- slickLoginInfos.filter(_.id === dbUserLoginInfo.loginInfoId)
} yield (dbUser, dbLoginInfo)
db.run(query.result).map { results =>
      var loginList = List[LoginInfo]()
results.foreach {
  case (user, loginInfo) =>
    loginList = LoginInfo(loginInfo.providerID, loginInfo.providerKey) :: loginList
}
results.headOption.map {
  case (user, loginInfo) =>
    User(
      UUID.fromString(user.userID),
      loginList,
      user.firstName,
      user.lastName,
      user.fullName,
      user.email,
      user.avatarURL)
}
}

Answer 1

我会这样做：

def find(userID: UUID) = {

  val query = for {
    dbUser <- slickUsers.filter(_.id === userID.toString)
    dbUserLoginInfo <- slickUserLoginInfos if(_.userID === dbUser.id) // join would also work
    dbLoginInfo <- slickLoginInfos if(_.id === dbUserLoginInfo.loginInfoId)
  } yield (dbUser, dbLoginInfo)

  db.run(query.result).map { results =>
    val grouped = results.groupBy(_._1) // group by user, but it will be max 1, so it's ok..
    grouped.map {
      case (user, group) =>
        val loginInfo = group.map{ case (dbUser, dbLoginInfo) => LoginInfo(dbLoginInfo.providerID, dbLoginInfo.providerKey) }
        User(
          UUID.fromString(user.userID),
          loginInfo,
          user.firstName,
          user.lastName,
          user.fullName,
          user.email,
          user.avatarURL)
    }.headOption
  }
}

看看我的blog post，我也做过类似的事情。 不要想当然，我没有你的表定义，但它应该可以工作..