cache-fn 0.1.1

A Rust library for caching functions
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
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

use crate::{error::CacheError, options::CacheOptions};
use reqwest::Client;
use serde::{de::DeserializeOwned, Deserialize, Serialize};
use serde_json::Value;

#[derive(Debug, Clone)]
pub struct WorkerClient {
    client: Client,
    options: CacheOptions,
}

#[derive(Debug, Serialize)]
struct CacheRequest<T> {
    key: String,
    value: T,
    tags: Vec<String>,
    ttl: Option<u64>,
}

#[derive(Debug, Deserialize)]
struct CacheResponse {
    success: bool,
    data: Option<Value>,
    message: Option<String>,
}

impl WorkerClient {
    pub fn new(options: CacheOptions) -> Self {
        Self {
            client: Client::new(),
            options,
        }
    }

    pub async fn get<T>(&self, key: &str) -> Result<Option<T>, CacheError>
    where
        T: DeserializeOwned,
    {
        let url = format!("{}/get/{}", self.options.worker, key);
        let resp = self
            .client
            .get(&url)
            .header("Authorization", format!("Bearer {}", self.options.api_key))
            .send()
            .await?;
        let cr: CacheResponse = resp.json().await?;

        if !cr.success {
            if let Some(msg) = cr.message {
                return Err(CacheError::Worker(msg));
            }
            return Ok(None);
        }

        match cr.data {
            Some(val) => Ok(Some(serde_json::from_value(val)?)),
            None => Ok(None),
        }
    }

    pub async fn set<T>(&self, key: &str, value: T) -> Result<(), CacheError>
    where
        T: Serialize,
    {
        let url = format!("{}/set", self.options.worker);
        let ttl = self.options.revalidate.map(|d| d.as_millis() as u64);
        let req = CacheRequest {
            key: key.to_string(),
            value,
            tags: self.options.tags.clone(),
            ttl,
        };

        let resp = self
            .client
            .post(&url)
            .header("Authorization", format!("Bearer {}", self.options.api_key))
            .json(&req)
            .send()
            .await?;
        let cr: CacheResponse = resp.json().await?;

        if !cr.success {
            if let Some(msg) = cr.message {
                return Err(CacheError::Worker(msg));
            }
        }

        Ok(())
    }

    pub async fn invalidate_tags(&self, tags: &[String]) -> Result<(), CacheError> {
        let url = format!("{}/invalidate", self.options.worker);
        let resp = self
            .client
            .post(&url)
            .header("Authorization", format!("Bearer {}", self.options.api_key))
            .json(&tags)
            .send()
            .await?;
        let cr: CacheResponse = resp.json().await?;

        if !cr.success {
            if let Some(msg) = cr.message {
                return Err(CacheError::Worker(msg));
            }
            return Err(CacheError::Cache("Failed to invalidate cache".into()));
        }

        Ok(())
    }
}