与 Quill 以多对多方式进行分组

Question

我正在尝试使用 Quill 实现以下 PostgreSQL 查询的功能：

select books.*, array_agg(authors.name) from books 
join authors_books on(books.id = authors_books.book_id)
join authors on(authors.id = authors_books.author_id)
group by books.id

现在我的 Quill 版本中有这个：

val books = quote(querySchema[Book]("books"))
val authorsBooks = quote(querySchema[AuthorBook]("authors_books"))
val authors = quote(querySchema[Author]("authors"))

val q: db.Quoted[db.Query[(db.Query[Book], Seq[String])]] = quote{
        books
            .join(authorsBooks).on(_.id == _.book_id)
            .join(authors).on(_._2.author_id == _.id)
            .groupBy(_._1._1.id)
            .map {
                case (bId, q) => {
                    (q.map(_._1._1), unquote(q.map(_._2.name).arrayAgg))
                }
            }
    }

如何摆脱结果中的嵌套查询 (db.Query[Book]) 并获得一本书？

Answer 1

我可能对 SQL 有点生疏，但您确定您的查询有效吗？特别是我发现您在 group by books.id 时使用 select books.* 表示怀疑，即您直接返回未分组的字段。尝试直接翻译错误的查询是导致事情出错的原因

解决此问题的一种方法是对所有字段执行group by。假设Book 被声明为：

case class Book(id: Int, name: String)

你可以的

  val qq: db.Quoted[db.Query[((Index, String), Seq[String])]] = quote {
    books
      .join(authorsBooks).on(_.id == _.book_id)
      .join(authors).on(_._2.author_id == _.id)
      .groupBy(r => (r._1._1.id, r._1._1.name))
      .map {
        case (bId, q) => {
          // (Book.tupled(bId), unquote(q.map(_._2.name).arrayAgg)) // doesn't work
          (bId, unquote(q.map(_._2.name).arrayAgg))
        }
      }
  }

  val res = db.run(qq).map(r => (Book.tupled(r._1), r._2))

不幸的是，您似乎无法在quote 中应用Book.tupled，因为您收到错误

单子组合不能使用应用连接来表达

但您可以在 db.run 调用后轻松完成此操作以取回您的 Book。

另一种选择是仅按Book.id 进行分组，然后再次加入Book 表以取回所有字段。如果Book中有很多字段，这实际上可能更干净、更快