Update example

hit-box · Sep 22, 2021 · 74f6d10 · 74f6d10
1 parent f89bf31
commit 74f6d10
Show file tree

Hide file tree

Showing 3 changed files with 18 additions and 3 deletions.
diff --git a/examples/examples/axum.rs b/examples/examples/axum.rs
@@ -11,6 +11,18 @@ use hitbox_axum::CacheLayer;
 #[tokio::main]
 async fn main() {
     let cache_layer = CacheLayer::new();
+    // let cache_layer = CacheLayer::build()
+    //     .with_ttl(30)
+    //     .with_stale(30)
+    //     .with_version(1)
+    //     .with_cache_key_prefix("тыдыщь")
+    //     .by_path()
+    //     .by_path_extended(parse)
+    //     .by_header("X-Request")
+    //     .by_header("X-Location")
+    //     .by_query()
+    //     .by_body()
+    //     .finish();
     let app = Router::new().route("/users/:user_id/", get(handler.layer(cache_layer)));
 
     let addr: SocketAddr = ([127, 0, 0, 1], 3000).into();

diff --git a/hitbox-axum/src/layer.rs b/hitbox-axum/src/layer.rs
@@ -2,12 +2,16 @@ use crate::service::CacheService;
 use tower_layer::Layer;
 use crate::CacheableRequest;
 
-#[derive(Debug, Clone, Default)]
+#[derive(Debug, Default)]
+pub struct CacheConfig;
+
+#[derive(Debug, Default)]
 pub struct CacheLayer {
     cache_key_prefix: Option<String>,
     ttl: Option<u32>,
     stale_ttl: Option<u32>,
     cache_version: Option<u32>,
+    cache_config: CacheConfig,
 }
 
 impl CacheLayer {

diff --git a/hitbox-axum/src/service.rs b/hitbox-axum/src/service.rs
@@ -4,7 +4,7 @@ use axum::http::{Request, Response};
 use futures::future::BoxFuture;
 use tower_service::Service;
 
-use crate::CacheableRequest;
+use crate::{CacheableRequest, CacheLayer};
 use hitbox::cache::Cacheable;
 
 #[derive(Clone)]
@@ -34,7 +34,6 @@ impl<S, ReqBody, ResBody> Service<Request<ReqBody>> for CacheService<S>
     }
 
     fn call(&mut self, request: Request<ReqBody>) -> Self::Future {
-        // self.service.call(wrapper.into_inner())
         let clone = self.service.clone();
         let mut service = std::mem::replace(&mut self.service, clone);