fix(codegen): #[system] should copy attributes to generated functions…

… too

codegen/src/lib.rs

@@ -1,4 +1,5 @@
 #![feature(extract_if)]
+#![allow(dead_code)]
 
 use proc_macro::TokenStream;
 

codegen/src/system.rs

@@ -22,7 +22,6 @@ pub(crate) fn imp(args: TokenStream, input: TokenStream) -> Result<TokenStream>
         return Err(Error::new_spanned(&input.sig.output, "system functions must return unit"));
     }
 
-    // 1. Parse item-level attributes.
     let item = item::Agg::parse(ident, args)?;
     let item::Agg { crate_name, name, state_maybe_uninit, deps, .. } = item;
 
@@ -384,6 +383,7 @@ pub(crate) fn imp(args: TokenStream, input: TokenStream) -> Result<TokenStream>
             #[automatically_derived]
             impl #ident {
                 #build_fn
+                #[allow(clippy::too_many_arguments)]
                 #call_fn
             }
 
@@ -393,6 +393,7 @@ pub(crate) fn imp(args: TokenStream, input: TokenStream) -> Result<TokenStream>
             #impl_referrer_for_local_state
 
             // The actual function is moved here.
+            #(#other_attrs)*
             fn __dynec_original(#(#input_args),*) {
                 #fn_body
             }

codegen/src/tracer_def.rs

@@ -315,7 +315,7 @@ impl Parse for Named<AssocFnOpt> {
     }
 }
 
-#[allow(clippy::enum_variant_names)] // TODO remove this when we add more implementations
+#[allow(clippy::enum_variant_names)]
 enum AssocFnParamOpt {
     LogSkip,
     LogElapsedDuration,

rust-toolchain

@@ -1 +1 @@
-nightly-2023-11-21
+nightly-2024-01-31

src/entity/ealloc/recycling.rs

@@ -2,7 +2,8 @@ use std::collections::BTreeSet;
 use std::sync::Arc;
 use std::{iter, ops};
 
-use parking_lot::Mutex;
+use parking_lot::lock_api::ArcMutexGuard;
+use parking_lot::{Mutex, RawMutex};
 
 use super::{iter_gaps, Ealloc, Shard, ShardAssigner, Snapshot};
 use crate::entity::raw::Atomic;
@@ -72,7 +73,11 @@ impl<RawT: Raw, T: Recycler<RawT>, S: ShardAssigner> Recycling<RawT, T, S> {
 impl<RawT: Raw, T: Recycler<RawT>, S: ShardAssigner> Ealloc for Recycling<RawT, T, S> {
     type Raw = RawT;
     type AllocHint = T::Hint;
-    type Shard = impl Shard<Raw = RawT, Hint = T::Hint>;
+    type Shard = RecyclingShard<
+        Arc<RawT::Atomic>,
+        ArcMutexGuard<RawMutex, T>,
+        ArcMutexGuard<RawMutex, Vec<RawT>>,
+    >;
 
     fn new(num_shards: usize) -> Self { Self::new_with_shard_assigner(num_shards, S::default()) }
 
@@ -81,13 +86,10 @@ impl<RawT: Raw, T: Recycler<RawT>, S: ShardAssigner> Ealloc for Recycling<RawT,
 
         vec.extend(
             iter::zip(self.recycler_shards.iter(), self.reuse_queue_shards.iter())
-                .map(|(recycler, reuse_queue)| -> Self::Shard {
-                    // The return type hint here is used to constrain the TAIT, don't delete it.
-                    RecyclingShard {
-                        global_gauge: Arc::clone(&self.global_gauge),
-                        recycler:     Arc::clone(recycler).lock_arc(),
-                        reuse_queue:  Arc::clone(reuse_queue).lock_arc(),
-                    }
+                .map(|(recycler, reuse_queue)| RecyclingShard {
+                    global_gauge: Arc::clone(&self.global_gauge),
+                    recycler:     Arc::clone(recycler).lock_arc(),
+                    reuse_queue:  Arc::clone(reuse_queue).lock_arc(),
                 })
                 .map(f),
         );