scala:提高这段代码的可读性和风格

时间:2012-10-04 14:39:14

标签: scala error-handling idiomatic

以下是一个非常常见的播放框架2控制器:

def save(ideaId : Long) = CORSAction { request =>
  Idea.findById(ideaId).map { idea =>
    request.body.asJson.map { json =>
      json.asOpt[Comment].map { comment =>
        comment.copy(idea = idea).save.fold(
          errors => JsonBadRequest(errors),
          comment => Ok(toJson(comment).toString)
        )
      }.getOrElse     (JsonBadRequest("Invalid Comment entity"))
    }.getOrElse       (JsonBadRequest("Expecting JSON data"))
  }.getOrElse         (JsonBadRequest("Could not find idea with id '%s'".format(ideaId)))
}

我发现所有嵌套的.maps有点烦人,而且我发现每个错误处理都在底部有点乏味

您如何改进它以使其更具可读性,同时保留为功能惯用的scala代码?

我在考虑这样的事情(这是seudo代码,仍然无法编译)

def save(ideaId : Long) = CORSAction { request =>

  val idea = Idea.findById(ideaId).getOrElse(
    return JsonBadRequest("Could not find idea with id '%s'".format(ideaId)))

  val json = request.body.asJson.getOrElse(
    return JsonBadRequest("Expecting JSON data"))

  val comment = json.asOpt[Comment].getOrElse(
    return JsonBadRequest("Invalid Comment entity"))

  comment.copy(idea = idea).save.fold(
    errors => JsonBadRequest(errors),
    comment => Ok(toJson(comment).toString)
  )

}

ps:我知道避免退货声明会好得多......

1 个答案:

答案 0 :(得分:7)

首先要简化。假设我有三种方法需要String并返回Option[String]

def foo(s: String): Option[String] = if (s.size >= 4) Some(s + "1") else None
def bar(s: String): Option[String] = if (s(0) != 'A') Some(s + "2") else None
def baz(s: String): Option[String] = if (s toSet ' ') Some(s + "3") else None

我想要一个通过这些方法管道字符串的方法,如果我一路上得到None,则返回相应的错误消息。我可以这样写:

def all(s: String): Either[String, String] =
  foo(s).map { x =>
    bar(x).map { y =>
      baz(y).map { z =>
        Right(z)
      } getOrElse Left("Doesn't contain a space!")
    } getOrElse   Left("Starts with an A!")
  } getOrElse     Left("Too short!")

但是,这不是很好。我们可以在for上使用toRight - 理解和Option方法来编写更清晰的版本:

def all(s: String): Either[String, String] = for {
  x <- (foo(s) toRight "Too short!"              ).right
  y <- (bar(x) toRight "Starts with an A!"       ).right
  z <- (baz(y) toRight "Doesn't contain a space!").right
} yield z

如果toRight(msg)Option,则Left(msg)会为空{+ 1}},否则为Right(whatever)。然后我们必须使用Either.right进行正确的投影,因为Scala的Either没有正确偏见。

你的情况中的等价物将是这样的:

def save(ideaId: Long) = CORSAction { request =>
  val saveResult = for {
    idea    <- (Idea.findById(ideaId) toRight "Could not find id"     ).right
    json    <- (request.body.asJson   toRight "Invalid Comment entity").right
    comment <- (json.asOpt[Comment]   toRight "Expecting JSON data"   ).right
    result  <- comment.copy(idea = idea).save().right
  } yield result

  saveResult.fold(
    error => JsonBadRequest(error),
    comment => Ok(toJson(comment).toString)
  )
}

不像你想要的语法那么简洁,但错误消息出现在一个更合乎逻辑的地方,我们已经摆脱了丑陋的嵌套。