现在是时候开始进行数据库操作，以便实现功能了。本章将实现`TodoList`的功能。 > 本章代码在[04/数据库连接及模型定义](https://github.com/axumrs/todo/tree/04/%E6%95%B0%E6%8D%AE%E5%BA%93%E8%BF%9E%E6%8E%A5%E5%8F%8A%E6%A8%A1%E5%9E%8B%E5%AE%9A%E4%B9%89)分支。 ## 增加数据库配置开始之前，我们需要加入数据库配置。在[src/config.rs](https://github.com/axumrs/todo/blob/04/%E6%95%B0%E6%8D%AE%E5%BA%93%E8%BF%9E%E6%8E%A5%E5%8F%8A%E6%A8%A1%E5%9E%8B%E5%AE%9A%E4%B9%89/src/config.rs#L15)中修改： ```rust /// 应用配置 #[derive(Deserialize)] pub struct Config { pub web: WebConfig, pub pg: deadpool_postgres::Config, } ``` ## 状态共享为了在整个应用间共享数据库连接池，我们需要定义[状态共享](https://github.com/axumrs/todo/blob/04/%E6%95%B0%E6%8D%AE%E5%BA%93%E8%BF%9E%E6%8E%A5%E5%8F%8A%E6%A8%A1%E5%9E%8B%E5%AE%9A%E4%B9%89/src/model.rs#L4-L9)： ```rust /// 应用状态共享 #[derive(Clone)] pub struct AppState { /// PostgreSQL 连接池 pub pool: deadpool_postgres::Pool, } ``` 同时，我们需要在`main`函数中[实例化](https://github.com/axumrs/todo/blob/04/%E6%95%B0%E6%8D%AE%E5%BA%93%E8%BF%9E%E6%8E%A5%E5%8F%8A%E6%A8%A1%E5%9E%8B%E5%AE%9A%E4%B9%89/src/main.rs#L24-L27)并[增加这个共享](https://github.com/axumrs/todo/blob/04/%E6%95%B0%E6%8D%AE%E5%BA%93%E8%BF%9E%E6%8E%A5%E5%8F%8A%E6%A8%A1%E5%9E%8B%E5%AE%9A%E4%B9%89/src/main.rs#L41)： ```rust #[tokio::main] async fn main() { // ... let pool = cfg .pg .create_pool(tokio_postgres::NoTls) .expect("初始化数据库连接池失败"); let app = Router::new() // ... .layer(AddExtensionLayer::new(AppState { pool })); // ... } ``` ## TodoList 模型我们定义两个[TodoList 相关的模型](https://github.com/axumrs/todo/blob/04/%E6%95%B0%E6%8D%AE%E5%BA%93%E8%BF%9E%E6%8E%A5%E5%8F%8A%E6%A8%A1%E5%9E%8B%E5%AE%9A%E4%B9%89/src/model.rs#L11-L24)： - `TodoList`：用于映射完整的 TodoList - `TodoListID`：用于映射新插入的 TodoList 的 ID ```rust /// 待办列表模型 #[derive(PostgresMapper, Serialize)] #[pg_mapper(table = "todo_list")] pub struct TodoList { pub id: i32, pub title: String, } /// 待办列表新ID模型 #[derive(PostgresMapper, Serialize)] #[pg_mapper(table = "todo_list")] pub struct TodoListID { pub id: i32, } ``` ## TodoList 数据库操作在 `src/db/todo_list.rs`中，我们为 TodoList 实现数据库相关操作： ### `create`：创建一个 TodoList ```rust pub async fn create(client: &Client, frm: form::CreateTodoList) -> Result { let stmt = client .prepare("INSERT INTO todo_list (title) VALUES ($1) RETURNING id") .await .map_err(AppError::from)?; let result = client .query(&stmt, &[&frm.title]) .await .map_err(AppError::from)? .iter() .map(|row| TodoListID::from_row_ref(row).unwrap()) .collect::>() .pop() .ok_or(AppError::not_found())?; Ok(result) } ``` 这里使用了一个名为`CreateTodoList`的结构体，它定义在`form`模块： ```rust #[derive(Deserialize)] pub struct CreateTodoList { pub title: String, } ``` ### `all`：所有 TodoList 的列表 ```rust pub async fn all(client: &Client) -> Result> { let stmt = client .prepare("SELECT id,title FROM todo_list ORDER BY id DESC") .await .map_err(AppError::from)?; let result = client .query(&stmt, &[]) .await .map_err(AppError::from)? .iter() .map(|row| TodoList::from_row_ref(row).unwrap()) .collect::>(); Ok(result) } ``` ### `find`：根据 ID 查找 TodoList ```rust pub async fn find(client: &Client, list_id: i32) -> Result { let stmt = client .prepare("SELECT id,title FROM todo_list WHERE id=$1") .await .map_err(AppError::from)?; let result = client .query(&stmt, &[&list_id]) .await .map_err(AppError::from)? .iter() .map(|row| TodoList::from_row_ref(row).unwrap()) .collect::>() .pop() .ok_or(AppError::not_found())?; Ok(result) } ``` ### `update`：修改 TodoList ```rust pub async fn update(client: &Client, frm: form::UpdateTodoList) -> Result { let stmt = client .prepare("UPDATE todo_list SET title=$1 WHERE id=$2") .await .map_err(AppError::from)?; let result = client .execute(&stmt, &[&frm.title, &frm.id]) .await .map_err(AppError::from)?; Ok(result > 0) } ``` 这里使用了一个名为 `UpdateTodoList` 的结构体，它定义在 `form` 模块： ```rust /// 修改待办列表 #[derive(Deserialize)] pub struct UpdateTodoList { pub id: i32, pub title: String, } ``` ### `delete`：删除 TodoList 及其相关的 TodoItem ```rust pub async fn delete(client: &mut Client, id: i32) -> Result { let tx = client.transaction().await.map_err(AppError::from)?; let stmt = tx .prepare("DELETE FROM todo_list WHERE id=$1") .await .map_err(AppError::from)?; let result = tx.execute(&stmt, &[&id]).await; if let Err(err) = result { tx.rollback().await.map_err(AppError::from)?; return Err(AppError::db_error(err)); }; let stmt = tx .prepare("DELETE FROM todo_item WHERE list_id=$1") .await .map_err(AppError::from)?; let result = tx.execute(&stmt, &[&id]).await; if let Err(err) = result { tx.rollback().await.map_err(AppError::from)?; return Err(AppError::db_error(err)); }; tx.commit().await.map_err(AppError::from)?; Ok(true) } ``` 由于要同时对两个表进行操作，为了保证操作的原子性，这里使用了事务。 ## TodoList 的 handler 首先，我们对`handler`模块进行了拆分，把之前的`src/handle.rs`单一文件，拆分成了目录，并在其中增加了`todo_list`模块。下面以`create`和`find`来对 handler 进行简要说明。 ### `create` 函数 #### 参数 - `Extension(state): Extension`：获取共享状态，里面包含有数据库连接池 - `Json(payload): Json`：以 JSON 方式从客户端获取输入的数据，并将其反序化成 `CreateTodoList` 结构体 #### 返回值 - `crate::Result`：这是我们自定义的`Result`，再次提示，它包含了我们自定义错误`AppError` - `Json`：axum 提供的功能，将泛型的数据自定义序列化成 JSON 格式 - `Response`：我们在上一章定义的响应 - `TodoListID`：里面包含了新增加的 TodoList 的 ID #### 错误转换在获取数据库客户端时，我们使用了`map_err`将某个类型的错误转换成了`AppError`，这是 Rust 编程中的常用技巧。 ```rust pub async fn create( Extension(state): Extension, Json(payload): Json, ) -> Result>> { let client = state.pool.get().await.map_err(AppError::from)?; let result = todo_list::create(&client, payload).await?; Ok(Json(Response::ok(result))) } ``` ### `find` 函数 #### 参数除了共享状态之外，我们还通过 `Path`来获取到了客户端传递过来的 TodoList 的 ID： ```rust Path(list_id): Path ``` #### 返回值这次，我们要返回的是`TodoList`，它包含了完整的 TodoList 数据。 ```rust pub async fn find( Extension(state): Extension, Path(list_id): Path, ) -> Result>> { let client = state.pool.get().await.map_err(AppError::from)?; let result = todo_list::find(&client, list_id).await?; Ok(Json(Response::ok(result))) } ``` ## 错误转换正如在讲解`create` handler 时所说，我们经常在 Rust 开发中使用`map_err`，它的作用是：将某个类型为`S`的错误，转换成类型为`E`的错误。比如： ```rust let client = state.pool.get().await.map_err(AppError::from)?; ``` 从连接池获取客户端时，可能发生错误，这个错误是由 deadpool-postgres 定义的`PoolError`，我们需要对其进行转换： ```rust let client = state.pool.get().await.map_err(|err| AppError{ cause: Some(err.to_string()), // 其它字段省略 })?; ``` 但我们的代码中，使用是 `AppError::from`。那是因为我们为 `AppError` 实现了 `From` trait： ```rust impl From for AppError { fn from(err: deadpool_postgres::PoolError) -> Self { Self::db_error(err) } } impl From for AppError { fn from(err: tokio_postgres::Error) -> Self { Self::db_error(err) } } ``` - 实现`From`：可以直接在`map_err`中使用`AppError::from`将`deadpool_postgres::PoolError`转换成`AppError`。在尝试从连接池中获取 Client 时，有可能出现这个错误。 - 实现`From`：可以直接在`map_err`中使用`AppError::from`将`tokio_postgres::Error`转换成`AppError`。在数据库操作时（比如`prepare`、`query`、`execute`、`rollback`、`commit`）有可能出现这个错误。虽然我们实现了 TodoList 的功能，但存在一些问题： - 重复代码太多 - 未对错误进行日志记录下一章我们将对代码进行重构，以及记录可能发生的错误。

标签

模型

数据库、模型、状态共享及TodoList