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
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
use std::fmt;
use std::str::FromStr;
use async_trait::async_trait;
use rand::Rng;
use destream::IntoStream;
use tc_error::*;
use tcgeneric::{Id, NetworkTime};
const INVALID_ID: &str = "Invalid transaction ID";
pub const MIN_ID: TxnId = TxnId {
timestamp: 0,
nonce: 0,
};
#[derive(Clone, Copy, Debug, Eq, PartialEq, Hash)]
pub struct TxnId {
timestamp: u64,
nonce: u16,
}
impl TxnId {
pub fn new(time: NetworkTime) -> TxnId {
TxnId {
timestamp: time.as_nanos(),
nonce: rand::thread_rng().gen(),
}
}
pub fn time(&self) -> NetworkTime {
NetworkTime::from_nanos(self.timestamp)
}
pub fn to_id(&self) -> Id {
self.to_string().parse().unwrap()
}
}
impl FromStr for TxnId {
type Err = TCError;
fn from_str(s: &str) -> TCResult<TxnId> {
let parts: Vec<&str> = s.split('-').collect();
if parts.len() == 2 {
let timestamp = parts[0]
.parse()
.map_err(|e| TCError::bad_request(INVALID_ID, e))?;
let nonce = parts[1]
.parse()
.map_err(|e| TCError::bad_request(INVALID_ID, e))?;
Ok(TxnId { timestamp, nonce })
} else {
Err(TCError::bad_request(INVALID_ID, s))
}
}
}
impl Ord for TxnId {
fn cmp(&self, other: &TxnId) -> std::cmp::Ordering {
if self.timestamp == other.timestamp {
self.nonce.cmp(&other.nonce)
} else {
self.timestamp.cmp(&other.timestamp)
}
}
}
impl PartialOrd for TxnId {
fn partial_cmp(&self, other: &TxnId) -> Option<std::cmp::Ordering> {
Some(self.cmp(other))
}
}
#[async_trait]
impl destream::de::FromStream for TxnId {
type Context = ();
async fn from_stream<D: destream::de::Decoder>(
context: (),
d: &mut D,
) -> Result<Self, D::Error> {
let s = <String as destream::de::FromStream>::from_stream(context, d).await?;
Self::from_str(&s).map_err(destream::de::Error::custom)
}
}
impl<'en> destream::en::IntoStream<'en> for TxnId {
fn into_stream<E: destream::en::Encoder<'en>>(self, e: E) -> Result<E::Ok, E::Error> {
self.to_string().into_stream(e)
}
}
impl<'en> destream::en::ToStream<'en> for TxnId {
fn to_stream<E: destream::en::Encoder<'en>>(&'en self, e: E) -> Result<E::Ok, E::Error> {
self.to_string().into_stream(e)
}
}
impl fmt::Display for TxnId {
fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
write!(f, "{}-{}", self.timestamp, self.nonce)
}
}