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

问题描述

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

我找不到解决方案。到目前为止，我能找到的最佳建议是"set a request extension"。似乎没有任何示例，有关此的文档也太少，无法在这里确定要做什么。

解决方法

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

req.extensions_mut().insert(user);

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

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) {}