How to use PrivateCookieJar in middleware

[michaelkimchi]

I have been trying to use PrivateCookieJar to manage cookie but wasn't able to get it compile.

My middleware looks like this

pub async fn cookie_middleware<B>(
    cookie_jar: PrivateCookieJar,
    mut request: Request<B>,
    next: Next<B>,
) -> impl IntoResponse {

I am adding as a layer like this.

.layer(ServiceBuilder::new().layer(axum::middleware::from_fn(cookie_middleware)));

I got this error

^ the trait `Service<axum::http::Request<_>>` is not implemented for 
`axum::middleware::FromFn<fn(axum_extra::extract::cookie::private::PrivateCookieJar, axum::http::Request<_>, Next<_>) -> 
impl std::future::Future<Output = impl IntoResponse> {cookie_middleware::<_>}, (), Route, _>`

It compiles again when I remove cookie_jar: PrivateCookieJar

[michaelkimchi]

I managed to get it compiled after adding an app_state and do impl FromRef<AppState> for Key

debug_middleware might be helpful in this case. #1993

[davidpdrsn]

Relevant section in docs: https://docs.rs/axum-extra/0.7.7/axum_extra/extract/cookie/struct.PrivateCookieJar.html#example