Rust Web 全栈开发(四):构建 REST API

Rust Web 全栈开发(四):构建 REST API

参考视频:https://www.bilibili.com/video/BV1RP4y1G7KF

需要使用的 crate:

  • serde
  • chrono

构建的内容

构建一个教师课程的 WebService,支持 3 种请求,每个请求对应一个 Handler:

  1. POST:/courses,提交一个课程
  2. GET:/courses/teacher_id,查询老师的所有课程
  3. GET:/courses/teacher_id/course_id,查询老师的一个课程的详细信息

在这里插入图片描述

构建项目

相关文件:

  • bin/teacher_service.rs
  • models.rs
  • state.rs
  • routers.rs
  • handlers.rs

继续使用上一节的项目。

在 webservice/src 目录下,创建 models.rs、state.rs、routers.rs、handlers.rs。

打开 webservice/Cargo.toml,添加一个 bin 的声明:

[[bin]]
name = "teacher_service"

打开 webservice/Cargo.toml,在 [package] 部分添加:

default-run = "teacher_service"

这样设置后, 当我们不带参数运行 webservice 包时,它默认执行 teacher_service.rs 中的代码。

在 webservice/src/bin 目录下创建 teacher_service.rs。

健康检查

打开 state.rs,编写代码:

use std::sync::Mutex;

pub struct AppState {
    pub health_check_response: String,
    pub visit_count: Mutex<u32>,
}

AppState 代表应用程序的状态。因为我们使用了 Actix 框架,在请求时应用程序的状态会被注入到 Handler 中,于是在处理请求时 Handler 能获取到 AppState。

health_check_response 是健康检查响应,不可变,在所有线程中共享。visit_count 是访问次数,可变,可以在线程间安全共享。

打开 routers.rs,编写代码:

use super::handlers::*;
use actix_web::web;

pub fn general_routes(cfg: &mut web::ServiceConfig) {
    cfg.route("/health", web::get().to(health_check_handler));
}

打开 handlers.rs,编写代码:

use super::state::AppState;
use actix_web::{web, HttpResponse};

pub async fn health_check_handler(app_state: web::Data<AppState>) -> HttpResponse {
    println!("incoming for health check");
    let health_check_response = &app_state.health_check_response;
    let mut visit_count = app_state.visit_count.lock().unwrap();
    let response = format!("{} {} times", health_check_response, visit_count);
    *visit_count += 1;
    HttpResponse::Ok().json(&response)
}

互斥锁在 health_check_handler 函数结束时自动释放。

打开 teacher_service.rs,编写代码:

use actix_web::{web, App, HttpServer};
use std::io;
use std::sync::Mutex;

#[path = "../handlers.rs"]
mod handlers;
#[path = "../routers.rs"]
mod routers;
#[path = "../state.rs"]
mod state;

use routers::*;
use state::AppState;

#[actix_rt::main]
async fn main() -> io::Result<()> {
    let shared_data = web::Data::new(AppState {
        health_check_response: "I'm OK.".to_string(),
        visit_count: Mutex::new(0),
    });
    let app = move || {
        App::new()
            .app_data(shared_data.clone())
            .configure(general_routes)
    };

    HttpServer::new(app).bind("127.0.0.1:3000")?.run().await
}

构建 APP 时,把 shared_data(也就是一个 AppState 实例)作为参数传递进去了。

在终端执行命令 cargo run -p webservice,因为之前的设置,默认执行 webservice 库中的 teacher_service。

或者 cd 到 webservice 目录,执行命令 cargo run。

在这里插入图片描述

再新建一个终端,cd 到 webservice,执行命令 curl 127.0.0.1:3000/health。

每次执行一次请求,服务器打印一句 incoming for health check。

然后,新终端会打印出完整的响应信息:

在这里插入图片描述

注意 Content 中的内容,每执行一次 curl 127.0.0.1:3000/health 命令,其计数就会加一。

完整输出结果:

在这里插入图片描述

PS C:\Users\81228\Documents\Program\Rust Project\Actix-Workspace\webservice> curl 127.0.0.1:3000/health


StatusCode        : 200
StatusDescription : OK
Content           : "I'm OK. 0 times"
RawContent        : HTTP/1.1 200 OK
                    Content-Length: 17
                    Content-Type: application/json
                    Date: Thu, 10 Jul 2025 08:52:34 GMT

                    "I'm OK. 0 times"
