7

我正在编写 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(...)
  }

}

当请求格式错误并且缺少某些标头时会出现问题，然后引发 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")

...

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

4

1 回答 1

16

您可以使用authHeader.get(key)which 返回一个 Option[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只有在没有值的情况下才会执行整个表达式。因此，如果所有标题都存在，您将得到一个resultof Some[A]，如果一个或多个不存在，您将得到None. result.getOrElse(Unauthorized)将返回包含的值 forSome[A]和Unauthorizedfor None。

于 2012-06-15T06:41:25.433 回答