Scala: вложенный getOrElse

Question

Я хотел бы создать метод, который принимает в качестве аргументов массив параметров и значение по умолчанию и возвращает первый непустой параметр, в противном случае значение по умолчанию:

def customGetOrElse[T](options : Array[Option[T]], defaultValue : T) : T = {
  // Example if options contain 2 elements
  options(0).getOrElse(options(1).getOrElse(defaultValue))
  // If options contain 3 elements
  options(0).getOrElse(options(1).getOrElse(options(2).getOrElse(defaultValue)))
}

Но я изо всех сил пытаюсь сделать этоМетод работает для любого размера массива.Есть идеи?

Спасибо!

Answer 1

Один с collectFirst

def customGetOrElse[T](options: Array[Option[T]], defaultValue: T): T =
 options.collectFirst { case Some(value) => value }.getOrElse(defaultValue)

Scala REPL

Welcome to Scala 2.12.6 (Java HotSpot(TM) 64-Bit Server VM, Java 1.8.0_162).
Type in expressions for evaluation. Or try :help.

scala> :paste
// Entering paste mode (ctrl-D to finish)


 def customGetOrElse[T](options: Array[Option[T]], defaultValue: T): T =
  options.collectFirst { case Some(value) => value }.getOrElse(defaultValue)

// Exiting paste mode, now interpreting.

customGetOrElse: [T](options: Array[Option[T]], defaultValue: T)T

scala> customGetOrElse[Int](Array(None, Some(1)), 2)
res0: Int = 1

scala> customGetOrElse[Int](Array(None, None), 2)
res1: Int = 2

Answer 2

Используйте flatten над options, чтобы уменьшить options до тех, которые имеют значение.Затем используйте headOption для возврата Option, содержащего первое имеющееся значение, или None.Наконец, получите значение T из Option или верните defaultValue.

def customGetOrElse[T](options : Array[Option[T]], defaultValue : T) : T = {
  options.flatten.headOption.getOrElse(defaultValue)
}

Примеры:

scala> customGetOrElse[Int](Array(None, None), 666)
res29: Int = 666

scala> customGetOrElse[Int](Array(None, Some(42)), 666)
res30: Int = 42

Answer 3

def customGetOrElse[T](options : Array[Option[T]], defaultValue : T) : T = {
  options.find(_.isDefined).flatten.getOrElse(defaultValue)
}