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
114
115
116
117
use super::errors::convert_bincode_error;
use super::{Error, Result};
use bytes::Bytes;
use multibase::{self, Base};
use rand::rngs::OsRng;
use rand::Rng;
use rayon::current_num_threads;
use serde::{de::DeserializeOwned, Deserialize, Serialize};
pub fn serialise<T: Serialize>(data: &T) -> Result<Vec<u8>> {
bincode::serialize(data).map_err(convert_bincode_error)
}
pub(crate) fn deserialise<'a, T>(bytes: &'a [u8]) -> Result<T>
where
T: Deserialize<'a>,
{
bincode::deserialize(bytes).map_err(convert_bincode_error)
}
pub(crate) fn encode<T: Serialize>(data: &T) -> Result<String> {
let bytes = serialise(&data)?;
Ok(multibase::encode(Base::Base32Z, &bytes))
}
pub(crate) fn decode<I: AsRef<str>, O: DeserializeOwned>(encoded: I) -> Result<O> {
let (base, decoded) =
multibase::decode(encoded).map_err(|e| Error::FailedToParse(e.to_string()))?;
if base != Base::Base32Z {
return Err(Error::FailedToParse(format!(
"Expected z-base-32 encoding, but got {:?}",
base
)));
}
deserialise(&decoded).map_err(|e| Error::FailedToParse(e.to_string()))
}
#[macro_export]
macro_rules! btree_set {
($($item:expr),*) => {{
let mut _set = ::std::collections::BTreeSet::new();
$(
let _ = _set.insert($item);
)*
_set
}};
($($item:expr),*,) => {
btree_set![$($item),*]
};
}
#[macro_export]
macro_rules! btree_map {
() => ({
::std::collections::BTreeMap::new()
});
($($key:expr => $value:expr),*) => {{
let mut _map = ::std::collections::BTreeMap::new();
$(
let _ = _map.insert($key, $value);
)*
_map
}};
($($key:expr => $value:expr),*,) => {
btree_map![$($key => $value),*]
};
}
pub fn random_bytes(length: usize) -> Bytes {
use rayon::prelude::*;
let threads = current_num_threads();
if threads > length {
let mut rng = OsRng;
return ::std::iter::repeat(())
.map(|()| rng.gen::<u8>())
.take(length)
.collect();
}
let per_thread = length / threads;
let remainder = length % threads;
let mut bytes: Vec<u8> = (0..threads)
.par_bridge()
.map(|_| vec![0u8; per_thread])
.map(|mut bytes| {
let bytes = bytes.as_mut_slice();
rand::thread_rng().fill(bytes);
bytes.to_owned()
})
.flatten()
.collect();
bytes.extend(vec![0u8; remainder]);
Bytes::from(bytes)
}