Как преобразовать X => Option[R] в PartialFunction[X,R]
-
19-09-2019 - |
Вопрос
До тех пор, пока у нас есть PartialFunction[X,R]
очень легко преобразовать его в функцию, возвращающую Option[R]
, например
def pfToOptf[X, R](f: PartialFunction[X,R])(x: X) =
if (f.isDefinedAt(x)) Some(f(x))
else None
Однако что, если задача прямо противоположна:предположим, у меня есть функция f
получение X
в качестве аргумента и возвращения Option[R]
в результате.И я хочу сделать PartialFunction[X,R]
выйти из этого состояния.Каков наилучший способ?
То, что я придумал, на мой вкус, выглядит довольно уродливо:
def optfToPf[X,R](f: X => Option[R]) : PartialFunction[X,R] = {
object extractor {
def unapply(x: X): Option[R] = f(x)
}
{ case extractor(r) => r }
}
Есть ли какой-нибудь лучший способ, который я пропустил?
Решение
Начиная Scala 2.9
, Function.unlift
делает именно это:
def unlift[T, R](f: (T) => Option[R]): PartialFunction[T, R]
Преобразует функцию T => Option[R] в частичную функцию[T, R].
Другие советы
Как насчет этого:
Welcome to Scala version 2.8.0.r19650-b20091114020153 (Java HotSpot(TM) Client VM, Java 1.6.0_17).
Type in expressions to have them evaluated.
Type :help for more information.
scala> def optfToPf[X,R](f: X => Option[R]): PartialFunction[X,R] = x => f(x) match {
| case Some(r) => r
| }
optfToPf: [X,R](f: (X) => Option[R])PartialFunction[X,R]
scala>
Я полагаю, вы могли бы переопределить apply и isDefinedAt вручную, но я бы сделал это так, как вы считаете некрасивым.
def optfToPf[X,R](f: X => Option[R]) = new PartialFunction[X,R] {
def apply(x: X): R = f(x).get
def isDefinedAt(x: X): Boolean = f(x) != None
}
Тестирование:
scala> val map = Map(1 -> 2)
map: scala.collection.immutable.Map[Int,Int] = Map(1 -> 2)
scala> map(1)
res0: Int = 2
scala> def mapOpt(key: Int) = map.get(key)
mapOpt: (key: Int)Option[Int]
scala> mapOpt(1)
res1: Option[Int] = Some(2)
scala> mapOpt(2)
res2: Option[Int] = None
scala> val mapPf = optfToPf(mapOpt _)
mapPf: java.lang.Object with PartialFunction[Int,Int] = <function1>
scala> mapPf.isDefinedAt(2)
res3: Boolean = false
scala> mapPf.isDefinedAt(1)
res4: Boolean = true
scala> mapPf(1)
res5: Int = 2
scala> mapPf(2)
java.util.NoSuchElementException: None.get