Forms             : {}
Images            : {}
InputFields       : {}
Links             : {}                                                                                                                                                              ParsedHtml        : mshtml.HTMLDocumentClass                                                                                                                                        RawContentLength  : 17                                                                                                                                                                                                                                                                                                                                                  


PS C:\Users\81228\Documents\Program\Rust Project\Actix-Workspace\webservice> curl 127.0.0.1:3000/health


StatusCode        : 200
StatusDescription : OK
Content           : "I'm OK. 1 times"
RawContent        : HTTP/1.1 200 OK
                    Content-Length: 17
                    Content-Type: application/json
                    Date: Thu, 10 Jul 2025 08:52:57 GMT

                    "I'm OK. 1 times"
Forms             : {}
Images            : {}
InputFields       : {}
Links             : {}                                                                                                                                                              ParsedHtml        : mshtml.HTMLDocumentClass                                                                                                                                        RawContentLength  : 17                                                                                                                                                                                                                                                                                                                                                  


PS C:\Users\81228\Documents\Program\Rust Project\Actix-Workspace\webservice> curl 127.0.0.1:3000/health


StatusCode        : 200
StatusDescription : OK
Content           : "I'm OK. 2 times"
RawContent        : HTTP/1.1 200 OK
                    Content-Length: 17
                    Content-Type: application/json
                    Date: Thu, 10 Jul 2025 08:53:10 GMT

                    "I'm OK. 2 times"
Forms             : {}
Headers           : {[Content-Length, 17], [Content-Type, application/json], [Date, Thu, 10 Jul 2025 08:53:10 GMT]}
Images            : {}
InputFields       : {}
Links             : {}
ParsedHtml        : mshtml.HTMLDocumentClass
RawContentLength  : 17

至此,健康检查的功能就完成了。

POST 资源:添加课程

打开 webservice/Cargo.toml,在 [dependencies] 部分添加:

# Data serialization library
serde = { version = "1.0.144", features = ["derive"] }
# Other utilities
chrono = {version = "0.4.22", features = ["serde"]}

打开 models.rs,编写代码:

use actix_web::web;
use chrono::NaiveDateTime;
use serde::{Deserialize, Serialize};

#[derive(Deserialize, Serialize, Debug, Clone)]
pub struct Course {
    pub teacher_id: i32,
    pub id: Option<i32>,
    pub name: String,
    pub time: Option<NaiveDateTime>,
}

impl From<web::Json<Course>> for Course {
    fn from(course: web::Json<Course>) -> Self {
        Course {
            teacher_id: course.teacher_id,
            id: course.id,
            name: course.name.clone(),
            time: course.time,
        }
    }
}

Course 结构体存储了课程的相关信息,我们为 Course 实现了 From trait。

修改 state.rs:

use std::sync::Mutex;
use crate::models::Course;

pub struct AppState {
    pub health_check_response: String,
    pub visit_count: Mutex<u32>,
    pub courses: Mutex<Vec<Course>>,
}

修改 teacher_service.rs:

use actix_web::{web, App, HttpServer};
use std::io;
use std::sync::Mutex;

#[path = "../handlers.rs"]
mod handlers;
#[path = "../models.rs"]
mod models;
#[path = "../routers.rs"]
mod routers;
#[path = "../state.rs"]
mod state;

use routers::*;
use state::AppState;

#[actix_rt::main]
async fn main() -> io::Result<()> {
    let shared_data = web::Data::new(AppState {
        health_check_response: "I'm OK.".to_string(),
        visit_count: Mutex::new(0),
        courses: Mutex::new(vec![]),
    });
    let app = move || {
        App::new()
            .app_data(shared_data.clone())
            .configure(general_routes)
    };

    HttpServer::new(app).bind("127.0.0.1:3000")?.run().await
}

到这先像上一节一样测试一下程序,是能成功执行的。

下面编写添加课程的功能。

修改 routers.rs,添加一个公共函数:

pub fn course_routes(cfg: &mut web::ServiceConfig) {
    cfg.service(
        web::scope("/courses")
            .route("/", web::post().to(new_course))
    );
}

我们声明了 /courses 范围,在该范围内再接着 / 路径,则执行 POST 请求,由 new_course 这个 Handler 进行处理。

这个 course_routes 同样需要注册,在 teacher_service.rs 的 main 函数的 .configure(general_routes) 语句后加上 .configure(course_routes)。

修改 handlers.rs,new_course 函数实现了新增课程的功能:

use super::state::AppState;
use actix_web::{web, HttpResponse};
use super::models::Course;
use chrono::Utc;

