如何使用Option或PartialFunction简化方法

时间:2018-01-19 11:10:52

标签: scala flatmap partialfunction

我需要一些帮助来修改(简化)我的代码 这是一个例子:

def getBids(rawBids: RDD[List[String]], exchangeRates: Map[String, Double]): RDD[BidItem] = {
    rawBids.filter(bidList => !bidList(2).matches(ERROR_REGEXP))
      .flatMap(bidList => {
        Constants.TARGET_LOSAS.map { losa =>
          val indexOfLosa = Constants.BIDS_HEADER.indexOf(losa)
          if (bidList(indexOfLosa) != null && !bidList(indexOfLosa).equals("") && isDoubleNumber(bidList(indexOfLosa))) {
            BidItem(bidList.head, bidList(1), losa, bidList(indexOfLosa).toDouble)
          } else {
            BidItem(bidList.head, bidList(1), losa, 0)
          }
        }
      })
      .filter(unit => unit.price != 0)
      .map(unit => {
        val convertedPrice = usdToEur(unit.price, unit.bidDate, exchangeRates)
        val convertedDate = changeDateFormat(unit.bidDate)
        BidItem(unit.motelId, convertedDate, unit.loSa, convertedPrice)
      })
  }

我需要更改flatMap方法并避免使用" ELSE"代码的一部分
 据我所知,我们可以使用部分功能或[选项]。它可以帮助我们避免情况如果我们的"如果"声明是假的。如果我从代码中删除ELSE部分,scala不会给我机会编译我的代码。谢谢!

1 个答案:

答案 0 :(得分:5)

val doubleValue = Option(bidList(indexOfLosa))
  .filterNot(_.equals(""))
  .filter(isDoubleNumber)
  .map(_.toDouble)
  .getOrElse(0)
BidItem(bidList.head, bidList(1), losa, doubleValue)