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) //, ...
Foi útil?

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) //, ...
Licenciado em: CC-BY-SA com atribuição
Não afiliado a StackOverflow
scroll top