Scala: mapa con dos o más opciones
-
27-10-2019 - |
Pregunta
Básicamente estoy buscando la forma más similar a Scala de hacer lo siguiente:
def sum(value1: Option[Int], value2: Option[Int]): Option[Int] =
if(value1.isDefined && value2.isDefined) Some(value1.get + value2.get)
else if(value1.isDefined && value2.isEmpty) value1
else if(value1.isEmpty && value2.isDefined) value2
else None
Esto da la salida correcta:
sum(Some(5), Some(3)) // result = Some(8)
sum(Some(5), None) // result = Some(5)
sum(None, Some(3)) // result = Some(3)
sum(None, None) // result = None
Sin embargo, para resumir más de dos opciones, tendría que usar demasiadas if
s o usar algún tipo de bucle.
Editar 1:
Mientras escribía la pregunta, se me ocurrió una especie de respuesta:
def sum2(value1: Option[Int], value2: Option[Int]): Option[Int] =
value1.toList ::: value2.toList reduceLeftOption { _ + _ }
Este se ve muy idiomático para mi ojo inexperto. Esto incluso funcionaría con más de dos valores. ¿Sin embargo, es posible hacer lo mismo sin convertir a listas?
Editar-2:
Terminé con esta solución (gracias a Ziggystar):
def sum(values: Option[Int]*): Option[Int] =
values.flatten reduceLeftOption { _ + _ }
Edición-3:
Otra alternativa gracias a Landei:
def sum(values: Option[Int]*): Option[Int] =
values collect { case Some(n) => n } reduceLeftOption { _ + _ }
Solución
Qué tal si:
scala> def sum(values: Option[Int]*): Option[Int] = values.flatten match {
| case Nil => None
| case l => Some(l.sum)
| }
sum: (values: Option[Int]*)Option[Int]
scala> sum(Some(1), None)
res0: Option[Int] = Some(1)
scala> sum(Some(1), Some(4))
res1: Option[Int] = Some(5)
scala> sum(Some(1), Some(4), Some(-5))
res3: Option[Int] = Some(0)
scala> sum(None, None)
res4: Option[Int] = None
Editar
Tal vez sería sano devolver 0 si todos los argumentos fueran ninguno. En ese caso, la función se reduciría a values.flatten.sum
.
Otros consejos
scala> def sum(a: Option[Int], b: Option[Int]) = (a,b) match {
| case (Some(x), Some(y)) => Some(x + y)
| case (Some(x), None) => Some(x)
| case (None, Some(y)) => Some(y)
| case _ => None
| }
sum: (a: Option[Int],b: Option[Int])Option[Int]
scala> sum(Some(5), Some(3))
res0: Option[Int] = Some(8)
scala> sum(Some(5), None)
res1: Option[Int] = Some(5)
scala> sum(None, Some(3))
res2: Option[Int] = Some(3)
scala> sum(None, None)
res3: Option[Int] = None
Otra solución es:
def sum(values: Option[Int]*): Int = values.collect{case Some(n) => n}.sum
Mientras que en el caso actual flatten
es claramente más conveniente, el collect
La versión es más flexible, ya que permite realizar asignaciones y tener condiciones de filtro adicionales o patrones complejos. Por ejemplo, imagine que desea tener la suma de los cuadrados de todos los números pares en valores:
values.collect{case Some(n) if n mod 2 == 0 => n*n}.sum
Puedes hacerlo muy conciso usando el hecho de que hay un Semigroup
instancia Option
Eso hace exactamente lo que quieres. Puedes usar escala o gatos. Aquí hay un ejemplo usando cats
:
import cats.std.option._
import cats.syntax.semigroup._
import cats.std.int._
Option(1) |+| Option(2) // Some(3)
Option(1) |+| None // Some(1)
None |+| Option(2) // Some(2)
Entonces tus sum
se convierte en:
def sum(v1: Option[Int], v2: Option[Int]): Option[Int] = v1 |+| v2
Solución reducida de Michael.Kebe con un pequeño aspecto a algunas reglas matemáticas básicas:
def sum(a: Option[Int], b: Option[Int]) = (a,b) match {
case (None,None) => None
case _ => Some(a.getOrElse(0)+b.getOrElse(0))
}
scala> sum(Some(5), Some(3)) // result = Some(8)
res6: Option[Int] = Some(8)
scala> sum(Some(5), None) // result = Some(5)
res7: Option[Int] = Some(5)
scala> sum(None, Some(3)) // result = Some(3)
res8: Option[Int] = Some(3)
scala> sum(None, None) // result = None
res9: Option[Int] = None