有没有更好的方法在Scala中提升PartialFunction?

时间:2011-01-18 07:14:48

标签: scala functional-programming partialfunction

我偶尔会遇到以下模式,我基本上有一个PartialFunction[SomeType,AnotherType],并希望将其视为Function[SomeType,Option[AnotherType],例如:

def f(s:SomeType):Option[AnotherType] = s match {
  case s1:SubType1 => Some(AnotherType(s1.whatever))
  case s2:SubType2 => Some(AnotherType(s2.whatever))
  case _ => None
}

有没有办法以避免默认情况的方式编写上述函数,并将结果包装在Some定义的位置?到目前为止,我提出的最好的是:

def f(s:SomeType):Option[AnotherType] = pf.lift(s)
def pf:PartialFunction[SomeType,AnotherType] = {
  case s1:SubType1 => AnotherType(s1.whatever)
  case s2:SubType2 => AnotherType(s2.whatever)
}

有没有办法在不定义中间函数的情况下完成?我已经尝试了以下各种方法,但还没有任何东西可以编译:

def f:Function[SomeType,Option[AnotherType]] = {
  case s1:SubType1 => AnotherType(s1.whatever)
  case s2:SubType2 => AnotherType(s2.whatever)
}.lift

2 个答案:

答案 0 :(得分:6)

对象scala.PartialFunction中的

condOpt。来自scaladoc:

def onlyInt(v: Any): Option[Int] = condOpt(v) { case x: Int => x }

答案 1 :(得分:4)

不是答案,而是解释为什么huynhjl的答案是正确的......

你的一部分困惑是你试图def部分功能。所有这一切都是为了创建一个返回PartialFunction对象的方法,当你也可以直接创建它时:

val pf: PartialFunction[SomeType,AnotherType] = {
  case s1:SubType1 => AnotherType(s1.whatever)
  case s2:SubType2 => AnotherType(s2.whatever)
}

虽然我个人更喜欢使用类型归属:

val pf = {
  case s1:SubType1 => AnotherType(s1.whatever)
  case s2:SubType2 => AnotherType(s2.whatever)
} : PartialFunction[SomeType,AnotherType]

无论哪种方式,您都必须指定输入类型,因此您必须提供PartialFunction的确切签名。我知道感觉应该可以推断出来,但是,唉,可悲的是不是这样!

使用归属版本,您可以在同一个地方定义和解除所有内容:

val pf = ({
  case s1:SubType1 => AnotherType(s1.whatever)
  case s2:SubType2 => AnotherType(s2.whatever)
} : PartialFunction[SomeType,AnotherType]).lift

PartialFunction.condOpt是更好的解决方案,因为它允许推理器为您完成大部分工作,留下更清晰的代码:)