scala - 如何以功能方式验证参数？

Question

我正在编写 Scala/Play 2.0 应用程序，我希望我的代码根据请求返回不同的页面。这是我的代码:

// Validate client and return temporary credentials
def requestToken = Action { request =>
  // Authorization header may present or not
  val authHeader = parseHeaders(request headers AUTHORIZATION)
  // Authorization header may contain such keys or not
  val clientKey = authHeader("oauth_consumer_key")
  val signature = authHeader("oauth_signature")

  if (authenticateClient(clientKey, signature)) {
    ...
    Ok(...)
  } else {
    Unauthorized(...)
  }

}

当请求格式错误并且缺少某些 header 时会出现问题，然后抛出 NoSuchElementException。

在命令式语言中，我会像这样验证每一步:

if (!request.headers.contains(AUTHORIZATION))
  return Unathorized

val authHeader = parseHeaders(request headers AUTHORIZATION)

if (!authHeader.contains("oauth_consumer_key") || !authHeader.contains("oauth_signature"))
  return Unathorized

val clientKey = authHeader("oauth_consumer_key")
val signature = authHeader("oauth_signature")

...

但是我应该怎么做才能以功能方式解决这样的问题？

Answer 1

您可以使用 authHeader.get(key)它返回一个选项[B]。您的代码将如下所示:

val result = for {
  auth <- request.headers.get(AUTHORIZATION)
  authHeader = parseHeaders(auth)
  clientKey <- authHeader.get("oauth_consumer_key")
  signature <- authHeader.get("oauth_signature")
} yield { ... }

result.getOrElse(Unauthorized)

解释:

只有在没有 None 的情况下才会执行整个表达式值(value)。因此，如果所有标题都存在，您将获得 result的 Some[A]如果缺少一个或多个，您将收到 None . result.getOrElse(Unauthorized)将返回 Some[A] 的包含值和 Unauthorized为 None .