Rust HTTP请求中间件库reqwest-conditional-middleware的使用，实现条件性请求处理和响应缓存控制

安装

在项目目录中运行以下Cargo命令：

cargo add reqwest-conditional-middleware

或者在Cargo.toml中添加以下行：

reqwest-conditional-middleware = "0.4.0"

完整示例代码

以下是一个使用reqwest-conditional-middleware实现条件性请求处理和响应缓存控制的完整示例：

use reqwest::{Client, Request, Response};
use reqwest_conditional_middleware::ConditionalMiddleware;
use std::time::{SystemTime, UNIX_EPOCH};

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // 创建带有条件中间件的客户端
    let client = Client::builder()
        .with(ConditionalMiddleware::default())
        .build()?;
    
    // 示例URL
    let url = "https://example.com/api/data";
    
    // 构建请求
    let request = client.get(url)
        // 添加If-None-Match条件
        .header("If-None-Match", "\"some-etag\"")
        // 添加If-Modified-Since条件
        .header("If-Modified-Since", "Wed, 21 Oct 2015 07:28:00 GMT")
        .build()?;
    
    // 发送请求
    let response = client.execute(request).await?;
    
    // 处理响应
    if response.status() == 304 {
        println!("资源未修改，可以使用缓存版本");
    } else {
        println!("获取到新资源");
        // 处理响应数据...
        if let Some(etag) = response.headers().get("ETag") {
            println!("ETag: {:?}", etag);
        }
        if let Some(last_modified) = response.headers().get("Last-Modified") {
            println!("Last-Modified: {:?}", last_modified);
        }
    }
    
    Ok(())
}

功能说明

1. 条件性请求处理：

   • 通过添加If-None-Match和If-Modified-Since等请求头，服务器可以返回304状态码表示资源未修改
   • 这可以减少不必要的数据传输，提高性能

2. 响应缓存控制：

   • 中间件会自动处理ETag和Last-Modified等响应头
   • 可以基于这些信息构建后续的条件请求

3. 缓存验证：

   • 当服务器返回304状态码时，表示客户端的缓存仍然有效
   • 可以避免重新下载未修改的资源

注意事项

1. 该中间件需要与reqwest库一起使用
2. 确保正确设置请求的条件头字段
3. 处理304响应时需要从本地缓存获取数据
4. 对于POST等非安全/非幂等方法，条件请求可能不适用

这个示例展示了如何使用reqwest-conditional-middleware来优化HTTP请求，减少不必要的网络传输，提高应用程序性能。

更完整的示例代码

以下是一个更完整的示例，包含本地缓存处理和自动条件头设置：

use reqwest::{Client, Request, Response};
use reqwest_conditional_middleware::ConditionalMiddleware;
use std::collections::HashMap;
use std::sync::Mutex;
use std::time::{SystemTime, UNIX_EPOCH};

// 简单的内存缓存结构
struct Cache {
    data: Mutex<HashMap<String, (String, String)>>,
}

impl Cache {
    fn new() -> Self {
        Self {
            data: Mutex::new(HashMap::new()),
        }
    }

    fn get(&self, url: &str) -> Option<(String, String)> {
        self.data.lock().unwrap().get(url).cloned()
    }

    fn set(&self, url: String, etag: String, last_modified: String) {
        self.data.lock().unwrap().insert(url, (etag, last_modified));
    }
}

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    // 创建缓存实例
    let cache = Cache::new();
    
    // 创建带有条件中间件的客户端
    let client = Client::builder()
        .with(ConditionalMiddleware::default())
        .build()?;
    
    // 示例URL
    let url = "https://example.com/api/data";
    
    // 检查缓存
    let mut request_builder = client.get(url);
    
    if let Some((etag, last_modified)) = cache.get(url) {
        // 如果缓存中存在，添加条件头
        request_builder = request_builder
            .header("If-None-Match", format!("\"{}\"", etag))
            .header("If-Modified-Since", last_modified);
    }
    
    // 构建请求
    let request = request_builder.build()?;
    
    // 发送请求
    let response = client.execute(request).await?;
    
    // 处理响应
    match response.status() {
        reqwest::StatusCode::NOT_MODIFIED => {
            println!("资源未修改，使用缓存版本");
            // 从缓存中获取数据...
        }
        _ => {
            println!("获取到新资源");
            // 处理新数据...
            
            // 更新缓存
            if let (Some(etag), Some(last_modified)) = (
                response.headers().get("ETag"),
                response.headers().get("Last-Modified"),
            ) {
                cache.set(
                    url.to_string(),
                    etag.to_str()?.to_string(),
                    last_modified.to_str()?.to_string(),
                );
            }
        }
    }
    
    Ok(())
}

这个更完整的示例添加了以下功能：

1. 简单的内存缓存实现
2. 自动从缓存中读取ETag和Last-Modified值
3. 自动为后续请求设置条件头
4. 根据响应状态码处理缓存逻辑
5. 更新缓存中的ETag和Last-Modified值