3 releases

0.1.3 Jul 16, 2022
0.1.2 Jul 15, 2022
0.1.1 Jul 14, 2022
0.1.0 Jul 14, 2022

#834 in Development tools

MIT license

25KB
432 lines

rid

Crate API License

A practical distributed id generator by rust, Snowflake based unique ID generator. It works as a component, and allows users to override workId bits and initialization strategy. As a result, it is much more suitable for virtualization environment, such as docker.

Snowflake

Snowflake algorithm: An unique id consists of worker node, timestamp and sequence within that timestamp. Usually, it is a 64 bits number(long), and the default bits of that three fields are as follows:

+------+----------------------+----------------+-----------+
| sign |     delta seconds    | worker node id | sequence  |
+------+----------------------+----------------+-----------+
  1bit          30bits              20bits         13bits

sign(1bit) The highest bit is always 0.

delta seconds (30 bits) The next 30 bits, represents delta seconds since a customer epoch(2016-05-20). The maximum time will be 34 years.

worker id (20 bits) The next 20 bits, represents the worker node id, maximum value will be 1.04 million. UidGenerator uses a build-in database based worker id assigner when startup by default, and it will reuse previous work node id after reboot.

sequence (13 bits) the last 13 bits, represents sequence within the one second, maximum is 8192 per second(per server)by default.

Features

  • light and easy to use
  • distributed id generator at local instead of by service or rpc
  • worker id persistence solution (in database like mysql instead of cache storage)
  • support clock moved backwards(can be disabled)
  • support id length customization lower than 64 bits

Design

Quick Start

Step1: Install rust, Mysql

Step2: Create table worker_node

DROP TABLE IF EXISTS `worker_node`;
CREATE TABLE `worker_node` (
  `id` bigint unsigned NOT NULL AUTO_INCREMENT COMMENT 'auto increment id',
  `host_name` varchar(64) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NOT NULL COMMENT 'host name',
  `port` varchar(64) CHARACTER SET utf8mb4 COLLATE utf8mb4_general_ci NOT NULL COMMENT 'port',
  `worker_type` int NOT NULL COMMENT 'node type: CONTAINER(1), ACTUAL(2), FAKE(3)',
  `modified` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP ON UPDATE CURRENT_TIMESTAMP COMMENT 'modified time',
  `created` timestamp NOT NULL DEFAULT CURRENT_TIMESTAMP COMMENT 'created time',
  PRIMARY KEY (`id`),
  UNIQUE KEY `UNIQ_IDX_HOST_PORT` (`host_name`,`port`) USING BTREE COMMENT 'host和端口的唯一索引'
) ENGINE=InnoDB AUTO_INCREMENT=7 DEFAULT CHARSET=utf8mb4 COLLATE=utf8mb4_general_ci COMMENT='DB WorkerID Assigner for UID Generator';

Step3: Install Lib

Add this to your Cargo.toml:

[dependencies]
rust-distributed-id = "0.1.2"

Step4: Usage

let config = rid_config::UidConfig::new("5000".to_string());
let rb: Rbatis = Rbatis::new();
rb.link("mysql://root:root@127.0.0.1:3306/test")
.await
.expect("Couldn't open database");
let mut idg = rid_generator::UidGenerator::new(&config, Arc::new(rb)).await;

let start = Local::now().timestamp_millis();
for _ in 1..10000 {
    //println!("{}", &idg.get_uid());
    let _ = idg.get_uid();
}

Customization

Change the time_bits, worker_bits, seq_bits of 'UidConfig' to get your customer uid, especially shorter uid.

let mut config = rid_config::UidConfig::new("5000".to_string());
config.worker_bits = 10;
config.seq_bits = 23;

let rb: Rbatis = Rbatis::new();
rb.link("mysql://root:root@127.0.0.1:3306/test")
.await
.expect("Couldn't open database");
let mut idg = rid_generator::UidGenerator::new(&config, Arc::new(rb)).await;

let start = Local::now().timestamp_millis();
for _ in 1..1000000 {
    //println!("{}", &idg.get_uid());
    let _ = idg.get_uid();
}

ChangeLog

License

rid is MIT licensed.

Dependencies

~26–39MB
~768K SLoC