Rust语言从入门到精通系列 - SeaORM框架实践(基础篇)

SeaORM是一个基于Rust语言的ORM（对象关系映射）框架，它提供了一种简单的方式来操作SQL数据库。SeaORM的设计理念是将SQL查询和Rust代码结合在一起，从而提供更好的类型安全和代码可读性。

在本教程中，我们将介绍SeaORM的基本用法和进阶用法。我们将使用SQLite数据库来演示这些用法。

基础用法#

在使用SeaORM之前，我们需要将其添加到我们的Rust项目中。cargo.toml添加依赖：

1
sea-orm = "0.11.3"

连接到数据库#

在使用SeaORM之前，我们需要连接到一个数据库。我们可以使用DatabaseConnection结构体来完成这个任务。

1
use sea_orm::Database;
2

3
#[async_std::main]
4
async fn main() -> Result<(), Box<dyn std::error::Error>> {
5
    let db_conn = Database::connect("mysql://username:password@hostname:port/database").await?;
6
    // ...
7
    Ok(())
8
}

在上面的代码中，我们使用DatabaseConnectionInfo::from_env()方法从环境变量中获取数据库连接信息，并使用DatabaseConnection::connect()方法连接到数据库。

定义表#

在SeaORM中，我们可以使用sea_query::Table结构体来定义表。我们可以定义表的名称、列名和列类型。

1
use sea_query::{ColumnDef, ColumnType, Table};
2

3
let users = Table::new("users")
4
    .add_column("id", ColumnType::Int(Some(11)).Unsigned(true).NotNull(true).AutoIncrement(true))
5
    .add_column("name", ColumnType::String(Some(256)).NotNull(true))
6
    .add_column("email", ColumnType::String(Some(256)).NotNull(true).Unique(true))
7
    .add_column("age", ColumnType::Int(Some(3)).Unsigned(true).NotNull(true))
8
    .set_primary_key(vec!["id"]);

在上面的代码中，我们定义了一个名为users的表，它有四个列：id、name、email和age。id列是自增的主键，name和email列是字符串类型，age列是整数类型。

插入数据#

在SeaORM中，我们可以使用InsertStatement结构体来插入数据。我们可以使用values()方法来设置要插入的值。

1
use sea_orm::{entity::*, query::*, DatabaseConnection};
2

3
let user = User {
4
    name: "John Doe".to_owned(),
5
    email: "john.doe@example.com".to_owned(),
6
    age: 30,
7
};
8

9
let insert = Insert::single_into(User::table())
10
    .values(user.clone())
11
    .build();
12

13
let result = User::insert(insert).exec(&db_conn).await?;

在上面的代码中，我们使用Insert::single_into()方法和values()方法来设置要插入的值。我们使用User::table()方法来获取User实体的表格。最后，我们使用User::insert()方法和exec()方法来执行插入操作。

查询数据#

在SeaORM中，我们可以使用SelectStatement结构体来查询数据。我们可以使用from()方法来设置要查询的表格，使用column()方法来设置要查询的列，使用where_expr()方法来设置查询条件。

1
use sea_orm::{entity::*, query::*, DatabaseConnection};
2

3
let query = Select::from_table(User::table())
4
    .column(User::all_columns)
5
    .where_expr(User::email.eq("john.doe@example.com"))
6
    .build();
7

8
let result = User::find_by_sql(query).one(&db_conn).await?;

在上面的代码中，我们使用Select::from_table()方法和column()方法来设置要查询的表格和列。我们使用where_expr()方法来设置查询条件。最后，我们使用User::find_by_sql()方法和one()方法来查询数据。

更新数据#

在SeaORM中，我们可以使用UpdateStatement结构体来更新数据。我们可以使用set()方法来设置要更新的值，使用where_expr()方法来设置更新条件。

1
use sea_orm::{entity::*, query::*, DatabaseConnection};
2

3
let update = Update::table(User::table())
4
    .set(User::name, "Jane Doe")
5
    .where_expr(User::email.eq("john.doe@example.com"))
6
    .build();
7

8
let result = User::update(update).exec(&db_conn).await?;

在上面的代码中，我们使用Update::table()方法和set()方法来设置要更新的值。我们使用where_expr()方法来设置更新条件。最后，我们使用User::update()方法和exec()方法来执行更新操作。

删除数据#

在SeaORM中，我们可以使用DeleteStatement结构体来删除数据。我们可以使用where_expr()方法来设置删除条件。

1
use sea_orm::{entity::*, query::*, DatabaseConnection};
2

3
let delete = Delete::from_table(User::table())
4
    .where_expr(User::email.eq("john.doe@example.com"))
5
    .build();
6

7
let result = User::delete(delete).exec(&db_conn).await?;

在上面的代码中，我们使用Delete::from_table()方法和where_expr()方法来设置删除条件。最后，我们使用User::delete()方法和exec()方法来执行删除操作。

进阶用法#

关联表查询#

在SeaORM中，我们可以使用JoinType枚举来设置关联类型。我们可以使用JoinOn结构体来设置关联条件。

1
use sea_orm::{entity::*, query::*, DatabaseConnection};
2

3
let query = Select::from_table(User::table())
4
    .inner_join(Post::table(), JoinOn::new(User::id, Post::user_id))
5
    .column((User::name, Post::title))
6
    .build();
7

8
let result = User::find_by_sql(query).all(&db_conn).await?;

在上面的代码中，我们使用Select::from_table()方法和inner_join()方法来设置关联表格。我们使用JoinOn::new()方法来设置关联条件。最后，我们使用User::find_by_sql()方法和all()方法来查询数据。

事务处理#

在SeaORM中，我们可以使用Transaction结构体来处理事务。我们可以使用begin()方法来开始事务，使用commit()方法来提交事务，使用rollback()方法来回滚事务。

1
use sea_orm::{entity::*, query::*, DatabaseConnection, Transaction};
2

3
let tx = Transaction::new(&db_conn).await?;
4

5
let user = User {
6
    name: "John Doe".to_owned(),
7
    email: "john.doe@example.com".to_owned(),
8
    age: 30,
9
};
10

11
let insert = Insert::single_into(User::table())
12
    .values(user.clone())
13
    .build();
14

15
let result = User::insert(insert).exec(&tx).await?;
16

17
let update = Update::table(User::table())
18
    .set(User::name, "Jane Doe")
19
    .where_expr(User::email.eq("john.doe@example.com"))
20
    .build();
21

22
let result = User::update(update).exec(&tx).await?;
23

24
tx.commit().await?;

在上面的代码中，我们使用Transaction::new()方法来开始事务。我们使用User::insert()方法和exec()方法来插入数据，使用User::update()方法和exec()方法来更新数据。最后，我们使用tx.commit()方法来提交事务。

总结#

在本教程中，我们介绍了SeaORM的基本用法和进阶用法。SeaORM提供了一种简单的方式来操作SQL数据库，它将SQL查询和Rust代码结合在一起，提供了更好的类型安全和代码可读性。通过本教程的学习，我们可以更好地理解SeaORM的使用方法，从而更好地使用它来开发Rust应用程序。

音乐

音乐