idworker 1.1.1

A high-performance distributed ID generator library implementing Snowflake algorithm variants with multiple optimization modes for different performance requirements.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30

use crate::{IdWorker, IdWorkerConfig, IdWorkerError, IdWorkerGenerator};
use tracing::debug;
use std::sync::{Arc, RwLock};

static ID_WORKER: RwLock<Option<Arc<dyn IdWorker>>> = RwLock::new(None);

/// 获取当前配置的只读访问
pub fn get_id_worker() -> Result<Arc<dyn IdWorker>, IdWorkerError> {
    let read_lock = ID_WORKER.read().map_err(|_| IdWorkerError::GetIdWorker())?;
    read_lock
        .as_ref()
        .map(Arc::clone)
        .ok_or(IdWorkerError::GetIdWorker())
}

/// 设置配置
pub fn set_id_worker(value: Arc<dyn IdWorker>) -> Result<(), IdWorkerError> {
    let mut write_lock = ID_WORKER
        .write()
        .map_err(|_| IdWorkerError::SetIdWorker())?;
    *write_lock = Some(value);
    Ok(())
}

/// 初始化id生成器
pub fn init_id_worker(id_worker_config: IdWorkerConfig) -> Result<(), IdWorkerError> {
    debug!("init id worker...");
    let id_worker = IdWorkerGenerator::generate(id_worker_config)?;
    set_id_worker(id_worker)
}