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

//! Atomic time types
#![cfg_attr(not(feature = "std"), no_std)]
#![cfg_attr(docsrs, feature(doc_cfg))]
#![cfg_attr(docsrs, allow(unused_attributes))]
#![deny(missing_docs, warnings)]
#![forbid(unsafe_code)]

pub use core::sync::atomic::Ordering;

use portable_atomic::AtomicU128;

mod duration;
pub use duration::AtomicDuration;
mod option_duration;
pub use option_duration::AtomicOptionDuration;

/// Utility functions for encoding/decoding [`Duration`] to other types.
pub mod utils {
  #[cfg(feature = "std")]
  use std::{
    sync::OnceLock,
    time::{Duration, Instant, SystemTime},
  };

  #[cfg(feature = "std")]
  fn init(now: Instant) -> (SystemTime, Instant) {
    static ONCE: OnceLock<(SystemTime, Instant)> = OnceLock::new();
    *ONCE.get_or_init(|| {
      let system_now = SystemTime::now();
      (system_now, now)
    })
  }

  /// Encode an [`Instant`] into a [`Duration`].
  #[cfg(feature = "std")]
  #[inline]
  pub fn encode_instant_to_duration(instant: Instant) -> Duration {
    let (system_now, instant_now) = init(instant);
    if instant <= instant_now {
      system_now.duration_since(SystemTime::UNIX_EPOCH).unwrap() + (instant_now - instant)
    } else {
      system_now.duration_since(SystemTime::UNIX_EPOCH).unwrap() + (instant - instant_now)
    }
  }

  /// Decode an [`Instant`] from a [`Duration`].
  #[cfg(feature = "std")]
  #[inline]
  pub fn decode_instant_from_duration(duration: Duration) -> Instant {
    let (system_now, instant_now) = init(Instant::now());
    let system_time = SystemTime::UNIX_EPOCH + duration;
    if system_time >= system_now {
      instant_now + system_time.duration_since(system_now).unwrap()
    } else {
      instant_now - system_now.duration_since(system_time).unwrap()
    }
  }

  pub use super::duration::{decode_duration, encode_duration};
  pub use super::option_duration::{decode_option_duration, encode_option_duration};
}

#[cfg(feature = "std")]
mod system_time;

#[cfg(feature = "std")]
#[cfg_attr(docsrs, doc(cfg(feature = "std")))]
pub use system_time::AtomicSystemTime;

#[cfg(feature = "std")]
mod option_system_time;
#[cfg(feature = "std")]
#[cfg_attr(docsrs, doc(cfg(feature = "std")))]
pub use option_system_time::AtomicOptionSystemTime;

#[cfg(feature = "std")]
mod instant;
#[cfg(feature = "std")]
#[cfg_attr(docsrs, doc(cfg(feature = "std")))]
pub use instant::AtomicInstant;

#[cfg(feature = "std")]
mod option_instant;
#[cfg(feature = "std")]
#[cfg_attr(docsrs, doc(cfg(feature = "std")))]
pub use option_instant::AtomicOptionInstant;

#[cfg(feature = "std")]
use utils::{decode_instant_from_duration, encode_instant_to_duration};