当前位置: 首页 > news >正文

3.Rust + Axum 提取器模式深度剖析

摘要

深入解读 Rust + Axum 提取器模式,涵盖内置提取器及自定义实现。

一、引言

在 Rust 的 Web 开发领域,Axum 作为一款轻量级且高效的 Web 框架,为开发者提供了强大的功能。其中,提取器(Extractor)模式是 Axum 中一个非常重要的特性,它允许开发者从 HTTP 请求中提取所需的数据,并将其转换为合适的 Rust 类型。通过提取器,开发者可以方便地处理查询参数、JSON 数据、表单数据等,同时还能进行数据验证和错误处理。本文将深入剖析 Axum 提取器模式,包括内置提取器的使用场景、自定义提取器的实现以及错误处理与 FromRequest trait 的设计。

二、内置提取器的使用场景

2.1 Query 提取器

Query 提取器用于从 URL 的查询字符串中提取数据。在处理 GET 请求时,查询字符串通常用于传递一些可选的参数。例如,我们有一个处理用户搜索请求的 API,用户可以通过查询字符串指定搜索关键词和分页信息。以下是一个示例:

use axum::{routing::get,Router,extract::Query,
};
use serde::Deserialize;#[derive(Deserialize)]
struct SearchParams {keyword: String,page: u32,
}async fn search(Query(params): Query<SearchParams>) -> String {format!("Searching for '{}' on page {}", params.keyword, params.page)
}#[tokio::main]
async fn main() {let app = Router::new().route("/search", get(search));axum::Server::bind(&"127.0.0.1:3000".parse().unwrap()).serve(app.into_make_service()).await.unwrap();
}

在这个示例中,SearchParams 结构体使用 serde::Deserialize 进行反序列化,Query 提取器会自动将查询字符串中的参数转换为 SearchParams 结构体的实例。当客户端访问 /search?keyword=rust&page=1 时,服务器会返回相应的搜索信息。

2.2 Json 提取器

Json 提取器用于从请求体中提取 JSON 数据。在处理 POST、PUT 等请求时,客户端通常会将数据以 JSON 格式发送到服务器。以下是一个处理用户注册请求的示例:

use axum::{routing::post,Router,extract::Json,
};
use serde::Deserialize;#[derive(Deserialize)]
struct UserRegistration {username: String,password: String,
}async fn register(Json(user): Json<UserRegistration>) -> String {format!("User '{}' registered successfully", user.username)
}#[tokio::main]
async fn main() {let app = Router::new().route("/register", post(register));axum::Server::bind(&"127.0.0.1:3000".parse().unwrap()).serve(app.into_make_service()).await.unwrap();
}

在这个示例中,UserRegistration 结构体使用 serde::Deserialize 进行反序列化,Json 提取器会自动将请求体中的 JSON 数据转换为 UserRegistration 结构体的实例。当客户端发送包含 JSON 数据的 POST 请求到 /register 时,服务器会处理注册逻辑。

2.3 Form 提取器

Form 提取器用于从表单数据中提取信息。在处理 HTML 表单提交时,表单数据通常以 application/x-www-form-urlencodedmultipart/form-data 格式发送到服务器。以下是一个处理用户登录表单的示例:

use axum::{routing::post,Router,extract::Form,
};
use serde::Deserialize;#[derive(Deserialize)]
struct LoginForm {username: String,password: String,
}async fn login(Form(form): Form<LoginForm>) -> String {format!("User '{}' is trying to log in", form.username)
}#[tokio::main]
async fn main() {let app = Router::new().route("/login", post(login));axum::Server::bind(&"127.0.0.1:3000".parse().unwrap()).serve(app.into_make_service()).await.unwrap();
}

在这个示例中,LoginForm 结构体使用 serde::Deserialize 进行反序列化,Form 提取器会自动将表单数据转换为 LoginForm 结构体的实例。当客户端提交登录表单时,服务器会处理登录逻辑。

三、自定义提取器实现请求数据验证

有时候,内置提取器可能无法满足我们的需求,我们需要自定义提取器来实现更复杂的请求数据验证。例如,我们希望对用户输入的邮箱地址进行格式验证。以下是一个自定义提取器的示例:

use axum::{extract::{FromRequest, RequestParts},http::Request,body::Body,response::{IntoResponse, Response},
};
use serde::Deserialize;
use validator::Validate;
use validator::email::Email;#[derive(Deserialize, Validate)]
struct UserRegistration {#[validate(email)]email: String,password: String,
}struct ValidatedUserRegistration(UserRegistration);#[axum::async_trait]
impl FromRequest<Body> for ValidatedUserRegistration {type Rejection = Response;async fn from_request(req: &mut RequestParts<Body>) -> Result<Self, Self::Rejection> {let Json(user) = axum::extract::Json::<UserRegistration>::from_request(req).await.map_err(|e| {(axum::http::StatusCode::BAD_REQUEST, e.to_string()).into_response()})?;if let Err(errors) = user.validate() {return Err((axum::http::StatusCode::BAD_REQUEST, errors.to_string()).into_response());}Ok(ValidatedUserRegistration(user))}
}async fn register(ValidatedUserRegistration(user): ValidatedUserRegistration) -> String {format!("User with email '{}' registered successfully", user.email)
}

在这个示例中,我们定义了一个 UserRegistration 结构体,并使用 validator 库对邮箱地址进行格式验证。然后,我们自定义了一个 ValidatedUserRegistration 提取器,在 from_request 方法中,首先使用 Json 提取器提取请求体中的 JSON 数据,然后对数据进行验证。如果验证失败,返回一个包含错误信息的响应;如果验证成功,返回一个包含验证后数据的 ValidatedUserRegistration 实例。

四、错误处理与 FromRequest trait 设计

4.1 FromRequest trait

FromRequest 是 Axum 中用于定义提取器的核心 trait。当我们自定义提取器时,需要实现 FromRequest trait。该 trait 有两个关联类型:Rejection 表示提取失败时返回的类型,from_request 方法用于从请求中提取数据。以下是 FromRequest trait 的基本定义:

#[async_trait::async_trait]
pub trait FromRequest<B>
whereB: Send + 'static,
{type Rejection: IntoResponse;async fn from_request(req: &mut RequestParts<B>) -> Result<Self, Self::Rejection>;
}

4.2 错误处理

在自定义提取器的 from_request 方法中,我们需要处理各种可能的错误。当提取数据失败或数据验证不通过时,我们需要返回一个合适的错误响应。在前面的自定义提取器示例中,我们使用 axum::http::StatusCode::BAD_REQUEST 表示请求数据无效,并将错误信息作为响应体返回。这样,客户端可以根据返回的状态码和错误信息进行相应的处理。

五、总结

Axum 的提取器模式为 Rust Web 开发提供了强大而灵活的功能。通过内置提取器,我们可以方便地处理查询参数、JSON 数据和表单数据;通过自定义提取器,我们可以实现更复杂的请求数据验证。同时,FromRequest trait 和错误处理机制确保了提取器的可靠性和可维护性。在实际开发中,合理运用提取器模式可以提高代码的可读性和可维护性,为构建高效、安全的 Web 应用提供有力支持。

http://www.xdnf.cn/news/4429.html

相关文章:

  • 【Python Cookbook】迭代器与生成器(一)
  • 【Qt】初识Qt(一)
  • Oracle 12.1.0.2补丁安装全流程
  • FPGA阵列
  • ZStack文档DevOps平台建设实践
  • 设计模式每日硬核训练 Day 14:组合模式(Composite Pattern)完整讲解与实战应用
  • 基于Django实现的图书分析大屏系统项目
  • Linux 常用命令总结
  • NLP高频面试题(四十六)——Transformer 架构中的位置编码及其演化详解
  • MCP和A2A是什么?
  • FreeRTOS事件标志组
  • 【Linux】第八章 监控和管理Linux进程
  • 关于Diamond机械手的运动学与动力学的推导
  • 【力扣刷题】49字母异位词分组,不用哈希,c语言实现
  • 《AI大模型应知应会100篇》第22篇:系统提示词(System Prompt)设计与优化
  • 基础知识 - 结构体
  • 首席人工智能官(Chief Artificial Intelligence Officer,CAIO)的详细解析
  • 从“链主”到“全链”:供应链数字化转型的底层逻辑
  • 智能sc一面
  • 【cocos creator 3.x】cocos creator2.x项目升级3.x项目改动点
  • 士兵乱斗(贪心)
  • 前端api(请求后端)简易template
  • Python高级爬虫之JS逆向+安卓逆向1.5节: 控制结构
  • docker harbor私有仓库登录报错
  • Ubuntu利用docker搭建Java相关环境问题记录
  • 如何有效防止服务器被攻击
  • 在激烈竞争下B端HMI设计怎样打造独特用户体验?
  • 数组理论基础
  • 从GPT到Gemini 大模型进化史
  • ADVB发送器设计