构建一个rust生产应用读书笔记四(实战4)

embedded/2024/12/20 3:02:30/

上一节我们已经将数据的基本配置完成,这一节我们继续完成邮件订阅功能的开发,这两天按照书中的代码编写终于调试成功了,把这其中遇到的问题总结记录下来

版本问题

在调试的过程中遇到了各种各样的版本问题,目前这个版本依赖可以编译通过,如果在学习过程中也遇到syn,macors等版本问题,可以参考如下版本

Cargo.toml

[package]
name = "zero2prod"
version = "0.1.0"
edition = "2021"[lib]
path = "src/lib.rs"[[bin]]
path = "src/main.rs"
name = "zero2prod"[dependencies]
actix-web = "4"
tokio = { version = "1", features = ["macros", "rt-multi-thread"] }
serde = "1.0.115"
config = { version = "0.14", default-features = false, features = ["yaml"] }
sqlx = { version = "0.8", default-features = false, features = ["runtime-tokio-rustls","macros","postgres","uuid","chrono","migrate",
] }
uuid = { version = "1", features = ["v4"] }
chrono = { version = "0.4.22", default-features = false, features = ["clock"] }[dev-dependencies]
reqwest = { version = "0.12", features = ["json"] }# On MacOS, `brew install michaeleisel/zld/zld`
# [target.x86_64-apple-darwin]
# rustflags = ["-C", "link-arg=-fuse-ld=/usr/local/bin/zld"]
# [target.aarch64-apple-darwin]
# rustflags = ["-C", "link-arg=-fuse-ld=/usr/local/bin/zld"]

有时候还会遇到编译问题,我通常执行cargo clean, cargo update来解决。

有时候,更新到最新版本的 Cargo 和 Rust 工具链也可以解决构建问题

执行

rustup update

使用配置文件yaml

在实际的项目中,将数据库的配置文件,服务器的ip端口统一在配置文件里面管理,方便应用部署的时候根据实际的情况调整

// ! configuration.yaml
application_port: 8000
database:host: '127.0.0.1'port: 5432username: 'postgres'password: 'postgres'database_name: 'newsletter'

配置文件读取

rust">//! src/configuration
#[derive(serde::Deserialize)]
pub struct Settings {pub database: DatabaseSettings,pub application_port: u16,
}#[derive(serde::Deserialize, Clone)]
pub struct DatabaseSettings {pub username: String,pub password: String,pub port: u16,pub host: String,pub database_name: String,
}pub fn get_configuration() -> Result<Settings, config::ConfigError> {let settings = config::Config::builder().add_source(config::File::new("configuration.yaml",config::FileFormat::Yaml,)).build()?;settings.try_deserialize::<Settings>()
}impl DatabaseSettings {pub fn connection_string(&self) -> String {format!("postgres://{}:{}@{}:{}/{}",self.username, self.password, self.host, self.port, self.database_name)}
}

函数说明

get_configuration 函数用于从 configuration.yaml 文件中读取配置,并将其解析为 Settings 结构体。

  • 使用 config::Config::builder() 创建一个配置构建器。
  • 使用 add_source 方法添加一个配置源,这里指定了配置文件的路径和格式(YAML)。
  • 调用 build 方法构建配置对象,因依赖库版本升级后原文提供的方法已过时,和这里不一样但实现功能一样。
  • 使用 try_deserialize 方法将配置对象反序列化为 Settings 结构体。
  • 如果任何步骤失败,函数将返回一个 Result 类型的 Err 变体,其中包含 config::ConfigError

注意的是:struct都要记得加上序列化attribute

lib.rs

在 lib.rs 文件中声明模块是常见的方式, 这些模块通常是项目中不同功能的逻辑划分,根据当前应用分成3个模块

rust">pub mod configuration;
pub mod routes;
pub mod startup;

routes

根据之前的代码,我们把代码重构,将health_check() 挪到health_check.rs中

rust">//! src/routes/health_check.rs
use actix_web::HttpResponse;pub async fn health_check() -> HttpResponse {HttpResponse::Ok().finish()
}

