spray - 如何在自定义指令中获取请求上下文？

Question

我正在用 Spray 编写一个自定义指令，用于管理任何用户请求的速率限制。

我要一个 LimitManager某个地方将处理每个请求的自定义限制和规则。这个唯一需要的信息LimitManager是 userInfo和 currentRoute ，即不同路线的不同限制。

所以我可能最终会有这样的指令:

def ensureLimit(): Directive0 =
  if (LimitManager.isAuthorized(userInfo, currentRoute)) {
    pass
  } else {
    reject
  }

如何在指令中获取请求上下文，以便向我的 LimitManager 提供正确的信息?

Answer 1

在 Spray 中，每个指令都是 Route 上的一个函数，它是函数 RequestContext => Unit 的别名.但是在强大的 Scala 隐含的帮助下，路由 DSL 有助于隐藏这一点，但您可以编写如下内容:

val route: Route = get { (ctx: RequestContext) => // this can be omitted, just for info
  ctx.complete("Hello")
}

它与:

val route: Route = get { complete("Hello") }

但是有一些复杂的语法技巧。

记住! 你永远不应该这样写:

val route = get { ctx =>
  complete("Alloha")
}

在这里 complete将转换为 ctx => ctx.complete("Hello") ，因此您将根据您的请求返回此函数，并且不会完成真正的请求。

因此，一种方法可以做到这一点，只需将其作为参数传递即可。您也可以使用 extract指令，获取上下文然后使用 map或 flatMap制作你自己的:

val myDirective = extract(identity) map { ctx => /* Your directive */ }