pub async fn health_check_handler(app_state: web::Data<AppState>) -> HttpResponse {
    println!("incoming for health check");
    
    let health_check_response = &app_state.health_check_response;
    let mut visit_count = app_state.visit_count.lock().unwrap();
    let response = format!("{} {} times", health_check_response, visit_count);
    *visit_count += 1;
    
    HttpResponse::Ok().json(&response)
}

pub async fn new_course(
    new_course: web::Json<Course>,
    app_state: web::Data<AppState>,
) -> HttpResponse {
    println!("Received new course");
    
    let course_count = app_state.courses
        .lock()
        .unwrap()
        .clone()
        .into_iter()
        .filter(|course| course.teacher_id == new_course.teacher_id)
        //.collect::<Vec<Course>>()
        //.len();
        .count();
    
    let new_course = Course {
        teacher_id: new_course.teacher_id,
        id: Some(course_count + 1),
        name: new_course.name.clone(),
        time: Some(Utc::now().naive_utc()), // 当前时间
    };
    
    app_state.courses.lock().unwrap().push(new_course);
    
    HttpResponse::Ok().json("Course added")
}

最后,为这个功能添加一个测试,在 handlers.rs 的最后加上如下代码:

#[cfg(test)]
mod tests {
    use super::*;
    use actix_web::http::StatusCode;
    use std::sync::Mutex;

    #[actix_rt::test]
    async fn post_course_test() {
        let course = web::Json(Course {
            teacher_id: 1,
            name: "Test course".into(),
            id: None,
            time: None,
        });
        
        let app_state: web::Data<AppState> = web::Data::new(AppState {
            health_check_response: "".to_string(),
            visit_count: Mutex::new(0),
            courses: Mutex::new(vec![]),
        });
        
        // 模拟添加课程的请求
        let response = new_course(course, app_state).await;
        
        assert_eq!(response.status(), StatusCode::OK);
    }
}

打开终端,cd 到 webservice,执行命令 cargo test,一个测试通过了:

在这里插入图片描述

我们在一个终端执行命令 cargo run 运行程序,然后在一个 bash 终端执行命令:

curl -X POST localhost:3000/courses/ -H "Content-Type: application/json" -d '{"teacher_id":1, "name":"First course"}'

来模拟一个添加课程的 POST 请求。

在这里插入图片描述

我们看到终端打印 Received new course,bash 终端打印 Course added,说明这个功能成功实现了。

GET 资源:查询课程

查询指定老师的所有课程

修改 routers.rs 中的 course_routes 函数:

pub fn course_routes(cfg: &mut web::ServiceConfig) {
    cfg.service(
        web::scope("/courses")
            .route("/", web::post().to(new_course))
            .route("/{user_id}", web::get().to(get_courses_for_teacher))
    );
}

新增 /courses/{user_id} 路径,对应一个 GET 请求,对应的 Handler 是 get_courses_for_teacher 函数。

修改 handlers.rs,get_courses_for_teacher 函数实现了查询指定 teacher_id 老师的所有课程的功能:

pub async fn get_courses_for_teacher(
    app_state: web::Data<AppState>,
    params: web::Path<i32>,
) -> HttpResponse {
    // let teacher_id: usize = params.0;
    let teacher_id: i32 = params.into_inner();
    
    let filtered_courses = app_state.courses
        .lock()
        .unwrap()
        .clone()
        .into_iter()
        .filter(|course| course.teacher_id == teacher_id)
        .collect::<Vec<Course>>();

    if filtered_courses.len() > 0 {
        HttpResponse::Ok().json(filtered_courses)
    } else {
        HttpResponse::Ok().json("No courses found for teacher".to_string())
    }
}

在 handlers.rs 的 test 模块为这个功能也添加一个测试:

    #[actix_rt::test]
    async fn get_all_courses_success() {
        let app_state: web::Data<AppState> = web::Data::new(AppState {
            health_check_response: "".to_string(),
            visit_count: Mutex::new(0),
            courses: Mutex::new(vec![]),
        });
        let teacher_id: web::Path<usize> = web::Path::from(1);
        let response = get_courses_for_teacher(app_state, teacher_id).await;
        assert_eq!(response.status(), StatusCode::OK);
    }

打开终端,cd 到 webservice,执行命令 cargo test,两个测试都通过了:

在这里插入图片描述

因为测试是顺序执行的,在第一个测试中插入了一个 teacher_id 为 1 的课程,所以第二个测试中查询 teacher_id 为 1 的课程可以通过。

我们在一个终端执行命令 cargo run 运行程序,然后在一个 bash 终端执行命令:

curl -X POST localhost:3000/courses/ -H "Content-Type: application/json" -d '{"teacher_id":1, "name":"First course"}'

来模拟一个添加课程的 POST 请求。

再在 bash 终端执行命令:

curl localhost:3000/courses/1

我们可以看到 bash 终端成功打印出了查询课程的信息:[{“teacher_id”:1,“id”:1,“name”:“First course”,“time”:“2025-07-10T11:47:21.168336600”}]

在这里插入图片描述

说明这个功能成功实现了。

查询指定老师的指定课程

修改 routers.rs 中的 course_routes 函数:

pub fn course_routes(cfg: &mut web::ServiceConfig) {
    cfg.service(
        web::scope("/courses")
            .route("/", web::post().to(new_course))
            .route("/{user_id}", web::get().to(get_courses_for_teacher))
            .route("/{user_id}/{course_id}", web::get().to(get_course_detail))
    );
}

新增 /courses/{user_id}/{course_id} 路径,对应一个 GET 请求,对应的 Handler 是 get_course_detail 函数。

修改 handlers.rs,get_course_detail 函数实现了查询指定 teacher_id 老师的指定 course_id 课程的功能:

pub async fn get_course_detail(
    app_state: web::Data<AppState>,
    params: web::Path<(i32, i32)>,
) -> HttpResponse {
    let (teacher_id, course_id) = params.into_inner();
    let selected_course = app_state.courses
        .lock()
        .unwrap()
        .clone()
        .into_iter()
        .find(|course| course.teacher_id == teacher_id && course.id == Some(course_id))
        .ok_or("Course not found");
    if let Ok(course) = selected_course {
        HttpResponse::Ok().json(course)
    } else {
        HttpResponse::Ok().json("Course not found".to_string())
    }
}

在 handlers.rs 的 test 模块为这个功能也添加一个测试:

    #[actix_rt::test]
    async fn get_one_course_success() {
        let app_state: web::Data<AppState> = web::Data::new(AppState {
            health_check_response: "".to_string(),
            visit_count: Mutex::new(0),
            courses: Mutex::new(vec![]),
        });
        let params: web::Path<(i32, i32)> = web::Path::from((1, 1));
        let resp = get_course_detail(app_state, params).await;
        assert_eq!(resp.status(), StatusCode::OK);
    }

打开终端,cd 到 webservice,执行命令 cargo test,三个测试都通过了:

在这里插入图片描述

我们在一个终端执行命令 cargo run 运行程序,然后在一个 bash 终端执行命令:

curl -X POST localhost:3000/courses/ -H "Content-Type: application/json" -d '{"teacher_id":1, "name":"First course"}'
curl -X POST localhost:3000/courses/ -H "Content-Type: application/json" -d '{"teacher_id":1, "name":"Second course"}'

来模拟两个添加课程的 POST 请求。

再在 bash 终端执行命令:

curl localhost:3000/courses/1/2

我们可以看到 bash 终端成功打印出了查询课程的信息:{“teacher_id”:1,“id”:2,“name”:“Second course”,“time”:“2025-07-10T11:51:18.608348”}

在这里插入图片描述

说明这个功能成功实现了。

尾声

视频教程 的写法,Course 结构体声明为:

#[derive(Deserialize, Serialize, Debug, Clone)]
pub struct Course {
    pub teacher_id: usize,
    pub id: Option<usize>,
    pub name: String,
    pub time: Option<NaiveDateTime>,
}

取值用的是 params.0,例如,handlers.rs 中的 get_courses_for_teacher 函数长这样:

pub async fn get_courses_for_teacher(
    app_state: web::Data<AppState>,
    params: web::Path<usize>,
) -> HttpResponse {
    let teacher_id: usize = params.0;

    let filtered_courses = app_state.courses
        .lock()
        .unwrap()
        .clone()
        .into_iter()
        .filter(|course| course.teacher_id == teacher_id)
        .collect::<Vec<Course>>();

    if filtered_courses.len() > 0 {
        HttpResponse::Ok().json(filtered_courses)
    } else {
        HttpResponse::Ok().json("No courses found for teacher".to_string())
    }
}

这样做会出现报错:

在这里插入图片描述

有几处不同的地方,改成前文我写的那样才行。

这里给出原版教程的官方源码地址:

https://github.com/peshwar9/rust-servers-services-apps/tree/master

后续如果还出现这种情况,建议对比源码看看。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

UestcXiye

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值