routes 模块化

rust">//! src/routes/mod.rs
mod health_check;
mod subscriptions;pub use health_check::*;
pub use subscriptions::*;

重构subscriptions.rs,把订购的用户信息存入到数据库

rust">use core::result::Result::{Err, Ok};use actix_web::{web, HttpResponse};
use chrono::Utc;
use sqlx::PgPool;
use uuid::Uuid;#[derive(serde::Deserialize)]
pub struct FormData {email: String,name: String,
}pub async fn subscribe(form: web::Form<FormData>, pool: web::Data<PgPool>) -> HttpResponse {match sqlx::query!(r#"insert into subscriptions (id,email,name,subscribed_at) values($1,$2,$3,$4)"#,Uuid::new_v4(),form.email,form.name,Utc::now()).execute(pool.as_ref()).await{Ok(_) => HttpResponse::Ok().finish(),Err(e) => {print!("Failed to execute query:{}", e);HttpResponse::InternalServerError().finish()}}
}

代码说明

  • subscribe 函数是一个异步处理函数,用于处理用户订阅请求。
  • form: web::Form<FormData>: 接收一个 web::Form<FormData> 类型的参数,类似于DTO对象,表示从 HTTP 请求中解析出来的表单数据。
  • pool: web::Data<PgPool>: 接收一个 web::Data<PgPool> 类型的参数,表示数据库连接池。
  • sqlx::query! 宏用于构建 SQL 查询,并绑定参数。该方法是一种预处理sql的方式。
  • Uuid::new_v4(): 生成一个唯一的 UUID 作为订阅 ID。
  • Utc::now(): 获取当前的 UTC 时间,用于记录订阅时间。
  • execute(pool.as_ref()): 执行 SQL 查询,并传递数据库连接池的引用,传递引用可以避免不必要的克隆操作,提高性能。PgPool 本身是一个较重的对象,包含了许多内部状态和资源管理逻辑。如果每次调用 execute 都克隆一次 PgPool,不仅会增加内存开销,还可能导致性能下降。而且你可以在不改变所有权的情况下使用资源。
  • match 表达式用于处理查询结果:
    • 如果查询成功(Ok),返回 HttpResponse::Ok().finish(),表示请求处理成功。
    • 如果查询失败(Err),打印错误信息并返回 HttpResponse::InternalServerError().finish(),表示服务器内部错误。

主函数

rust">//! main.rs
use sqlx::postgres::PgPool;
use std::net::TcpListener;
use zero2prod::configuration::get_configuration;
use zero2prod::startup::run;#[tokio::main]
async fn main() -> std::io::Result<()> {let configuration = get_configuration().expect("Failed to read configuration.");let connection_pool = PgPool::connect(&configuration.database.connection_string()).await.expect("Failed to connect to Postgres.");let address = format!("127.0.0.1:{}", configuration.application_port);let listener = TcpListener::bind(address)?;run(listener, connection_pool)?.await?;Ok(())
}

startup里面监听和路由注册

rust">use crate::routes::{ health_check, subscribe };
use actix_web::dev::Server;
use actix_web::web::Data;
use actix_web::{ web, App, HttpServer };
use sqlx::PgPool;
use std::net::TcpListener;pub fn run(listener: TcpListener, db_pool: PgPool) -> Result<Server, std::io::Error> {let db_pool = Data::new(db_pool);let server = HttpServer::new(move || {App::new().route("/health_check", web::get().to(health_check)).route("/subscriptions", web::post().to(subscribe)).app_data(db_pool.clone())}).listen(listener)?.run();Ok(server)
}

函数说明

  • run 函数是一个启动 HTTP 服务器的函数,它接受一个 TcpListener 和一个 PgPool 作为参数。
  • let db_pool = Data::new(db_pool);: 将 PgPool 包装成 Data 类型,以便在应用中共享。
  • HttpServer::new(move || { ... }): 创建一个新的 HttpServer 实例,并传入一个闭包来配置应用。前面的文章提到过,闭包暂时可以理解为java中的lambda表达式
  • move || { ... }: 使用 move 关键字将闭包内的变量捕获到闭包中,确保闭包拥有这些变量的所有权。
  • App::new(): 创建一个新的 App 实例。
  • .route("/health_check", web::get().to(health_check)): 配置一个 GET 路由 /health_check,并将其映射到 health_check 处理函数。
  • .route("/subscriptions", web::post().to(subscribe)): 配置一个 POST 路由 /subscriptions,并将其映射到 subscribe 处理函数。
  • .app_data(db_pool.clone()): 将 db_pool 作为应用数据注册到 App 中,使其在处理函数中可用。

代码改进

以上代码在测试的过程中可以做如下的改进,后面会继续优化

  1. 日志记录:添加日志记录,以便在启动和运行时记录重要信息。
  2. 错误处理:使用更详细的错误处理和日志记录,以便更好地调试和监控。
  3. 配置文件:确保配置文件路径和格式正确,避免因配置文件问题导致启动失败。

测试结果

运行cargo run

在api调试工具中添加email和name参数

查看数据库结果

单元测试代码

rust">use sqlx::{Connection, Executor, PgConnection, PgPool};
use std::net::TcpListener;
use uuid::Uuid;
use zero2prod::configuration::{get_configuration, DatabaseSettings};
use zero2prod::startup::run;pub struct TestApp {pub address: String,pub db_pool: PgPool,
}async fn spawn_app() -> TestApp {let listener = TcpListener::bind("127.0.0.1:0").expect("Failed to bind random port");// We retrieve the port assigned to us by the OSlet port = listener.local_addr().unwrap().port();let address = format!("http://127.0.0.1:{}", port);let mut configuration = get_configuration().expect("Failed to read configuration.");configuration.database.database_name = Uuid::new_v4().to_string();let connection_pool = configure_database(&configuration.database).await;let server = run(listener, connection_pool.clone()).expect("Failed to bind address");let _ = tokio::spawn(server);TestApp {address,db_pool: connection_pool,}
}pub async fn configure_database(config: &DatabaseSettings) -> PgPool {// Create databaselet maintenance_settings = DatabaseSettings {database_name: "postgres".to_string(),username: "postgres".to_string(),password: "password".to_string(),..config.clone()};let mut connection = PgConnection::connect(&maintenance_settings.connection_string()).await.expect("Failed to connect to Postgres");connection.execute(format!(r#"CREATE DATABASE "{}";"#, config.database_name).as_str()).await.expect("Failed to create database.");// Migrate databaselet connection_pool = PgPool::connect(&config.connection_string()).await.expect("Failed to connect to Postgres.");sqlx::migrate!("./migrations").run(&connection_pool).await.expect("Failed to migrate the database");connection_pool
}#[tokio::test]
async fn health_check_works() {// Arrangelet app = spawn_app().await;let client = reqwest::Client::new();// Actlet response = client// Use the returned application address.get(&format!("{}/health_check", &app.address)).send().await.expect("Failed to execute request.");// Assertassert!(response.status().is_success());assert_eq!(Some(0), response.content_length());
}#[tokio::test]
async fn subscribe_returns_a_200_for_valid_form_data() {// Arrangelet app = spawn_app().await;let client = reqwest::Client::new();let body = "name=le%20guin&email=ursula_le_guin%40gmail.com";// Actlet response = client.post(&format!("{}/subscriptions", &app.address)).header("Content-Type", "application/x-www-form-urlencoded").body(body).send().await.expect("Failed to execute request.");// Assertassert_eq!(200, response.status().as_u16());let saved = sqlx::query!("SELECT email, name FROM subscriptions",).fetch_one(&app.db_pool).await.expect("Failed to fetch saved subscription.");assert_eq!(saved.email, "ursula_le_guin@gmail.com");assert_eq!(saved.name, "le guin");
}#[tokio::test]
async fn subscribe_returns_a_400_when_data_is_missing() {// Arrangelet app = spawn_app().await;let client = reqwest::Client::new();let test_cases = vec![("name=le%20guin", "missing the email"),("email=ursula_le_guin%40gmail.com", "missing the name"),("", "missing both name and email"),];for (invalid_body, error_message) in test_cases {// Actlet response = client.post(&format!("{}/subscriptions", &app.address)).header("Content-Type", "application/x-www-form-urlencoded").body(invalid_body).send().await.expect("Failed to execute request.");// Assertassert_eq!(400,response.status().as_u16(),// Additional customised error message on test failure"The API did not fail with 400 Bad Request when the payload was {}.",error_message);}
}

注:各位亲爱的小伙伴们,今年是我从事软件行业的第20年,通过博客记录的方式将我知道的、理解的、有帮助的都分享给大家。同时,也提供就业指导,专业简历优化服务。你们的支持是我最大的动力 


http://www.ppmy.cn/embedded/147168.html

相关文章

基于Spring Boot的小区车辆管理系统

一、系统背景与目的 随着城市化进程的加快&#xff0c;小区内的车辆数量急剧增加&#xff0c;车辆管理问题日益凸显。传统的车辆管理方式存在效率低、易出错、信息不透明等问题。为了解决这些问题&#xff0c;基于Spring Boot的小区车辆管理系统应运而生。该系统旨在通过信息化…

apache中的Worker 和 Prefork 之间的区别是什么?

文章目录 内存使用稳定性兼容性适用场景 Apache中的Worker和Prefork两种工作模式在内存使用、稳定性以及兼容性等方面存在区别 内存使用 Worker&#xff1a;由于使用线程&#xff0c;内存占用较少。Prefork&#xff1a;每个进程独立运行&#xff0c;内存消耗较大。 稳定性 W…

在 SQL Server 中获取指定字符所在有位置索引

在 SQL Server 中&#xff0c;要获取指定字符在字符串中的位置&#xff0c;可以使用 CHARINDEX 函数。这个函数返回一个整数&#xff0c;表示子字符串在字符串或表达式中的开始位置。如果没有找到子字符串&#xff0c;则函数结果为0。以下是 CHARINDEX 函数的基本语法&#xff…

31.攻防世界php_rce

进入场景 不是&#xff0c;这是个啥啊 index.php?sindex/think\app/invokefunction&functioncall_user_func_array&vars[0]system&vars[1][]whoami index.php?sindex/think\app/invokefunction&functioncall_user_func_array&vars[0]system&vars[1]…

vue3使用自定义指令简单实现一个可拖拽弹窗组件

vue3使用自定义指令简单实现一个可拖拽弹窗组件 <!DOCTYPE html> <html lang"zh-CN"><head><meta charset"UTF-8"><meta name"viewport" content"widthdevice-width, initial-scale1.0"><title>可…

概率论得学习和整理27:关于离散的数组 随机变量数组的均值,方差的求法3种公式,思考和细节。

目录 1 例子1&#xff1a;最典型的&#xff0c;最简单的数组的均值&#xff0c;方差的求法 2 例子1的问题&#xff1a;例子1只是1个特例&#xff0c;而不是普遍情况。 2.1 例子1各种默认假设&#xff0c;导致了求均值和方差的特殊性&#xff0c;特别简单。 2.2 我觉得 加权…

13.继承和多态的实例 C#

这是一个动物园的动物发出不同的声音&#xff0c;使用了继承和多态 using System; using System.Collections.Generic;namespace InheritanceAndPolymorphismExample {//一个动物类&#xff0c;包含属性&#xff1a;名称。包含方法&#xff1a;发出叫声public class Animal{pub…

Element@2.15.14-tree checkStrictly 状态实现父项联动子项,实现节点自定义编辑、新增、删除功能

背景&#xff1a;现在有一个新需求&#xff0c;需要借助树结构来实现词库的分类管理&#xff0c;树的节点是不同的分类&#xff0c;不同的分类可以有自己的词库&#xff0c;所以父子节点是互不影响的&#xff1b;同样为了选择的方便性&#xff0c;提出了新需求&#xff0c;选择…