我的表数据形成了一个树结构,其中一行可以引用同一个表中的父行。
我想要使用 Slick 实现的是编写一个查询,该查询将返回一行及其所有子项。此外,我也想做同样的事情,但编写一个查询,该查询将返回一个孩子及其所有祖先。
换句话说:
findDown(1)
应该返回
List(Group(1, 0, "1"), Group(3, 1, "3 (Child of 1)"))
findUp(5)
应该返回
List(Group(5, 2, "5 (Child of 2)"), Group(2, 0, "2"))
这是一个功能齐全的工作表(缺少的解决方案除外 ;-)。
package com.exp.worksheets
import scala.slick.driver.H2Driver.simple._
object ParentChildTreeLookup {
implicit val session = Database.forURL("jdbc:h2:mem:test1;", driver = "org.h2.Driver").createSession()
session.withTransaction {
Groups.ddl.create
}
Groups.insertAll(
Group(1, 0, "1"),
Group(2, 0, "2"),
Group(3, 1, "3 (Child of 1)"),
Group(4, 3, "4 (Child of 3)"),
Group(5, 2, "5 (Child of 2)"),
Group(6, 2, "6 (Child of 2)"))
case class Group(
id: Long = -1,
id_parent: Long = -1,
label: String = "")
object Groups extends Table[Group]("GROUPS") {
def id = column[Long]("ID", O.PrimaryKey, O.AutoInc)
def id_parent = column[Long]("ID_PARENT")
def label = column[String]("LABEL")
def * = id ~ id_parent ~ label <> (Group, Group.unapply _)
def autoInc = id_parent ~ label returning id into {
case ((_, _), id) => id
}
def findDown(groupId: Long)(implicit session: Session) = { ??? }
def findUp(groupId: Long)(implicit session: Session) = { ??? }
}
}
一个非常糟糕的静态尝试findDown
可能是这样的:
private def groupsById = for {
group_id <- Parameters[Long]
g <- Groups; if g.id === group_id
} yield g
private def childrenByParentId = for {
parent_id <- Parameters[Long]
g <- Groups; if g.id_parent === parent_id
} yield g
def findDown(groupId: Long)(implicit session: Session) = { groupsById(groupId).list union childrenByParentId(groupId).list }
但是,我正在寻找一种让 Slick 使用 id 和 id_parent 链接递归搜索同一个表的方法。任何其他解决问题的好方法都非常受欢迎。但请记住,最好尽量减少数据库往返次数。