arcodange-org/url-shortener
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

new: Made the Cache-Control headers fully configurable

Browse Source
This commit is contained in:
SinTan1729
2024-05-31 20:30:09 -05:00
parent 69fc25a264
commit 96495b037d
3 changed files with 14 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
actix/src/main.rs
View File

@@ -36,8 +36,7 @@ async fn main() -> Result<()> {
let cache_control_header = env::var("cache_control_header")
.ok()
.filter(|s| !s.trim().is_empty())
.unwrap_or(String::from("Enable"));
.filter(|s| !s.trim().is_empty());
// Actually start the server
HttpServer::new(move || {
@@ -54,10 +53,10 @@ async fn main() -> Result<()> {
.app_data(web::Data::new(AppState {
db: database::open_db(db_location.clone()),
}))
.wrap(if cache_control_header == "Disable" {
middleware::DefaultHeaders::new()
.wrap(if let Some(header) = &cache_control_header {
middleware::DefaultHeaders::new().add(("Cache-Control", header.to_owned()))
} else {
middleware::DefaultHeaders::new().add(("Cache-Control", "no-cache, private"))
middleware::DefaultHeaders::new()
})
.service(services::link_handler)
.service(services::getall)