Question

我尝试将应该返回单个项目的迭代器转换为等效选项。

我能做的最好的就是这个。我应该使用标准API吗？

def toUniqueOption[T](a: Iterator[T]): Option[T] =
    if (a.size > 1)
      throw new RuntimeException("The iterator should be emtpy or contain a single item but contained ${a.size} items.")
    else if (a.size > 0)
      Option(a.toList(0))
    else
      Option.empty

更新了试用

def toUnique[T](a: Iterator[T]): Try[Option[T]] =
    if (a.size > 1)
      Failure(new RuntimeException("The iterator should be emtpy or contain a single item but contained ${a.size} items."))
    else if (a.size > 0)
      Success(Option(a.toList(0)))
    else
      Success(Option.empty)

Answer 1

调用size是有风险的，因为它无法保证有效甚至停止。

怎么样：

def toUniqueOption[T](a: Iterator[T]): Option[T] =
  a.take(2).toList match {
    case Nil => None
    case x :: Nil => Some(x)
    case _ => throw new RuntimeException("Iterator size > 1")
  }

Answer 2

您实际上可以使用标准API：

a.toStream.headOption

其中a：迭代器[T]

Answer 3

您可以使用hasNext和next来避免迭代整个序列：

def toUniqueOption[T](a: Iterator[T]): Option[T] = {
    if(a.hasNext) {
        val f = a.next()
        if(a.hasNext) throw new RuntimeException("Iterator should contain at most one element")
        Some(f)
    }
    else None
}

Answer 4

不完全是你要求的，但为什么不使用类似的东西：

  def nextAsOption[T](i: Iterator[T]) : Option[T] = {
    i.hasNext match {
      case true  => Some(i.next)
      case false => None
    }
  }

这只是为您提供了一个Iterator“next”操作，它返回一个Option而不是一个Boolean。当你需要传递选项时非常方便。

如何转换scala Iterator [T]到Option [T]

4 个答案: