aboutsummaryrefslogtreecommitdiffstats
path: root/src/cache.rs
blob: 631f90031c2e42f2f9408a1f578d2916f7764679 (plain) (blame)
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
118
119
120
121
122
123
124
125
126
use std::{future::Future, time::*, collections::HashMap, hash::Hash};
use reqwest::StatusCode;
#[derive(Debug)]
pub struct AsyncCache<K, V, F> {
    func: F,
    cache: HashMap<K, (Instant, V)>,
    interval: Duration
}

impl<K, V, F, Fut> AsyncCache<K, V, F>
where
    for<'a> F: FnMut(&'a K) -> Fut + 'a,
    K: Hash + PartialEq + Eq + Clone,
    Fut: Future<Output = Result<V, (StatusCode, &'static str)>> + Send + Sync
{
    pub fn new(interval: Duration, func: F) -> Self {
        Self{
            cache: HashMap::new(),
            interval, func
        }
    }
    
    pub async fn get(&mut self, key: &K) -> Result<&V, (StatusCode, &'static str)> {
        if self.is_stale(&key) {
            log::trace!(target: "lfm::cache", "MISS : interval = {:?}", self.interval);
            self.renew(&key).await
        } else {
            log::trace!(target: "lfm::cache", "HIT : interval = {:?}", self.interval);
            Ok(&self.cache.get(&key).unwrap().1)
        }
    }

    pub async fn renew(&mut self, key: &K) -> Result<&V, (StatusCode, &'static str)> {
        let val = (self.func)(&key).await?;
        self.cache.insert(key.clone(), (Instant::now(), val));
        Ok(&self.cache.get(key).unwrap().1)
    }

    pub fn is_stale(&self, key: &K) -> bool {
        if let Some((last_update, _)) = self.cache.get(key) {
            let now = Instant::now();
            log::trace!(target: "lfm::cache", "Key exists, last update {:?} ago.", now - *last_update);
            now > (*last_update + self.interval)
        }
        else { true }
    }
    
    pub async fn get_opt(&self, key: &K) -> Option<&V> {
        if self.is_stale(key) {
            self.cache.get(key).map(|(_, v)| v)
        }
        else { None }
    }

    pub fn interval(&self) -> Duration { self.interval }
}

impl<K, V, F, Fut> AsyncCache<K, V, F>
where
    for<'a> F: FnMut(&'a K) -> Fut + 'a,
    K: Hash + PartialEq + Eq + Clone,
    V: Clone,
    Fut: Future<Output = Result<V, (StatusCode, &'static str)>> + Send + Sync
{
    pub async fn get_owned(&mut self, key: &K) -> Result<V, (StatusCode, &'static str)> {
        self.get(key).await.cloned()
    }
}
/*
pub struct AsyncCache<K, V, F> {
    func: F,
    cache: HashMap<K, (Instant, V)>,
    interval: Duration
}

impl<K, V, F> AsyncCache<K, V, F>
where
    for<'a> F: FnMut(&'a K) -> Fut + 'a,
    Fut: Future<Output = V>
{
    pub fn new(interval: Duration, mut func: F) -> Self {
        Self{
            cache: HashMap::new(),
            interval, func
        }
    }

    pub async fn get(&mut self, key: &K) -> &V {
        if self.is_stale(key) {
            self.renew().await
        } else {
            self.cache.get(key)
        }
    }

    pub async fn renew(&mut self, key: &K) -> &V {
        self.cache.get_mut(key).0 = now;
        self.cache.get_mut(key).1 = (self.func)(key).await;
        self.cache.get(key)
    }

    pub fn is_stale(&self, key: &K) -> bool {
        let now = Instant::now();
        let last_update = self.cache.get(key).0;
        now < (last_update + self.interval)
    }
    
    pub fn get_opt(&self, key: &K) -> Option<&T> {
        if self.is_stale(key) {
            Some(self.cache.get(key))
        }
        else { None }
    }
}

impl<K, V, F> AsyncCache<K, V, F>
where
    F: for<'a> FnMut(&'a K) -> Fut + 'a,
    Fut: Future<Output = V>,
    V: Clone
{
    pub async fn get_owned(&mut self, key: &K) -> V {
        self.get(key).await.clone()
    }
}
*/