idgenerator_thin/
lib.rs

1pub mod idgen;
2pub use idgen::*;
3
4#[cfg(test)]
5mod tests {
6    use chrono::{DateTime, Utc};
7    use crate::{IdGeneratorOptions, YitIdHelper};
8
9    #[test]
10    fn test_next_id() {
11        // 创建 IdGeneratorOptions 对象,请在构造函数中输入 WorkerId:
12        let mut options = IdGeneratorOptions::new(1);
13        options.seq_step = 10;
14        // options.WorkerIdBitLength = 10; // WorkerIdBitLength 默认值6,支持的 WorkerId 最大值为2^6-1,若 WorkerId 超过64,可设置更大的 WorkerIdBitLength
15        // ...... 其它参数设置参考 IdGeneratorOptions 定义,一般来说,只要再设置 WorkerIdBitLength (决定 WorkerId 的最大值)。
16
17        let base_datetime = DateTime::<Utc>::from_timestamp(options.base_time / 1000, 0).unwrap();
18        println!("{:?}", base_datetime);
19
20        // 保存参数(必须的操作,否则以上设置都不能生效):
21        YitIdHelper::set_id_generator(options);
22        // 以上初始化过程只需全局一次,且必须在第2步之前设置。
23
24        // 初始化以后,即可在任何需要生成ID的地方,调用以下方法:
25        for _ in 0..100 {
26            let next_id = YitIdHelper::next_id();
27            println!("next_id: {}", next_id);
28            assert!(next_id > 0);
29        }
30    }
31}