如何使用Rust框架 Actix构建web后端
rust的生态里,后端框架使用量最大的两个框架是Rocket
和Actix
,本文使用的是Actix
框架。
本文部分展示代码为,曾经公司王者荣耀比赛写的后端项目逻辑。展示的代码隐藏了部分实现,这里大家有兴趣的话可以直接在我的github repo 中寻找源码,并欢迎大家给我提供意见(点赞一键三连🐶)。也欢迎大家对于本文积极提出意见。
环境
- macOS环境下安装rust简单,一行命令
curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh
rust的相关工具就下载好了。这里如果需要可以更新下载源为国内源。 - IDE使用JetBrains的
RustRover
,目前EAP版本下免费。
启动
创建项目之后,项目的依赖都在Cargo.toml
中配置,数据库依赖方面使用sqlx
,features使用mysql
。:
[dependencies]
actix-web = "4.4"
dotenvy = "0.15"
env_logger = "0.10"
sqlx = { version = "0.7.1", features = ["mysql", "runtime-tokio", "chrono"] }
······
关于架构方面,使用多层架构:
.
├── Cargo.lock
├── Cargo.toml
└── src
├── controller
├── main.rs
├── model
├── repository
└── service
入口文件
main文件书写如下,分别是函数签名和内容:
#[actix_web::main]
async fn main() -> io::Result<()> {
init_environment();
let server_addr = env::var(SERVER_ADDR).expect(SERVER_ADDR_NOT_SET_MSG);
let database_url = env::var(DATABASE_URL).expect(DATABASE_URL_NOT_SET_MSG);
let pool = init_database(database_url).await;
log::info!("{}{}", STARTING_SERVER_LOG, server_addr);
HttpServer::new(move || { create_app(pool.clone()) })
.bind(server_addr)?
.run()
.await
}
先用一个属性宏将main函数转换为一个异步函数,允许函数使用异步操作。函数签名中,函数名为main,返回类型为io::Result<()>
, ::
在rust里是关联函数调用(类似于静态方法),与之对应的是实例函数(类似于成员方法)。
继续看函数内容,获得环境里的值进行初始化。然后HttpServer::new
初始化一个新的HTTP服务器实例;move || { create_app(pool.clone()) }
这是一个rust的闭包写法,获取了作用域外的变量的所有权(所有权是rust 核心机制,这里就不展开了)。然后绑定指定的地址、启动服务器、以及声明启动服务器的异步。
定义controller
我们定义一下controller的代码,这里的思路和其他后端框架实现差不多。
用属性宏指定请求路径,获得接收的参数PostParam
,然后传递这个param给对应的service处理,响应成功。
#[post("/")]
async fn pick_heroes(
web::Json(param): web::Json<PostParam>,
app_state: web::Data<AppState>,
) -> actix_web::Result<impl Responder> {
let response_data = app_state.service.pick.pick_heroes(param).await?;
Ok(web::Json(response_data))
}
定义model
这是我们前端需要用到的MyResult
结构体,包含多个字端,包括整数、字符串、时间结构体、数组Vec包裹的日志结构体;属性宏赋予了调试和序列化的特性。
#[derive(Debug, Serialize)]
pub struct MyResult {
pub team_id: i32,
pub data: String,
pub time: NaiveDateTime,
pub logs: Vec<Log>,
}
定义service
下面是service的核心代码片段,定义PickService
trait特征和PickServiceImpl
结构体,然后先给PickServiceImpl
结构体加一个返回自身的关联函数。然后实现PickService这个定义好的trait。
#[async_trait]
宏在这里帮助我们在trait里支持异步函数;: Sync + Send
意为有两个约束,Sync
和Send
是两个并发特性(trait),能确保类型在多线程环境中的安全使用;Arc
是一个用于线程安全的指针,可以用来在多线程环境中分享数据;<dyn HeroRepository>
表示实现了这个trait。
#[async_trait]
pub trait PickService: Sync + Send {
async fn pick_heroes(&self, param: PostParam) -> Result<MyResult, actix_web::Error>;
}
pub struct PickServiceImpl {
pub hero_repository: Arc<dyn HeroRepository>,
pub team_repository: Arc<dyn TeamRepository>,
pub log_repository: Arc<dyn LogRepository>,
}
impl PickServiceImpl {
pub fn new(
hero_repository: Arc<dyn HeroRepository>,
team_repository: Arc<dyn TeamRepository>,
log_repository: Arc<dyn LogRepository>,
) -> Self {
PickServiceImpl { hero_repository, team_repository, log_repository }
}
}
下面代码实现了方法pick_heroes
,可以将team_repository
查询到的team
数据,放入MyResult
结构体之中,之后又将team
和result
的可变引用放入check_team_is_picked
方法,进一步处理,处理之后的rusult
则成功响应返回。方法返回Result<MyResult, sqlx::Error>
类型,即成功时返回MyResult
对象,失败则返回sqlx::Error
。
#[async_trait]
impl PickService for PickServiceImpl {
async fn pick_heroes(&self, param: PostParam) -> Result<MyResult, actix_web::Error> {
let mut team = self.team_repository.get_by_encrypt_code(param.encrypt_code).await
.map_err(actix_web::error::ErrorInternalServerError)?;
let mut result = MyResult {
team_id: team.id,
data: team.pick_content.clone(),
time: current_time(),
logs: self.log_repository.get_by_team_id(team.id).await
.expect(GET_LOGS_FAILED_ERROR),
};
self.check_team_is_picked(&mut team, &mut result).await;
Ok(result)
}
}
定义repository
这是team
相关的repo的一个函数,在sqlx
这个包的使用习惯下,直接把sql写在代码里,返回我们想要查询的Team
数据,如果报错,则返回sqlx::Error
类型。
#[async_trait]
impl TeamRepository for TeamRepositoryImpl {
async fn get_by_encrypt_code(&self, encrypt_code: String) -> Result<Team, sqlx::Error> {
sqlx::query_as::<_, Team>("SELECT * FROM `team` WHERE `encrypt_code` = ?")
.bind(encrypt_code)
.fetch_one(&*self.pool)
.await
}
}
End
代码风格和项目依赖架构就到此为止了,以上举了一些代码的例子,展示Actix
框架中的一些风格和功能,以及rust
语言在开发项目中的特性。在学习rust
的路上,老实说我遇到不小的挑战,很多语法和特性习惯跟之前掌握的语言语法差距不小,以及老生常谈地跟rust编译器斗智斗勇、debug的漫长经历。
作为StackOverflow
调查中连续多年的最受欢迎语言,整体来说算是国外雷声大,国内雨点小。不过随着一些国内的大厂(如字节)入场,和一些市场和生态的兴起,或许Rust
语言未来在国内兴盛也未可知(希望别像"风暴要火")。