rust - 如何使用 Warp 检查授权标头？

Question

我正在用 Rust 和 Warp 构建一个 graphql api。我查看了文档，但我仍然没有弄清楚如何链接过滤器，特别是检查authorization请求标头。

let context_extractor = warp::any()
    // this code rejects all request which doesn't contain the authorization in header
    // I'd like to make to check if authorization in header
    .and(warp::header::<String>("authorization"))
    .map(|token: String| -> Context {
        let token_data = match verify_jwt(token) {
            Ok(t) => t,
            Err(_) => return Context { user_id: 0 },
        };

        Context {
            user_id: token_data.claims.user_id,
        }
    });

let handle_request = move |context: Context,
                           request: juniper::http::GraphQLRequest|
      -> Result<Vec<u8>, serde_json::Error> {
    serde_json::to_vec(&request.execute(&schema, &context))
};

warp::post2()
    .and(warp::path(path.into()))
    .and(context_extractor)
    .and(warp::body::json())
    .map(handle_request)
    .map(build_response)
    .boxed()

这是我的部分代码。它工作正常，但有一个问题。我用 设置了一条路由context_extractor，.and(warp::header::<String>("authorization")然后它拒绝所有不包含authorization在标头中的请求。

我该怎么做

1. 如果请求标头有一个authorizationin 标头，则返回Context正确的user_id

2. Context如果没有，返回user_id: 0?

Answer 1

我在Warp 的 github 问题中找到了解决方案。

这是一个小片段。

let context_extractor = warp::any().and(
    warp::header::<String>("authorization")
        .map(|token: String| -> Context {
            let token_data = match verify_jwt(token) {
                Ok(t) => t,
                Err(_) => return Context { user_id: 0 },
            };

            Context {
                user_id: token_data.claims.user_id,
            }
        })
        .or(warp::any().map(|| Context { user_id: 0 }))
        .unify(),
);