rust_cutil/cutil/
generator.rs

1use crate::cutil::meta::R;
2use async_trait::async_trait;
3use idgenerator_thin::{IdGeneratorOptions, YitIdHelper};
4use once_cell::sync::Lazy;
5use rand::distributions::Alphanumeric;
6use rand::{thread_rng, Rng};
7
8#[async_trait]
9pub trait Generator: Send + Sync {
10  async fn next_id(&self) -> R<i64>;
11}
12
13pub struct GeneratorImpl {}
14
15impl GeneratorImpl {
16  pub fn new() -> Self {
17    Self {}
18  }
19}
20
21impl GeneratorImpl {
22  fn configure(&self) {
23    let mut options = IdGeneratorOptions::new(1);
24    options.worker_id_bit_length = 1;
25    options.base_time = 1710034500000;
26    YitIdHelper::set_id_generator(options);
27  }
28}
29
30#[async_trait]
31impl Generator for GeneratorImpl {
32  async fn next_id(&self) -> R<i64> {
33    Ok(YitIdHelper::next_id())
34  }
35}
36
37static SHARED: Lazy<GeneratorImpl> = Lazy::new(|| {
38  let gen = GeneratorImpl::new();
39  gen.configure();
40  gen
41});
42
43pub async fn next_id() -> R<i64> {
44  SHARED.next_id().await
45}
46
47pub fn rand_string(length: usize) -> String {
48  let rand_string: String = thread_rng().sample_iter(&Alphanumeric).take(length).map(char::from).collect();
49  rand_string
50}
rust_cutil/cutil/generator.rs

rust_cutil/cutil/
generator.rs
