在actix-web中处理异步处理程序错误的最惯用的方法是什么?

时间:2019-06-02 17:23:58

标签: rust future actix-web

我在actix_web中有一个异步处理程序,如果未设置多个标头,该处理程序将失败。我不知道应该如何处理返回Future的函数中的错误的最佳方法。我基本上想要与?运算符等效的期货。

这是我当前的代码:

r.post().with_async(
    move |req: HttpRequest, path: Path<EventPath>, body: Json<EventCreationRequest>| {
        let headers = req.headers();
        let client_id = match headers
            .get("x-client-id")
            .ok_or("Header not found")
            .and_then(|v| v.to_str().map_err(|_| "Invalid header content"))
        {
            Err(e) => return ok(HttpResponse::BadRequest().body(e)).responder(),
            Ok(v) => v.to_string(),
        };
        operation_that_returns_future()
            .map(|_| HttpResponse::Ok().body("OK!"))
            .responder()
    },
);

我通过匹配早期回报来解决了期货缺乏?运算符的问题。但是,在我的代码中,我实际上需要确保存在许多其他标头。

理想情况下,我想将匹配和早期返回逻辑提取到可重用的东西中,但是在这种情况下,这迫使我创建一个宏。这似乎有点过大,尤其是如果该语言中已有某种东西可以让我做自己想做的事。

处理这种情况的最惯用的方法是什么?

1 个答案:

答案 0 :(得分:1)

要处理错误,请返回失败的Future。例如,将标头检查为Future,然后将期货与.and_then链接起来。一个技巧是保持期货的错误类型相同,以避免map_err。例如:

fn handler(req: HttpRequest) -> impl Future<Item = HttpResponse, Error = Error> {
    has_client_header(&req)
        .and_then(|client| operation_that_returns_future(client))
        .map(|result| HttpResponse::Ok().body(result))
}

fn has_client_header(req: &HttpRequest) -> impl Future<Item = String, Error = Error> {
    if let Some(Ok(client)) = req.headers().get("x-client-id").map(|h| h.to_str()) {
        future::ok(client.to_owned())
    } else {
        future::failed(ErrorBadRequest("invalid x-client-id header"))
    }
}

fn operation_that_returns_future(client: String) -> impl Future<Item = String, Error = Error> {
    future::ok(client)
}

结果:

$ curl localhost:8000
invalid x-client-id header⏎
$ curl localhost:8000 -H 'x-client-id: asdf'
asdf⏎

operation_that_returns_future具有另一种错误类型时:

fn handler(req: HttpRequest) -> impl Future<Item = HttpResponse, Error = Error> {
    has_client_header(&req)
        .and_then(|client| {
            operation_that_returns_future(client)
                .map_err(|_| ErrorInternalServerError("operation failed"))
        })
        .map(|result| HttpResponse::Ok().body(result))
}

另一种技巧是使用failure crate,它提供了failure::Error::from并将所有错误映射到一种类型failure::Error

最后,您可能会发现actix_web::guards对检查标头值很有用:

.guard(guard::Header("x-client-id", "special client"))