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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
use rand::Rng;
#[derive(Clone, Debug, PartialEq)]
pub enum Temporal {
Bson(crate::BsonDateTime),
Datetime(crate::DateTime<crate::Utc>),
Timestamp(i64),
Timezone(crate::Utc),
}
impl Temporal {
pub fn now() -> crate::DateTime<crate::Utc> {
crate::Utc::now()
}
pub fn bson_datetime() -> Self {
Self::Bson(Self::now().into())
}
pub fn datetime() -> Self {
Self::Datetime(Self::now())
}
pub fn timestamp() -> Self {
Self::Timestamp(Self::now().timestamp())
}
}
impl Default for Temporal {
fn default() -> Self {
Self::Timestamp(Self::now().timestamp())
}
}
#[derive(Clone, Debug, Hash, PartialEq, crate::Deserialize, crate::Serialize)]
pub enum Id {
Obj(crate::ObjectId),
Std(u64),
Str(String),
}
impl Id {
fn random_u64() -> u64 {
let mut rnd = rand::thread_rng();
rnd.gen::<u64>()
}
pub fn generate_object_id() -> Self {
Self::Obj(crate::ObjectId::new())
}
pub fn random_std() -> Self {
Self::Std(Self::random_u64())
}
}
impl Default for Id {
fn default() -> Self {
Self::Std(Self::random_u64())
}
}
#[cfg(test)]
mod tests {
use super::{Id, Temporal};
#[test]
fn test_ids() {
let actual = Id::generate_object_id();
let expected = actual.clone();
assert_eq!(actual, expected)
}
#[test]
fn test_temporal() {
let actual = Temporal::default();
let expected = actual.clone();
assert_eq!(&actual, &expected)
}
}
