libblobd_direct/
util.rs

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