Scala Some[Int]+Some[Int]=Some[Int]な関数を作りたい。

Question

タイトルの通りですが、以下のようなOption型でラップした変数の演算を行いたいのですがこの書き方のほかにきれいな方法はありますか？

def optSum(x: Option[Int], y: Option[Int]):Option[Int] = {
  (x, y) match {
    case (None, None) => None
    case _ => Some(x.getOrElse(0) + y.getOrElse(0))
  }
}

println(optSum(Some(1), Some(2))) // => 3
println(optSum(Some(3), None)) // => 3
println(optSum(None, None)) // => None

Answer 1

両方が None のときのみ None を返して、それ以外のケースでは None を 0 として扱うということですね。
私は yay さんが書かれている方法が一番シンプルだと思います。

両方が Some のときのみ計算してそれ以外を None で返す、というのであればもう少しシンプルになります。

def optSum(x: Option[Int], y: Option[Int]):Option[Int] = {
  (x, y) match {
    case (Some(x), Some(y)) => Some(x + y)
    case _ => None
  }
}

または

def optSum(x: Option[Int], y: Option[Int]):Option[Int] = {
  for {
    x <- x
    y <- y
  } yield x + y
}

Comments

[yay]

ありがとうございました！