Como escrever uma versão argumento preguiçoso, variável de “OrElse”
-
22-07-2019 - |
Pergunta
É possível escrever um método orElse
generalizada de Option
que leva um número variável de argumentos? Ou seja, em vez de:
lazy val o1 = { println("foo"); None }
lazy val o2 = { println("bar"); Some("bar") }
lazy val o3 = { println("baz"); Some("baz") }
// ...
o1 orElse o2 orElse o3 // orElse ...
Você pode usar:
orElse(o1, o2, o3) //, ...
Solução
De acordo com a O Scala especificação da linguagem (Declarações 4.6 Função e Definições) você não pode definir varargs parâmetros de-name:
ParamType ::= Type
| ‘=>’ Type
| Type ‘*’
scala> def orElse(x : (=> String)*)
<console>:1: error: no by-name parameter type allowed here
def orElse(x : (=> String)*)
Você poderia substituir o arg preguiçoso com função e uma conversão de tipo implícito:
def orElse[T](x : (()=> Option[T])*) : Option[T] =
if(x.isEmpty) None else x.first.apply.orElse(orElse((x drop 1) :_*))
implicit def anyToFun0[T](t : => T) : (() => T) = () => t
orElse(o1, o2, o3)
Outras dicas
Eu encontrei a pergunta um pouco tarde :). Uma possibilidade é envolver => A
em uma classe auxiliar, juntamente com uma função auxiliar para simplificar a sua criação:
import scala.language.implicitConversions
class Helper[+A](value: => A) extends Function0[A] {
override def apply(): A = value;
}
object Helper {
def unapply[A](h: Helper[A]): Option[A] = Some(h());
}
implicit def toHelper[A](body: => A) = new Helper(body);
O exaustor não é necessário, ele só permite fácil correspondente no helper. Então, podemos escrever
def orElse[A](xs: Helper[Option[A]]*): Option[A] =
xs.collectFirst[A]({
case Helper(Some(r)) => r;
})
lazy val o1 = { println("foo"); None }
lazy val o2 = { println("bar"); Some("bar") }
lazy val o3 = { println("baz"); Some("baz") }
orElse(o1, o2, o3) //, ...
Esta é apenas uma solução simplificada, um mais realista seria
def orElse[A](x: Option[A], xs: Helper[Option[A]]*): Option[A]
com uma implementação mais eficiente.
Já existe uma classe semelhante ao Helper
em Scalaz, chamado Name
com Need
implementação que garante que o corpo é avaliado no máximo uma vez. Assim, com Scalaz, poderia ser implementado como
import scala.language.implicitConversions
import scalaz._
import scalaz.Scalaz._
implicit def toNeed[A](body: => A): Name[A] = Need(body);
def orElse[A](xs: Name[Option[A]]*): Option[A] =
xs.collectFirst[A]({
case Name(Some(r)) => r;
})
lazy val o1 = { println("foo"); None }
lazy val o2 = { println("bar"); Some("bar") }
lazy val o3 = { println("baz"); Some("baz") }
orElse(o1, o2, o3) //, ...