scala - 将函数转换为部分函数 scala

Question

我有一个密封的特质:

sealed trait ActorMessage
case class AddX(x: Int) extends ActorMessage
case class RemoveX(x: Int) extends ActorMessage

我还有一个函数来处理所有消息并警告我有关非详尽匹配的信息:

def handleMessage: ActorMessage => Unit = {
  case AddX(x) => ...
  case RemoveX(x) => ...
}

Actor 需要 PartialFunction[Any, Unit]。 PartialFunction 扩展了 Function，这意味着我无法将我的 Function 分配给 PartialFunction。

我编写了简单的转换器:

def liftToPartialFunction[FUND <: PFUND, B, PFUND](f: Function[FUND, B]): PartialFunction[PFUND, B] = new PartialFunction[PFUND, B] {
  override def isDefinedAt(x: PFUND): Boolean = x.isInstanceOf[FUND]
  override def apply(v1: PFUND): B = f(v1.asInstanceOf[FUND])
}

但是有更好的方法吗？或者标准 scala 库中是否有等效项？

Answer 1

您可以使用Function.unlift -

val f: Throwable => Option[String] = {
  case e: NullPointerException => Some("nah it's ok")
  case e => None
}

Future(null.toString).recover(Function.unlift(f))
// Future(Success(nah it's ok))