使用 Rust 和 reqwest 构建命令行 HTTP 客户端

在当今互联世界中，从命令行与网络服务进行交互的能力是无价的。无论是测试 API、调试网络问题还是自动化任务，拥有一个轻量级的 HTTP 客户端能够极大地简化工作流程。在本篇博客中，我们将探讨如何使用 Rust 和 reqwest crate 构建一个简单而强大的命令行 HTTP 客户端。

介绍

近年来，Rust 因其性能、安全性保证和丰富的库生态系统而备受青睐。reqwest 是 Rust 中一个高级的 HTTP 客户端库。结合 Rust 强大的类型系统和模式匹配能力，reqwest 允许我们创建简洁高效的 HTTP 客户端。

入门指南

在开始编码之前，请确保你的系统上已安装了 Rust 和 Cargo。你可以通过 官方 Rust 网站 上的说明来安装它们。

设计 CLI 接口

我们的 HTTP 客户端将接受各种命令行参数来定制请求。我们将使用 clap crate 来轻松解析命令行参数。以下是客户端支持的参数列表：

• url：要请求的资源的 URL（必需）。
• -X, --method：要使用的 HTTP 方法（默认为 GET）。
• -H, --header：要包含在请求中的额外标头。
• -d, --data：要包含在请求体中的数据。

构建 HTTP 客户端

我们将定义一个名为 Args 的结构体，使用 clap 的 Parser trait 来解析命令行参数。这个结构体将保存解析参数的值。然后，我们将使用 reqwest 来根据提供的参数构建和发送 HTTP 请求。

#[derive(Parser, Debug)]
#[command(author, version, about, long_about = None)]
struct Args {
    #[arg(required = true)]
    url: String,

    #[arg(short = 'X', long = "method")]
    method: Method,

    #[arg(short = 'H', long = "header")]
    header: Option<Vec<String>>,

    #[arg(short = 'd', long = "data")]
    data: Option<String>,
}

发送请求

我们的 run 函数接受解析后的参数，并使用它们来构建一个 HTTP 请求，使用 reqwest 的构建器模式。我们设置 URL、方法、标头和请求体（如果提供）。最后，我们异步发送请求并处理响应。

async fn run(
    url: String,
    method: Method,
    header: Option<Vec<String>>,
    data: Option<String>
){
    let client = reqwest::Client::new();
    let mut request_builder = client
        .request(method, url);
    if let Some(data) = data{
        request_builder = request_builder
            .header("Content-Type","application/x-www-form-urlencoded");
        request_builder = request_builder.body(data);
    }
    if let Some(header) = header {
        request_builder = header.iter().fold(request_builder, |builder, item| {
            let parts: Vec<&str> = item.split(':').collect();
            if parts.len() == 2 {
                builder.header(parts[0].trim().to_string(), parts[1].trim().to_string())
            } else {
                panic!("header格式错误: {}", item);
            }
        });
    }

    let mut res = request_builder.send().await.expect("请求错误");
    println!("status: {:#?}",res.status());
    println!("headers: {:#?}",res.headers());
    println!("content_length: {:#?}",res.content_length().expect("文本长度获取失败"));
    println!("remote_addr: {:#?}",res.remote_addr().expect("远程地址获取失败"));
    println!("body:");
    while let Some(chunk) = res.chunk().await.expect("响应失败") {
        if let Ok(utf8_string) = String::from_utf8(Vec::from(chunk.clone())) {
            println!("{:#?}", utf8_string);
        } else {
            println!("{:#?}", String::from_utf8_lossy(&chunk));
        }
    }
}

处理响应

一旦收到响应，我们将打印出相关信息，如状态码、标头、内容长度和响应体。我们以块的形式读取响应体，以有效地处理可能很大的响应。

结论

只需几行 Rust 代码，我们就创建了一个命令行 HTTP 客户端，能够发出各种类型的请求。这个客户端可以轻松扩展，支持更多功能，如身份验证、HTTPS 和处理不同的内容类型。

源代码：https://github.com/golangwebLK/rurl