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
use crate::{CacheManager, HttpResponse, Result};
use http_cache_semantics::CachePolicy;
use serde::{Deserialize, Serialize};
use url::Url;
#[cfg_attr(docsrs, doc(cfg(feature = "manager-cacache")))]
#[derive(Debug, Clone)]
pub struct CACacheManager {
pub path: String,
}
impl Default for CACacheManager {
fn default() -> Self {
Self { path: "./http-cacache".into() }
}
}
#[derive(Debug, Deserialize, Serialize)]
struct Store {
response: HttpResponse,
policy: CachePolicy,
}
fn req_key(method: &str, url: &Url) -> String {
format!("{}:{}", method, url)
}
#[allow(dead_code)]
impl CACacheManager {
pub async fn clear(&self) -> Result<()> {
cacache::clear(&self.path).await?;
Ok(())
}
}
#[async_trait::async_trait]
impl CacheManager for CACacheManager {
async fn get(
&self,
method: &str,
url: &Url,
) -> Result<Option<(HttpResponse, CachePolicy)>> {
let store: Store =
match cacache::read(&self.path, &req_key(method, url)).await {
Ok(d) => bincode::deserialize(&d)?,
Err(_e) => {
return Ok(None);
}
};
Ok(Some((store.response, store.policy)))
}
async fn put(
&self,
method: &str,
url: &Url,
response: HttpResponse,
policy: CachePolicy,
) -> Result<HttpResponse> {
let data = Store { response: response.clone(), policy };
let bytes = bincode::serialize(&data)?;
cacache::write(&self.path, &req_key(method, url), bytes).await?;
Ok(response)
}
async fn delete(&self, method: &str, url: &Url) -> Result<()> {
Ok(cacache::remove(&self.path, &req_key(method, url)).await?)
}
}