在Scala中实现ifTrue,ifFalse,ifSome,ifNone等,以避免if(…)和简

这是一个实施尝试.首先,使用pimp-my-library模式将ifTrue和ifFalse添加到Boolean.它们在返回类型R上进行参数化并接受单个名称参数,如果实现了指定的条件,则应对其进行评估.但在这样做的时候,我们必须允许另外一个电话.因此,我们返回一个名为OTHER0的新对象(为什么0将在后面解释),它将可能的中间结果存储为Option [R].定义当前条件(ifTrue或ifFalse)是否已实现,否则为空.

class BooleanWrapper(b: Boolean) {
  def ifTrue[R](f: => R) = new Otherwise0[R](if (b) Some(f) else None)
  def ifFalse[R](f: => R) = new Otherwise0[R](if (b) None else Some(f))
}
implicit def extendBoolean(b: Boolean): BooleanWrapper = new BooleanWrapper(b)

现在,这有效,让我写

someTest ifTrue {
  println("OK")
}

但是,如果没有以下其他条款,它当然不能返回R类型的值.所以这里是otherwise0的定义：

class Otherwise0[R](intermediateResult: Option[R]) {
  def otherwise[S >: R](f: => S) = intermediateResult.getOrElse(f)
  def apply[S >: R](f: => S) = otherwise(f)
}

当且仅当从前面的ifTrue或ifFalse获得的中间结果未定义时,它才会计算其传递的命名参数,这正是所需的.类型参数化[S>：R]具有以下效果：S被推断为命名参数的实际类型的最特定的常见超类型,例如,此片段中的r具有推断类型Fruit：

class Fruit
class Apple extends Fruit
class Orange extends Fruit

val r = someTest ifTrue {
  new Apple
} otherwise {
  new Orange
}

apply()别名甚至允许您为短代码块完全跳过其他方法名称：

someTest.ifTrue(10).otherwise(3)
// equivalently:
someTest.ifTrue(10)(3)

最后,这是选项的相应pimp：

class OptionExt[A](option: Option[A]) {
  def ifNone[R](f: => R) = new Otherwise1(option match {
    case None => Some(f)
    case Some(_) => None
  },option.get)
  def ifSome[R](f: A => R) = new Otherwise0(option match {
    case Some(value) => Some(f(value))
    case None => None
  })
}

implicit def extendOption[A](opt: Option[A]): OptionExt[A] = new OptionExt[A](opt)

class Otherwise1[R,A1](intermediateResult: Option[R],arg1: => A1) {
  def otherwise[S >: R](f: A1 => S) = intermediateResult.getOrElse(f(arg1))
  def apply[S >: R](f: A1 => S) = otherwise(f)
}

请注意,我们现在还需要otherwise1,以便我们可以方便地将unwrapped值传递给ifSome函数参数,还可以传递给ifNone后面的函数参数.