Rust学习笔记 --雪花ID生成

雪花ID现在应该大家用的都不少了，雪花ID的作用是生成唯一标识符，一般用在生成订单上面，使订单不重复，但用rust应该怎么写一个雪花ID的生成器呢，下面是我的学习思路，首先我们要导入相关的依赖库rs-snowflake，在cargo.toml导入这个依赖。

要生成雪花ID的话需要用到机器码，我从redis里面捞取机器码，redis的安装需要自己下载安装，不过redis目前在Windows的安装是通过第三方的，所以我选择把redis装在了阿里云ECS服务器里面，安装好之后，就可以往redis里面存入值来捞取机器码了，为了用rust连接上redis，需要导入redis这个依赖库，redis的连接代码如下

ini 复制代码

let client = redis::Client::open(redis://default:password@localhost:6379/).unwrap();
let mut connection = client.get_async_connection().await.unwrap();

一般redis用户默认名为default，测试连接通过之后，就可以编写代码来捞取机器码，并通过机器码来生成雪花ID了，废话不多说，直接上代码

rust 复制代码

use snowflake::SnowflakeIdBucket;
use std::sync::Mutex;
use tokio::sync::OnceCell;
use crate::service:: CONTEXT;

static SNOWFLAKE_BUCKET: OnceCell<Mutex<SnowflakeIdBucket>> = OnceCell::const_new();

pub struct SonwflakeUtil {}

impl SonwflakeUtil {         
    async fn get_machine_id_from_redis() -> i32 {
        CONTEXT.cache_service.get_string("MACHINE_ID")
            .await
            .expect("faild")
            .parse::<i32>()
            .expect("faild")
        }  
   pub async fn get_id() -> u64 {
        SNOWFLAKE_BUCKET.get_or_init(|| async{
                let machine_id = SonwflakeUtil::get_machine_id_from_redis().await;
                Mutex::new(SnowflakeIdBucket::new(machine_id, 0)
            )})
            .await
            .lock()
            .expect("faild get_id")
            .get_id() as u64
        }
}

我的逻辑是创建一个全局静态变量SNOWFLAKE_BUCKET来存储我的SnowflakeIdBucket，SnowflakeIdBucket的作用是存储获取的机器码，我在创建SNOWFLAKE_BUCKET的时候使用了OnceCell，这个是为了能够调用get_or_init（）这个函数，我调用这个函数的作用是什么呢，是因为我想只初始化一次SnowflakeIdBucket，意思就是只捞取一次机器码，后面就把捞取下来的机器码保存下来复用，还用到Mutex互斥锁，是为了避免有多个对象同时竞争同一资源的事件发生，也就是数据竞争，

rust 复制代码

 async fn get_machine_id_from_redis() -> i32 {
        CONTEXT.cache_service.get_string("MACHINE_ID")
            .await
            .expect("faild")
            .parse::<i32>()
            .expect("faild")
        }

这一段代码是我从redis里面捞取机器码的过程，get_string()这个端口是连接redis并获取"MACHINE_ID"这个对象的机器码的，是已经编写好的一个端口，涉及到比较多文件，就不细讲了，实现这个的原理还是比较简单的，首先要连接上自己的redis，例子上面有，连接上之后，用get（）函数获取对象的值，例如

ini 复制代码

value = redis_client.get('my_key')

获取到值后把值转换成i32就是你拿到的机器码了，拿到机器码后把机器码存到SnowflakeIdBucket中，

php 复制代码

let machine_id = SonwflakeUtil::get_machine_id_from_redis().await;
                Mutex::new(SnowflakeIdBucket::new(machine_id, 0)

因为我是只拿一次机器码，所以我调用了get_or_init()这个函数，

rust 复制代码

 SNOWFLAKE_BUCKET.get_or_init(|| async{
                let machine_id = SonwflakeUtil::get_machine_id_from_redis().await;
                Mutex::new(SnowflakeIdBucket::new(machine_id, 0)
            )})

闭包里的语句就是将获得的机器码初始化，存进SnowflakeIdBucket里面，这些语句只在get_or_init（）第一次执行时启动，后面都是只会获取SnowflakeIdBucket里面的值，最后可以直接通过get_id()这个函数来直接生成雪花ID了

rust 复制代码

 .get_id() as u64

最后的测试生成结果如下

以上仅为个人学习总结，欢迎大家一起来讨论