~aleteoryx/lfm_embed

40372f0cf8a95f4d5bdf78fc522328b776482dbf — alyx 1 year, 4 months ago d796f67
fix packages, implement caching
5 files changed, 128 insertions(+), 2 deletions(-)

M .gitignore
M Cargo.toml
A src/cache.rs
M src/deserialize.rs
M src/lib.rs
M .gitignore => .gitignore +1 -0
@@ 1,3 1,4 @@
/Cargo.lock
/target
*~
.#*

M Cargo.toml => Cargo.toml +2 -0
@@ 6,3 6,5 @@ edition = "2021"
# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html

[dependencies]
serde = { version = "1.0.183", features = ["derive", "rc", "alloc"] }
serde_json = "1.0.104"

A src/cache.rs => src/cache.rs +120 -0
@@ 0,0 1,120 @@
use std::{future::Future, time::*, collections::HashMap, hash::Hash};

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
    F: for<'a> FnMut(&'a K) -> Fut,
    K: Hash + PartialEq + Eq + Clone,
    Fut: Future<Output = Result<V, &'static str>>
{
    pub fn new(interval: Duration, mut func: F) -> Self {
        Self{
            cache: HashMap::new(),
            interval, func
        }
    }
    
    pub async fn get(&mut self, key: &K) -> Result<&V, &'static str> {
        if self.is_stale(&key) {
            self.renew(&key).await
        } else {
            Ok(&self.cache.get(&key).unwrap().1)
        }
    }

    pub async fn renew(&mut self, key: &K) -> Result<&V, &'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();
            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 }
    }
}

impl<K, V, F, Fut> AsyncCache<K, V, F>
where
    F: for<'a> FnMut(&'a K) -> Fut,
    K: Hash + PartialEq + Eq + Clone,
    V: Clone,
    Fut: Future<Output = Result<V, &'static str>>
{
    pub async fn get_owned(&mut self, key: &K) -> Result<V, &'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()
    }
}
*/

M src/deserialize.rs => src/deserialize.rs +2 -2
@@ 74,8 74,8 @@ pub struct Artist {
    #[serde(default)]
    #[serde(rename = "image")]
    pub images: Vec<Image>,
    pub #[serde(default)]
    url: Option<Arc<str>>
    #[serde(default)]
    pub url: Option<Arc<str>>
}

#[derive(Deserialize, Debug)]

M src/lib.rs => src/lib.rs +3 -0
@@ 1,1 1,4 @@
#![feature(entry_insert)]

mod deserialize;
mod cache;