Skip to content

Commit 8849824

Browse files
committed
Still generate native registration functions
1 parent 46be42f commit 8849824

1 file changed

Lines changed: 14 additions & 11 deletions

File tree

node-graph/node-macro/src/codegen.rs

Lines changed: 14 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -678,22 +678,25 @@ fn generate_register_node_impl(parsed: &ParsedNodeFn, field_names: &[&Ident], st
678678
}
679679
let registry_name = format_ident!("__node_registry_{}_{}", NODE_ID.fetch_add(1, std::sync::atomic::Ordering::SeqCst), struct_name);
680680

681+
let native = quote! {
682+
#[cfg_attr(not(target_family = "wasm"), ctor)]
683+
fn register_node() {
684+
let mut registry = NODE_REGISTRY.lock().unwrap();
685+
registry.insert(
686+
#identifier(),
687+
vec![
688+
#(#constructors,)*
689+
]
690+
);
691+
}
692+
};
681693
if cfg!(feature = "disable-registration") {
682-
return Ok(quote!());
694+
return Ok(native);
683695
}
684696

685697
Ok(quote! {
698+
#native
686699

687-
#[cfg_attr(not(target_family = "wasm"), ctor)]
688-
fn register_node() {
689-
let mut registry = NODE_REGISTRY.lock().unwrap();
690-
registry.insert(
691-
#identifier(),
692-
vec![
693-
#(#constructors,)*
694-
]
695-
);
696-
}
697700
#[cfg(target_family = "wasm")]
698701
#[unsafe(no_mangle)]
699702
extern "C" fn #registry_name() {

0 commit comments

Comments
 (0)