rust 如何将结构从Actix中间件传递到处理程序?

wgmfuz8q  于 2023-10-20  发布在  其他
关注(0)|答案(1)|浏览(115)

我正在尝试为我的Actix应用程序编写一个身份验证中间件。当在中间件中验证请求时,我调用数据库来检索必要的用户数据以验证传入的请求。一旦请求被授权,我希望能够将此用户数据传递给处理程序,因为这将允许我避免对同一数据进行两次查询。
我找不到解决办法。到目前为止,我能找到的最好的建议是"set a request extension"。似乎没有任何例子,也有太少的文档围绕这一点,以工作在这里做什么。

cyvaqqii

cyvaqqii1#

您可以通过扩展将数据从中间件(服务)传递到处理程序。
首先,您必须插入扩展(在服务中)。HttpMessage trait是为ServiceRequest结构体实现的,提供了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..."))
        };
    }

}

然后你就可以在handler中使用它了。

pub async fn get_user_handler(user: User) {}

相关问题