feat(rust): disable HTTP caching

agama-project · Jan 10, 2025 · cc69ca8 · cc69ca8
1 parent d4771bd
commit cc69ca8
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/rust/agama-server/Cargo.toml b/rust/agama-server/Cargo.toml
@@ -24,7 +24,12 @@ macaddr = { version = "1.0", features = ["serde_std"] }
 async-trait = "0.1.83"
 axum = { version = "0.7.7", features = ["ws"] }
 serde_json = "1.0.128"
-tower-http = { version = "0.5.2", features = ["compression-br", "fs", "trace"] }
+tower-http = { version = "0.5.2", features = [
+    "compression-br",
+    "fs",
+    "trace",
+    "set-header",
+] }
 tracing-subscriber = { version = "0.3.18", features = ["env-filter"] }
 tracing-journald = "0.3.0"
 tracing = "0.1.40"

diff --git a/rust/agama-server/src/web/service.rs b/rust/agama-server/src/web/service.rs
@@ -21,6 +21,7 @@
 use super::http::{login, login_from_query, logout, session};
 use super::{config::ServiceConfig, state::ServiceState, EventsSender};
 use agama_lib::auth::TokenClaims;
+use axum::http::HeaderValue;
 use axum::{
     body::Body,
     extract::Request,
@@ -29,12 +30,14 @@ use axum::{
     routing::{get, post},
     Router,
 };
+use hyper::header::CACHE_CONTROL;
 use std::time::Duration;
 use std::{
     convert::Infallible,
     path::{Path, PathBuf},
 };
 use tower::Service;
+use tower_http::set_header::SetResponseHeaderLayer;
 use tower_http::{compression::CompressionLayer, services::ServeDir, trace::TraceLayer};
 use tracing::Span;
 
@@ -128,6 +131,10 @@ impl MainServiceBuilder {
                     ),
             )
             .layer(CompressionLayer::new().br(true))
+            .layer(SetResponseHeaderLayer::if_not_present(
+                CACHE_CONTROL,
+                HeaderValue::from_static("no-store"),
+            ))
             .with_state(state)
     }
 }