mirrors/leptos
1
0
Fork 0
mirror of https://github.com/leptos-rs/leptos.git synced 2025-12-27 11:04:40 -05:00
Code Issues Packages Projects Releases Wiki Activity

feat: support both sync and async lazy functions

Browse Source
This commit is contained in:
Greg Johnston
2025-07-19 08:21:55 -04:00
parent 10c13bbca2
commit 232b603a25
4 changed files with 80 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
leptos_macro/src/lazy.rs
View File

@@ -7,7 +7,7 @@ use std::{
hash::{DefaultHasher, Hash, Hasher},
mem,
};
use syn::{parse_macro_input, spanned::Spanned, ItemFn};
use syn::{parse_macro_input, ItemFn};
pub fn lazy_impl(args: proc_macro::TokenStream, s: TokenStream) -> TokenStream {
let name = if !args.is_empty() {
@@ -19,12 +19,8 @@ pub fn lazy_impl(args: proc_macro::TokenStream, s: TokenStream) -> TokenStream {
let mut fun = syn::parse::<ItemFn>(s).unwrap_or_else(|e| {
abort!(e.span(), "`lazy` can only be used on a function")
});
if fun.sig.asyncness.is_none() {
abort!(
fun.sig.asyncness.span(),
"`lazy` can only be used on an async function"
)
}
let was_async = fun.sig.asyncness.is_some();
let converted_name = name.unwrap_or_else(|| {
Ident::new(
@@ -56,6 +52,10 @@ pub fn lazy_impl(args: proc_macro::TokenStream, s: TokenStream) -> TokenStream {
#fun
}
} else {
if !was_async {
fun.sig.asyncness = Some(Default::default());
}
let statements = &mut fun.block.stmts;
let old_statements = mem::take(statements);
statements.push(