本文详细介绍如何使用 Axum 框架在 Rust 中构建一个通用化的 Web 应用模板,包括:
- 构建 RESTful API
- 使用 Sqlx 和 Postgres 数据库实现数据库交互
- 采用类似 Nest.js 的项目组织结构,以提升代码可维护性
- 包含丰富的单元测试和集成测试
- 使用 Github Actions 实现 CI/CD 流程
基础开发环境搭建
为了快速开始,可以参考我的 Rust 项目模板,点击这里获取项目代码。该模板包含基础项目结构和一些配置,帮助你迅速搭建开发环境。
类似 Nest.js 的项目组织方式
为了提升项目的可维护性和扩展性,本文中采用了类似 Nest.js 的项目组织结构:
├── docs # 文档文件
├── fixtures # 必要文件,比如公私钥,测试 SQL 脚本
├── migrations # 数据库迁移文件,适用于 Sqlx
├── rest_client # VS Code REST Client 测试 API 文件
├─ ─ src # 源代码
│ ├── common # 公共模块,如加解密、错误处理、配置等
│ └── modules # 业务模块
│ ├── auth # 认证模块,包括: `handlers`, `services`, `dto`, `tests`, `middleware` 等
│ └── users # 用户管理模块,包括: `handlers`, `services`, `dto`, `tests`, `entity` 等
业务模块文件说明
mod.rs
:模块的导入导出,创建当前模块的路由器。handlers.rs
:处理用户请求的逻辑。services.rs
:处理业务逻辑以及与数据库的交互。entity.rs
:定义与数据库表对应的结构体,便于结构操作。dto.rs
:定义数据转换对象,用于请求和响应的序列化和反序列化。tests.rs
:包含单元测试和集成测试。middleware.rs
:定义模块的中间件,用于认证、权限控制等。src/lib.rs
:定义项目的路由器,加载配置文件、初始化全局状态、错误处理等。src/main.rs
:应用入口。
开发经验分享
在项目开发过程中,我积累了一些经验,分享如下,帮助大家更好地构建 Axum Web 应用:
1. 测试的排序和并发控制
- 测试数量多或者测试较为复杂时,可能会遇到并发导致的错误,特别是在集成测试中。可以使用
serial_test
来对测试进行排序,防止这些问题。
2. 测试数据库的创建与销毁
- 在测试时需要创建临时数据库,并在测试结束后销毁它。可以使用
sqlx-db-tester
,并为测试初始化一个连接到测试数据库的全局状态。
3. 全局配置与数据库连接
- 全局状态中应包含全局配置和数据库连接,这样可以统一管理数据库操作,并将所有
services.rs
中的数据库操作通过全局状态来执行,方便管理。
4. Github Actions CI 配置
- 集成测试中,
reqwest
默认依赖 OpenSSL,这在 CI 环境下可能会导致问题。因此,我们需要在 OpenSSL 和 BoringSSL 之间做出选择。由于jwt-simple
依赖于 BoringSSL,所以应禁用掉reqwest
的默认 OpenSSL 依赖。
5. 公私钥生成
- 将公私钥生成的逻辑放在
build.rs
中,这样每次执行cargo build
或cargo run
等构建操作时,都会自动生成证书文件到fixtures
目录下,方便开发和部署。