在 Scala 中进行结果分页的更好方法

Question

我经常进行结果分页（给定页码和页面大小，计算开始、结束和总页数），我从 Java 移植了这个小函数来提供帮助：

def page(page: Int, pageSize: Int, totalItems: Int) = {
    val from = ((page - 1) * pageSize) + 1
    var to = from + pageSize - 1
    if (to > totalItems) to = totalItems
    var totalPages: Int = totalItems / pageSize
    if (totalItems % pageSize > 0) totalPages += 1
    (from, to, totalPages)
}

在接收方：

val (from, to, totalPages) = page(page, pageSize, totalItems)

虽然它有效，但我确信在 Scala 中有更多可读性和功能性的方法来做同样的事情。什么是更类似于 scala 的方法？

特别是，我试图找到一种更好的说法：

var to = from + pageSize - 1
if (to > totalItems) to = totalItems

在 Java 中我可以这样做：

from + pageSize - 1 + (to > totalItems) ? 1 : 0;

Answer 1

问题的一半在于识别模式：

def pageCalc(page: Int, pageSize: Int, totalItems: Int) = {
    val pages = 1 to totalItems by pageSize
    val from = pages(page - 1)
    val to = from + pageSize - 1 min totalItems
    val totalPages = pages.size
    (from, to, totalPages)
}

不过，真的，也许您可​​以直接使用Range 代替？

Answer 2

最简单的改进就是使用函数而不是 vars（并避免用参数名称遮蔽方法名称，因此更清楚您是否调用递归函数）：

def pageCalc(page: Int, pageSize: Int, totalItems: Int) = {
  val from = ((page - 1) * pageSize) + 1
  val to = totalItems min (from + pageSize - 1)
  val totalPages = (totalItems / pageSize) + (if (totalItems % pageSize > 0) 1 else 0)
  (from, to, totalPages)
}

关键更改只是使用min 函数而不是单独的var，并让if 语句返回0 或1，而不是让它更新一个var。