我问这个问题,因为我发现自己需要类似于普通函数andThen
的功能(但仍然符合部分函数的语义)。我的意思应该解释如下:
andThen
:
val x: Int => Int = ???
val y: Int => Double = ???
val z: Int => Double = x andThen y
部分函数的 andThen
(由于andThen
已经存在且具有不同的语义,因此命名不同):
val x: PartialFunction[A, B] = ???
val y: PartialFunction[B, C] = ???
val z: PartialFunction[A, C] = x andThenExtact y
在参数z
上定义了a: A
,其中包含以下属性:
x在
y在x(a)
我提出了以下扩展,原则上应该提供所需的功能:
type =>?[-A, +B] = PartialFunction[A, B]
class RichPartialFunction[A, B](private val lhs: A =>? B) extends AnyVal {
def andThenWiden[BB >: B, R](widening: BB =>? R): A =>? R = {
val InnerMatcher = matchArgument(lhs)
val pf: A =>? R = {
case InnerMatcher(res) if widening.isDefinedAt(res) =>
widening(res)
}
pf
}
def andThenExact[R](pipeTo: B =>? R): A =>? R = {
val InnerMatcher = matchArgument(lhs)
val pf: A =>? R = {
case InnerMatcher(res) if pipeTo.isDefinedAt(res) =>
pipeTo(res)
}
pf
}
}
private def matchArgument[A, B](fun: A =>? B): PartiallyUnapply[A, B] = new PartiallyUnapply(fun)
private class PartiallyUnapply[A, B](private val fun: A =>? B) extends AnyVal {
def unapply(arg: A): Option[B] = {
fun.andThen(Some(_)).applyOrElse(arg, (_: A) => None)
}
}
我的问题是这个问题有更好的解决方案,如果我的方法不好,那我怎么能在我的代码中替换x andThenExtact y
?