JakeHillion/scx-upstream
1
0
Fork 0
mirror of https://github.com/sched-ext/scx.git synced 2024-11-25 04:00:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

scx_stats_derive: Each _AssScxStastMeta assertion should have an unique ID

Browse Source
This commit is contained in:
Tejun Heo 2024-08-22 13:51:37 -10:00
parent 16c07a5cd9
commit 4678476ca7
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
rust/scx_stats/scx_stats_derive/src/lib.rs
View File

@ -1,8 +1,11 @@
use quote::{quote, quote_spanned};
use quote::{format_ident, quote, quote_spanned};
use scx_stats::{ScxStatsData, ScxStatsKind, ScxStatsMetaAux};
use std::sync::atomic::{AtomicU64, Ordering};
use syn::parse_macro_input;
use syn::spanned::Spanned;
static ASSERT_IDX: AtomicU64 = AtomicU64::new(0);
#[proc_macro_derive(Stats, attributes(stat))]
pub fn stat(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
let stats_aux = parse_macro_input!(input as ScxStatsMetaAux);
@ -17,9 +20,11 @@ pub fn stat(input: proc_macro::TokenStream) -> proc_macro::TokenStream {
| ScxStatsData::Dict { key: _, datum } => {
if let ScxStatsKind::Struct(name) = &datum {
let path = &paths[name.as_str()];
let idx = ASSERT_IDX.fetch_add(1, Ordering::Relaxed);
let assert_id = format_ident!("_AssertScxStatsMeta_{}", idx);
#[rustfmt::skip]
let assert = quote_spanned! {path.span()=>
struct _AssertScxStatsMeta where #path: scx_stats::Meta;
struct #assert_id where #path: scx_stats::Meta;
};
output.extend(assert.into_iter());
}