feat: add object id(test) (#4)

Cargo.toml

@@ -1,6 +1,6 @@
 [package]
 name = "fastid"
-version = "0.0.3"
+version = "0.0.4"
 edition = "2021"
 
 # See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
@@ -15,6 +15,8 @@ ulid="1.1.0"
 pyo3="0.20.0"
 rs-snowflake = "0.6.0"
 once_cell = "1.19.0"
+rand = "0.8.5"
+hex = "0.4.3"
 
 [dependencies.uuid]
 version = "1.6.1"

README.md

@@ -4,4 +4,4 @@ Created for the easy and fast generation of various IDs
 # Supported IDs
 - ulid
 - snowflake
-- uuid v7 (If you intend to use uuid in your database, this can be a great choice)
+- uuid v7 (If you intend to use uuid in your database, this can be a great choice)

fastid.pyi

@@ -12,3 +12,7 @@ def snowflake_int(machine_id: int = 1, node_id: int = 1) -> int:
 
 def uuid_v7() -> str:
     ...
+
+
+def object_id() -> str:
+    """test"""

python/test/test_fastid.py

@@ -1,13 +1,14 @@
 import timeit
 import uuid
 
+import bson
 import fastid
-from ulid import microsecond
+import ulid
 
 
 def test_ulid():
     print()
-    print("ulid-py    ", timeit.timeit(lambda: microsecond.new().str, number=10000))
+    print("ulid-py    ", timeit.timeit(lambda: ulid.microsecond.new().str, number=10000))
     print("fastid.ulid", timeit.timeit(lambda: fastid.ulid(), number=10000))
 
 
@@ -21,3 +22,9 @@ def test_uuid():
     print()
     print("uuid.uuid4    ", timeit.timeit(lambda: uuid.uuid4(), number=10000))
     print("fastid.uuid_v7", timeit.timeit(lambda: fastid.uuid_v7(), number=10000))
+
+
+def test_objectid():
+    print()
+    print("bson.ObjectId", timeit.timeit(lambda: bson.ObjectId(), number=10000))
+    print("fastid.object_id", timeit.timeit(lambda: fastid.object_id(), number=10000))

src/lib.rs

@@ -1,10 +1,15 @@
 use std::sync::Mutex;
+
 use once_cell::sync::Lazy;
 use pyo3::prelude::*;
 use snowflake::SnowflakeIdGenerator;
 use ulid::Ulid;
 use uuid::Uuid;
 
+use crate::objectid::ObjectId;
+
+mod objectid;
+
 static SNOWFLAKE_GENERATOR: Lazy<Mutex<SnowflakeIdGenerator>> = Lazy::new(|| {
     Mutex::new(SnowflakeIdGenerator::new(1, 1))
 });
@@ -40,6 +45,10 @@ fn uuid_v7() -> PyResult<String> {
     Ok(Uuid::now_v7().to_string())
 }
 
+#[pyfunction]
+fn object_id() -> PyResult<String> {
+    Ok(ObjectId::new().to_hex_string())
+}
 
 #[pymodule]
 fn fastid(_py: Python, m: &PyModule) -> PyResult<()> {
@@ -50,5 +59,7 @@ fn fastid(_py: Python, m: &PyModule) -> PyResult<()> {
 
     m.add_function(wrap_pyfunction!(uuid_v7, m)?)?;
 
+    m.add_function(wrap_pyfunction!(object_id, m)?)?;
+
     Ok(())
 }

src/objectid.rs

@@ -0,0 +1,29 @@
+use std::time::{SystemTime, UNIX_EPOCH};
+use rand::Rng;
+
+#[derive(Debug)]
+pub struct ObjectId([u8; 12]);
+
+impl ObjectId {
+    pub fn new() -> Self {
+        let timestamp = SystemTime::now()
+            .duration_since(UNIX_EPOCH)
+            .expect("SystemTime before UNIX EPOCH!")
+            .as_secs() as u32;
+
+        let mut rng = rand::thread_rng();
+        let random_bytes = rng.gen::<[u8; 5]>();
+        let counter = rng.gen::<[u8; 3]>();
+
+        let mut bytes = [0; 12];
+        bytes[0..4].copy_from_slice(&timestamp.to_be_bytes());
+        bytes[4..9].copy_from_slice(&random_bytes);
+        bytes[9..12].copy_from_slice(&counter);
+
+        ObjectId(bytes)
+    }
+
+    pub fn to_hex_string(&self) -> String {
+        hex::encode(self.0)
+    }
+}