Mirrors/leptos
2
0
Fork 0
mirror of https://github.com/leptos-rs/leptos synced 2024-11-10 06:44:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

feat(axum): provide state to server fn context (#2257)

Browse source 
Note that this is a minimal implementation and will __not__ allow the
user to `expect_state` if they have external calls to rendering their
app (i.e. using `render_app_to_*` directly).
This commit is contained in:
Chris 2024-02-04 19:26:21 -05:00 committed by GitHub
parent 85c3755f6d
commit 15d4ca0638
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 15 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

22
integrations/axum/src/lib.rs
View file

@ -1594,6 +1594,14 @@ where
where
IV: IntoView + 'static,
{
// S represents the router's finished state allowing us to provide
// it to the user's server functions.
let state = options.clone();
let cx_with_state = move || {
provide_context::<S>(state.clone());
additional_context();
};
let mut router = self;
// register router paths
@ -1609,7 +1617,7 @@ where
path,
LeptosOptions::from_ref(options),
app_fn.clone(),
additional_context.clone(),
cx_with_state.clone(),
method,
static_mode,
)
@ -1629,7 +1637,7 @@ where
SsrMode::OutOfOrder => {
let s = render_app_to_stream_with_context(
LeptosOptions::from_ref(options),
additional_context.clone(),
cx_with_state.clone(),
app_fn.clone(),
);
match method {
@ -1643,7 +1651,7 @@ where
SsrMode::PartiallyBlocked => {
let s = render_app_to_stream_with_context_and_replace_blocks(
LeptosOptions::from_ref(options),
additional_context.clone(),
cx_with_state.clone(),
app_fn.clone(),
true
);
@ -1658,7 +1666,7 @@ where
SsrMode::InOrder => {
let s = render_app_to_stream_in_order_with_context(
LeptosOptions::from_ref(options),
additional_context.clone(),
cx_with_state.clone(),
app_fn.clone(),
);
match method {
@ -1672,7 +1680,7 @@ where
SsrMode::Async => {
let s = render_app_async_with_context(
LeptosOptions::from_ref(options),
additional_context.clone(),
cx_with_state.clone(),
app_fn.clone(),
);
match method {
@ -1691,9 +1699,9 @@ where
// register server functions
for (path, method) in server_fn::axum::server_fn_paths() {
let additional_context = additional_context.clone();
let cx_with_state = cx_with_state.clone();
let handler = move |req: Request<Body>| async move {
handle_server_fns_with_context(additional_context, req).await
handle_server_fns_with_context(cx_with_state, req).await
};
router = router.route(
path,