Scala slick join table 与分组查询

Scala slick join table with grouped query

我有两个表:

  1. 商店

    class ShopTable(tag: Tag) extends GenericTableShop, UUID {

     def * = (id.?, name, address) <> (Shop.tupled, Shop.unapply _)
    
     def name = column[String]("name")
    
     def address = column[String]("address")
    

    }

    val shopTable = TableQuery[ShopDAO.ShopTable]

  2. 订单

    class OrderTable(tag: Tag) extends GenericTableOrder, UUID {

     def * = (id.?, shopId, amount) <> (Order.tupled, Order.unapply _)
    
     def shopId = column[UUID]("shop_id")
    
     def amount = column[Double]("amount")
    

    }

    val orderTable = TableQuery[OrderDAO.OrderTable]

我可以获得商店的静态信息(订单数、订单金额总和):

def getStatisticForShops(shopIds: List[UUID]): Future[Seq(UUID, Int, Double)] = {
  searchStatisticsByShopIds(shopIds).map(orderStatistics => 
    shopIds.map(shopId => {
      val o = orderStatistics.find(_._1 == shopId)
      (
        shopId,
        o.map(_._2).getOrElse(0),
        o.map(_._3).getOrElse(0.0)
      )
    })
  )
}

 def searchStatisticsByShopIds(shopIds: List[UUID]): Future[Seq(UUID, Int, Double)] = 
  db.run(searchStatisticsByShopIdsCompiled(shopIds).result)

 private val searchStatisticsByShopIdsCompiled = Compiled((shopIds: Rep[List[(UUID)]]) =>
  orderTable.filter(_.shopId === shopIds.any)
    .groupBy(_.shopId)
    .map { case (shopId, row) =>
      (shopId, row.length, row.map(_.amount).sum.get)
    }
 )

我需要按订单数对 shopTable 进行排序和过滤。

如何将分组的 orderTable 加入到缺少商店的零值的 shopTable 中?

我想有这样的要求:

| id(shopId) | name | address | ordersCount | ordersAmount |
|     id1    | name | address |       4     |    200.0     |
|     id2    | name | address |       0     |     0.0      |
|     id3    | name | address |       2     |    300.0     |

我使用 scala 2.12.6、slick 2.12:3.0.1、play-slick 3.0.1、slick-pg 0.16.3

P.S.我可能找到了解决办法

val shopsOrdersQuery: Query[(Rep[UUID], Rep[Int], Rep[Double]), (UUID, Int, Double), Seq] = searchShopsOrdersCompiled.extract

// Query shops with orders for sorting and filtering
val allShopsOrdersQueryQuery = shopTable.joinLeft(shopsOrdersQuery).on(_.id === _._1)
  .map(s => (s._1, s._2.map(_._2).getOrElse(0), s._2.map(_._3).getOrElse(0.0)))

private val searchShopsOrdersCompiled = Compiled(
  orderTable.groupBy(_.shopId)
    .map { case (shopId, row) =>
      (shopId, row.length, row.map(_.amount).sum.get)
    }
)

是的,这个解决方案工作正常

val shopsOrdersQuery: Query[(Rep[UUID], Rep[Int], Rep[Double]), (UUID, Int, Double), Seq] = searchShopsOrdersCompiled.extract

// Query shops with orders for sorting and filtering
val allShopsOrdersQueryQuery = shopTable.joinLeft(shopsOrdersQuery).on(_.id === _._1)
  .map(s => (s._1, s._2.map(_._2).getOrElse(0), s._2.map(_._3).getOrElse(0.0)))

private val searchShopsOrdersCompiled = Compiled(
  orderTable.groupBy(_.shopId)
    .map { case (shopId, row) =>
      (shopId, row.length, row.map(_.amount).sum.get)
    }
)