Cache

提供緩存功能的中間件.

Cache 中間件可以對 Response 中的 StatusCode, Headers, Body 提供緩存功能. 對於已經緩存的內容, 當下次處理請求時, Cache 中間件會直接把緩存在內存中的內容發送給客戶端.

注意, 此插件不會緩存 BodyResBody::StreamResponse. 如果應用到了這一類型的 Response, Cache 不會處理這些請求, 也不會引起錯誤.

主要功能

  • CacheIssuer 提供了對分配的緩存鍵值的抽象. RequestIssuer 是它的一個實現, 可以定義依據請求的 URL 的哪些部分以及請求的 Method 生成緩存的鍵. 你也可以定義你自己的緩存鍵生成的邏輯. 緩存的鍵不一定是字符串類型, 任何滿足 Hash + Eq + Send + Sync + 'static 約束的類型都可以作爲鍵.

  • CacheStore 提供對數據的存取操作. MokaStore 是內置的基於 moka 的一個內存的緩存實現. 你也可以定義自己的實現方式.

  • Cache 是實現了 Handler 的結構體, 內部還有一個 skipper 字段, 可以指定跳過某些不需要緩存的請求. 默認情況下, 會使用 MethodSkipper 跳過除了 Method::GET 以外的所有請求.

    內部實現示例代碼:

    impl<S, I> Cache<S, I> {
      pub fn new(store: S, issuer: I) -> Self {
          let skipper = MethodSkipper::new().skip_all().skip_get(false);
          Cache {
              store,
              issuer,
              skipper: Box::new(skipper),
          }
      }
    }
    

示例代碼

use std::time::Duration;

use salvo::cache::{Cache, MokaStore, RequestIssuer};
use salvo::prelude::*;
use salvo::writing::Text;
use time::OffsetDateTime;

#[handler]
async fn home() -> Text<&'static str> {
    Text::Html(HOME_HTML)
}
#[handler]
async fn short() -> String {
    format!(
        "Hello World, my birth time is {}",
        OffsetDateTime::now_utc()
    )
}
#[handler]
async fn long() -> String {
    format!(
        "Hello World, my birth time is {}",
        OffsetDateTime::now_utc()
    )
}

#[tokio::main]
async fn main() {
    tracing_subscriber::fmt().init();

    let short_cache = Cache::new(
        MokaStore::builder()
            .time_to_live(Duration::from_secs(5))
            .build(),
        RequestIssuer::default(),
    );
    let long_cache = Cache::new(
        MokaStore::builder()
            .time_to_live(Duration::from_secs(60))
            .build(),
        RequestIssuer::default(),
    );
    let router = Router::new()
        .get(home)
        .push(Router::with_path("short").hoop(short_cache).get(short))
        .push(Router::with_path("long").hoop(long_cache).get(long));
    let acceptor = TcpListener::new("127.0.0.1:5800").bind().await;
    Server::new(acceptor).serve(router).await;
}

static HOME_HTML: &str = r#"
<!DOCTYPE html>
<html>
    <head>
        <title>Cache Example</title>
    </head>
    <body>
        <h2>Cache Example</h2>
        <p>
            This examples shows how to use cache middleware. 
        </p>
        <p>
            <a href="/short" target="_blank">Cache 5 seconds</a>
        </p>
        <p>
            <a href="/long" target="_blank">Cache 1 minute</a>
        </p>
    </body>
</html>
"#;
[package]
name = "example-cache-simple"
version = "0.1.0"
edition = "2021"
publish = false

[dependencies]
salvo = {workspace = true, features = ["cache"] }
tokio = { version = "1", features = ["macros"] }
time = "0.3"
tracing = "0.1"
tracing-subscriber = "0.3"