salrucc 0.13.0

Sourced Asynchronous Least Recently Used Cache Control
Documentation 
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

use std::cell::Cell;
use std::sync::Arc;
use std::sync::Mutex;

use tokio::sync::broadcast::Sender;

use crate::CacheClock;

pub struct FakeClock {
	current_now: Mutex<Cell<i32>>,
	sender: Sender<()>,
}

pub type FakeClockDuration = i32;
pub type FakeClockInstant = i32;

impl FakeClock {
	pub fn new() -> Arc<Self> {
		let (sender, _) = tokio::sync::broadcast::channel(16);
		let fake_clock = Self {
			current_now: Mutex::new(Default::default()),
			sender,
		};
		Arc::new(fake_clock)
	}

	/// Sets the current time index on the fake clock
	pub fn set_time(&self, time_index: FakeClockInstant) {
		self.current_now.lock().unwrap().set(time_index);
		let _ = self.sender.send(());
	}

	pub async fn wait_until(&self, time_index: FakeClockInstant) {
		let mut receiver = self.sender.subscribe();

		while time_index > self.now() {
			receiver.recv().await.unwrap();
		}
	}

	pub fn now(&self) -> FakeClockInstant {
		self.current_now.lock().unwrap().get()
	}
}

impl CacheClock for Arc<FakeClock> {
	type Duration = FakeClockDuration;
	type Instant = FakeClockInstant;

	fn now(&self) -> FakeClockInstant {
		self.as_ref().now()
	}
}