为分页 API 创建 Scala 迭代器

Question

我正在编写一个 Scala 库，以便更轻松地查询分页 JSON API。每个 API 调用都会返回一个如下所示的对象：

{
  "count": 100,
  "current_page": 1,
  "total_pages": 2,
  "records": [
    ...
  ]
}

我想要一个返回某种迭代器（例如 MyIterator[Record]）的函数。在 Scala 世界中是否有任何标准方法可以做到这一点，甚至标准库中的构造可以帮助我？

我通常使用 lift-json 进行 JSON 解析，如果这有帮助的话。

谢谢。

Answer 1

如果您有一个 Iterator[Iterator[A]]，您可以使用 flatten 生成一个 Iterator[A]，它将所有嵌套的迭代器链接在一起：

scala> Iterator(Iterator(1, 2), Iterator(3, 4)).flatten.toList
res0: List[Int] = List(1, 2, 3, 4)

与迭代器上的工厂方法之一结合 伴随对象< /a>，您可能可以将其变成单行：

Iterator.range(1, pageCount).map(i: Int => fetchPage(i)).flatten

如果您只能通过检索页面来获取页数...则可能需要更复杂一点...类似：

val (firstPage, pageCount) = getFirstPageWithCount()
firstPage ++ (Iterator.range(2, pageCount).map(fetchPage).flatten)