mirror/rtic
1
0
Fork 0
mirror of https://github.com/rtic-rs/rtic.git synced 2025-12-16 21:05:35 +01:00
Code Issues Projects Releases Packages Wiki Activity

Make generated names stable when sorting.

Browse source
This commit is contained in:
Hugo van der Wijst 2019-02-10 15:25:33 -08:00 • committed by Jorge Aparicio
parent a654d13eef
commit 577d188f72
2 changed files with 35 additions and 12 deletions
Download patch file Download diff file Expand all files Collapse all files

35
macros/src/codegen.rs
View file

@ -3,6 +3,7 @@
use proc_macro::TokenStream;
use std::{
collections::{BTreeMap, HashMap},
time::{SystemTime, UNIX_EPOCH},
};
use proc_macro2::Span;
@ -1993,30 +1994,48 @@ fn mk_typenum_capacity(capacity: u8, power_of_two: bool) -> proc_macro2::TokenSt
}
struct IdentGenerator {
call_count: u32,
rng: rand::rngs::SmallRng,
}
impl IdentGenerator {
fn new() -> IdentGenerator {
let crate_name = env!("CARGO_PKG_NAME");
let seed = [0u8; 16];
for (i, b) in crate_name.bytes().enumerate() {
seed[i%seed.len()].wrapping_add(b);
let elapsed = SystemTime::now().duration_since(UNIX_EPOCH).unwrap();
let secs = elapsed.as_secs();
let nanos = elapsed.subsec_nanos();
let mut seed: [u8; 16] = [0; 16];
for (i, v) in seed.iter_mut().take(8).enumerate() {
*v = ((secs >> (i * 8)) & 0xFF) as u8
}
for (i, v) in seed.iter_mut().skip(8).take(4).enumerate() {
*v = ((nanos >> (i * 8)) & 0xFF) as u8
}
let rng = rand::rngs::SmallRng::from_seed(seed);
IdentGenerator {
call_count: 0,
rng,
}
IdentGenerator { rng: rand::rngs::SmallRng::from_seed(seed) }
}
fn mk_ident(&mut self, name: Option<&str>) -> Ident {
let n;
let mut s = if let Some(name) = name {
let s = if let Some(name) = name {
n = 4;
format!("{}_", name)
} else {
let crate_name = env!("CARGO_PKG_NAME").replace("-", "_").to_lowercase();
n = 16;
n = 4;
format!("{}__internal__", crate_name)
};
let mut s = format!("{}{}_", s, self.call_count);
for i in 0..n {
if i == 0 || self.rng.gen() {
s.push(('a' as u8 + self.rng.gen::<u8>() % 25) as char)
@ -2025,6 +2044,8 @@ impl IdentGenerator {
}
}
self.call_count += 1;
Ident::new(&s, Span::call_site())
}
}