rust - 如何将结构从 Actix 中间件传递给处理程序？

Question

我正在尝试为我的 Actix 应用程序编写身份验证中间件。在中间件中验证请求时，我调用数据库以检索必要的用户数据以验证传入请求。一旦请求被授权，我希望能够将此用户数据传递给处理程序，因为这将允许我避免两次查询相同的数据。

我找不到解决方案。到目前为止我能找到的最好的建议是“设置请求扩展”。似乎没有任何示例，并且围绕此的文档也太少，无法确定在这里做什么。

Answer 1

您可以通过扩展将数据从中间件（服务）传递到处理程序。首先，您必须插入扩展名（服务中）。对于 ServiceRequest 结构，实现了 HttpMessage 巫婆extensions_mut()功能。它必须是可变的，因为您将插入新的扩展。它可能看起来像这样：

req.extensions_mut().insert(user);

然后你必须为你的数据结构实现 FromRequest 特征。

impl FromRequest for User {
    type Error = actix_web::Error;
    type Future = futures::future::Ready<Result<Self, Self::Error>>;
    type Config = ();

    fn from_request(req: &HttpRequest, payload: &mut Payload) -> Self::Future {
        match req.extensions().get::<User>() {
            Some(user) => return ok(user.clone()),
            None => return err(actix_web::error::ErrorBadRequest("ups..."))
        };
    }

}

然后你就可以在处理程序中使用它了。

pub async fn get_user_handler(user: User) {}