sycamore-query 0.1.1

Data fetching and caching for sycamore
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
54
55
56

use crate::client::ClientOptions;
use fluvio_wasm_timer::Instant;
use fnv::FnvHashMap;
use std::{any::Any, rc::Rc, time::Duration};

type Cache = FnvHashMap<Vec<u64>, CacheEntry>;

pub struct CacheEntry {
    created_at: Instant,
    lifetime: Duration,
    value: Rc<dyn Any>,
}

#[derive(Default)]
pub struct QueryCache {
    inner: Cache,
}

impl QueryCache {
    pub fn get(&self, id: &[u64]) -> Option<Rc<dyn Any>> {
        let entry = self.inner.get(id)?;
        let age = Instant::now().duration_since(entry.created_at);
        if age > entry.lifetime {
            None
        } else {
            Some(entry.value.clone())
        }
    }

    pub fn insert(
        &mut self,
        id: Vec<u64>,
        value: Rc<dyn Any>,
        options: &ClientOptions,
    ) -> Rc<dyn Any> {
        self.inner.insert(
            id,
            CacheEntry {
                created_at: Instant::now(),
                lifetime: options.cache_expiration,
                value: value.clone(),
            },
        );
        value
    }

    pub fn invalidate_keys(&mut self, keys: &[&[u64]]) {
        self.inner
            .retain(|key, _| !keys.iter().any(|&prefix| key.starts_with(prefix)));
    }

    pub fn collect_garbage(&mut self) {
        self.inner
            .retain(|_, entry| Instant::now().duration_since(entry.created_at) < entry.lifetime);
    }
}