libblobd_lite/
util.rs

1#![allow(unused)]
2
3use chrono::Utc;
4use off64::u32;
5use off64::u64;
6
7pub(crate) fn get_now_ms() -> u64 {
8  u64!(Utc::now().timestamp_millis())
9}
10
11pub(crate) fn get_now_sec() -> u64 {
12  u64!(Utc::now().timestamp())
13}
14
15pub(crate) fn get_now_hour() -> u32 {
16  u32!(Utc::now().timestamp() / 60 / 60)
17}
18
19pub(crate) fn div_ceil(n: u64, d: u64) -> u64 {
20  (n / d) + ((n % d != 0) as u64)
21}
22
23pub(crate) fn div_mod_pow2(val: u64, pow2: u8) -> (u64, u64) {
24  (val >> pow2, val & ((1 << pow2) - 1))
25}
26
27pub(crate) fn div_pow2(val: u64, pow2: u8) -> u64 {
28  let (div, _) = div_mod_pow2(val, pow2);
29  div
30}
31
32pub(crate) fn mod_pow2(val: u64, pow2: u8) -> u64 {
33  let (_, mod_) = div_mod_pow2(val, pow2);
34  mod_
35}
36
37// Round up to next `2^pow2`.
38pub(crate) fn ceil_pow2(val: u64, pow2: u8) -> u64 {
39  let (mut div, mod_) = div_mod_pow2(val, pow2);
40  if mod_ != 0 {
41    div += 1;
42  };
43  div << pow2
44}
45
46// Round down to previous `2^pow2`.
47pub(crate) fn floor_pow2(val: u64, pow2: u8) -> u64 {
48  let (mut div, mod_) = div_mod_pow2(val, pow2);
49  div << pow2
50}
51
52pub(crate) fn is_multiple_of_pow2(val: u64, pow2: u8) -> bool {
53  let (_, mod_) = div_mod_pow2(val, pow2);
54  mod_ == 0
55}
56
57#[cfg(test)]
58mod tests {
59  use crate::util::ceil_pow2;
60  use crate::util::div_ceil;
61  use crate::util::div_mod_pow2;
62
63  #[test]
64  fn test_div_ceil() {
65    assert_eq!(div_ceil(0, 1), 0);
66    assert_eq!(div_ceil(0, 2), 0);
67    assert_eq!(div_ceil(0, 3), 0);
68    assert_eq!(div_ceil(1, 2), 1);
69    assert_eq!(div_ceil(1, 3), 1);
70    assert_eq!(div_ceil(10, 3), 4);
71    assert_eq!(div_ceil(2, 2), 1);
72    assert_eq!(div_ceil(3, 2), 2);
73  }
74
75  #[test]
76  fn test_div_mod_pow2() {
77    assert_eq!(div_mod_pow2(0, 0), (0, 0));
78    assert_eq!(div_mod_pow2(0, 1), (0, 0));
79    assert_eq!(div_mod_pow2(0, 2), (0, 0));
80    assert_eq!(div_mod_pow2(1, 0), (1, 0));
81    assert_eq!(div_mod_pow2(1, 1), (0, 1));
82    assert_eq!(div_mod_pow2(1, 2), (0, 1));
83    assert_eq!(div_mod_pow2(5, 2), (1, 1));
84    assert_eq!(div_mod_pow2(10, 2), (2, 2));
85    assert_eq!(div_mod_pow2(16392, 6), (256, 8));
86    assert_eq!(div_mod_pow2(16393, 6), (256, 9));
87  }
88
89  #[test]
90  fn test_ceil_pow2() {
91    assert_eq!(ceil_pow2(0, 0), 0);
92    assert_eq!(ceil_pow2(1, 0), 1);
93    assert_eq!(ceil_pow2(2, 0), 2);
94    assert_eq!(ceil_pow2(3, 0), 3);
95    assert_eq!(ceil_pow2(3, 1), 4);
96    assert_eq!(ceil_pow2(0, 2), 0);
97    assert_eq!(ceil_pow2(1, 2), 4);
98    assert_eq!(ceil_pow2(3, 2), 4);
99    assert_eq!(ceil_pow2(4, 2), 4);
100    assert_eq!(ceil_pow2(4, 8), 256);
101    assert_eq!(ceil_pow2(5, 8), 256);
102  }
103}
libblobd_lite/util.rs

libblobd_lite/
util.rs
