将"pimp my Iterable"扩展到选项

Question

最近我发现在我的代码中有几个地方我首先收集了一些解决方案，然后只有当解决方案是唯一的(解决方案集合只包含一个元素)时才继续处理它们。下面的代码试图以更实用的方式解决这个问题。

  implicit class GetOnlyOne[A](val coll: Iterable[A]) {
    def getonlyone = {
      if (coll.isEmpty) None
      else if (coll.tail.isEmpty) coll.headOption
      else None
    }
  }

函数的用法如下:

Seq(1).getonlyone
Seq(1,2).getonlyone
Set(1).getonlyone

当前不能工作的是:

Some(1).getonlyone

除了Iterable之外，是否可以改进该函数以接受Option，也许可以使用视图边界?

Answer 1

您可能能够为Option添加视图边界，但一般的解决方案是定义一个类型类:您定义一个接口并为您想要支持的每种类型提供该接口的隐式实例:

trait CanGetOnlyOne[F[_]] {
  def getOnlyOne[A](fa: F[A]): Option[A]
}
object CanGetOnlyOne {
  implicit object CanGetOnlyOneIterable extends CanGetOnlyOne[Iterable]{
    def getOnlyOne[A](fa: Iterable[A]) = ...
  }
  implicit object CanGetOnlyOneOption extends CanGetOnlyOne[Option] {
    def getOnlyOne[A](fa: Option[A]) = fa
  }
}
implicit class GetOnlyOne[F[_], A](fa: F[A])(implicit cgoo: CanGetOnlyOne[F]) {
  def getonlyone = cgoo.getOnlyOne(fa)
} 

Answer 2

Option可以隐式转换为Iterable，因此如下工作:

implicit class GetOnlyOne[A, Coll](coll: Coll)
                                  (implicit view: Coll => Iterable[A]) {
  def getonlyone = {
    val it: Iterable[A] = coll
    if (it.isEmpty) None
    else if (it.tail.isEmpty) it.headOption
    else None
  }
}

但是对于Option，这是非常低效的，因为您的getonlyone本质上是identity函数。因此，我将为options引入第二个方法扩展:

implicit class GetOnlyOneOption[A](private val opt: Option[A]) extends AnyVal {
  def getonlyone = opt
}