yume/yumechi-no-kuni-proxy-worker
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

set a more realistic cache-control for URL summary

Browse source 
Signed-off-by: eternal-flame-AD <yume@yumechi.jp>
This commit is contained in:
ゆめ 2024-12-21 20:00:28 -06:00
parent e6afa180bb
commit 791c0183c9
No known key found for this signature in database
1 changed files with 18 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

29
src/lib.rs
View file

@ -19,7 +19,7 @@ use axum::extract::ConnectInfo;
use axum::{
body::Body,
extract::{Path, Query, State},
http::{self, HeaderMap, StatusCode},
http::{self, HeaderMap, HeaderValue, StatusCode},
response::{IntoResponse, Redirect, Response},
routing::get,
Json, Router,
@ -250,20 +250,27 @@ pub async fn set_cache_control<
next: axum::middleware::Next,
) -> Response {
let mut resp = next.run(request).await;
let ok = resp.status().is_success();
let headers = resp.headers_mut();
if enabled {
if resp.status() == StatusCode::OK {
let headers = resp.headers_mut();
headers.insert(
"Cache-Control",
"public, max-age=31536000, immutable".parse().unwrap(),
);
if ok {
if headers.get("content-type").map_or(false, |x| {
[b"image/", b"video/", b"audio/"]
.into_iter()
.any(|y| x.as_bytes().starts_with(y))
}) {
headers.insert(
"cache-control",
HeaderValue::from_static("public, max-age=31536000, immutable"),
);
} else {
headers.insert("cache-control", HeaderValue::from_static("max-age=86400"));
}
} else {
let headers = resp.headers_mut();
headers.insert("Cache-Control", "max-age=300".parse().unwrap());
headers.insert("cache-control", HeaderValue::from_static("max-age=300"));
}
} else {
let headers = resp.headers_mut();
headers.insert("Cache-Control", "no-store".parse().unwrap());
headers.insert("cache-control", HeaderValue::from_static("no-store"));
}
resp
}