axum/axum-macros/tests/from_request/pass/state_infer.rs at ba8e9c1b212377631d43718246d01ba92ae02ec7 - OpenSource/axum - Moe's Git

OpenSource/axum

mirror of https://github.com/tokio-rs/axum.git synced 2024-11-23 07:39:25 +01:00

David Pedersen c3f3db79ec

Support State with #[derive(FromRequest[Parts])] (#1391 )

* Support `State` with `#[derive(FromRequest[Parts])]`

Fixes https://github.com/tokio-rs/axum/issues/1314

This makes it possible to extract things via `State` in
`#[derive(FromRequet)]`:

```rust
struct Foo {
    state: State<AppState>,
}
```

The state can also be inferred in a lot of cases so you only need to
write:

```rust
struct Foo {
    // since we're using `State<AppState>` we know the state has to be
    // `AppState`
    state: State<AppState>,
}
```

Same for

```rust
struct Foo {
    #[from_request(via(State))]
    state: AppState,
}
```

And

```rust
struct AppState {}
```

I think I've covered all the edge cases but there are (unsurprisingly) a
few.

* make sure things can be combined with other extractors

* main functions in ui tests don't need to be async

* Add test for multiple identicaly state types

* Add failing test for multiple states

2022-09-23 23:50:50 +02:00

18 lines

331 B

Rust

Raw Blame History

 use axum_macros::FromRequest;
 use axum::extract::State;
 #[derive(FromRequest)]
 struct Extractor {
     inner_state: State<AppState>,
 }
 #[derive(Clone)]
 struct AppState {}
 fn assert_from_request()
 where
     Extractor: axum::extract::FromRequest<AppState, axum::body::Body, Rejection = axum::response::Response>,
 {
 }
 fn main() {}