Rust语言从入门到精通系列 - SQLx模块SQLite入门
SQLx是一个Rust语言的异步SQL数据库访问库，支持多种数据库，包括PostgreSQL、MySQL、SQLite等。本教程将以SQLite为例，介绍SQLx的基础用法和进阶用法。
基础用法
连接数据库
首先，需要在Rust项目中添加SQLx库的依赖：
1
2
3
4
5
[dependencies]
sqlx = "0.6"
sqlx-core = "0.6"
sqlx-derive = "0.6"
sqlx-macros = "0.6"
然后，可以使用以下代码连接SQLite数据库：
1
2
3
4
5
6
7
8
9
10
use sqlx::{SqlitePool, sqlite::SqliteConnectOptions};
#[tokio::main]
async fn main() -> Result<(), sqlx::Error> {
let database_url = "sqlite:mydatabase.db";
let options = SqliteConnectOptions::new()
.filename(database_url);
let pool = SqlitePool::connect_with(options).await?;
Ok(())
}
这里使用了SqlitePool连接池，可以在多个线程中共享连接。SqliteConnectOptions用于配置连接选项，这里指定了SQLite数据库文件的路径。
创建表
下面的代码演示了如何使用SQLx创建一个名为users的表：
1
2
3
4
5
6
7
8
9
10
11
12
use sqlx::{query, SqlitePool};
#[tokio::main]
async fn main() -> Result<(), sqlx::Error> {
let pool = SqlitePool::connect("sqlite:mydatabase.db").await?;
query("CREATE TABLE users (
id INTEGER PRIMARY KEY,
name TEXT NOT NULL,
email TEXT NOT NULL UNIQUE
)").execute(&pool).await?;
Ok(())
}
这里使用了query宏执行SQL语句，execute方法用于执行语句。可以看到，SQL语句与普通的SQL语句没有太大区别。
插入数据
下面的代码演示了如何使用SQLx向users表中插入一条数据：
1
2
3
4
5
6
7
8
9
10
11
12
use sqlx::{query, SqlitePool};
#[tokio::main]
async fn main() -> Result<(), sqlx::Error> {
let pool = SqlitePool::connect("sqlite:mydatabase.db").await?;
query("INSERT INTO users (name, email) VALUES (?, ?)")
.bind("Alice")
.bind("alice@example.com")
.execute(&pool)
.await?;
Ok(())
}
这里使用了bind方法绑定参数，可以避免SQL注入攻击。
查询数据
下面的代码演示了如何使用SQLx查询users表中所有数据：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
use sqlx::{query_as, SqlitePool};
#[derive(Debug, sqlx::FromRow)]
struct User {
id: i32,
name: String,
email: String,
}
#[tokio::main]
async fn main() -> Result<(), sqlx::Error> {
let pool = SqlitePool::connect("sqlite:mydatabase.db").await?;
let users = query_as::<_, User>("SELECT * FROM users")
.fetch_all(&pool)
.await?;
println!("{:?}", users);
Ok(())
}
这里使用了query_as宏查询数据，并使用FromRow特性将查询结果转换为User结构体。fetch_all方法用于获取所有查询结果。
更新数据
下面的代码演示了如何使用SQLx更新users表中的数据：
1
2
3
4
5
6
7
8
9
10
11
12
use sqlx::{query, SqlitePool};
#[tokio::main]
async fn main() -> Result<(), sqlx::Error> {
let pool = SqlitePool::connect("sqlite:mydatabase.db").await?;
query("UPDATE users SET email = ? WHERE name = ?")
.bind("alice@example.org")
.bind("Alice")
.execute(&pool)
.await?;
Ok(())
}
这里使用了UPDATE语句更新数据。
删除数据
下面的代码演示了如何使用SQLx删除users表中的数据：
1
2
3
4
5
6
7
8
9
10
11
use sqlx::{query, SqlitePool};
#[tokio::main]
async fn main() -> Result<(), sqlx::Error> {
let pool = SqlitePool::connect("sqlite:mydatabase.db").await?;
query("DELETE FROM users WHERE name = ?")
.bind("Alice")
.execute(&pool)
.await?;
Ok(())
}
这里使用了DELETE语句删除数据。
进阶用法
事务
下面的代码演示了如何使用SQLx进行事务操作：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
use sqlx::{query, SqlitePool, SqliteTransaction};
#[tokio::main]
async fn main() -> Result<(), sqlx::Error> {
let pool = SqlitePool::connect("sqlite:mydatabase.db").await?;
let mut tx = pool.begin().await?;
query("INSERT INTO users (name, email) VALUES (?, ?)")
.bind("Alice")
.bind("alice@example.com")
.execute(&mut tx)
.await?;
query("INSERT INTO users (name, email) VALUES (?, ?)")
.bind("Bob")
.bind("bob@example.com")
.execute(&mut tx)
.await?;
tx.commit().await?;
Ok(())
}
这里使用了begin方法开启一个事务，可以在事务中执行多条SQL语句。如果所有语句执行成功，可以使用commit方法提交事务，否则可以使用rollback方法回滚事务。
批量插入
下面的代码演示了如何使用SQLx进行批量插入操作：
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
use sqlx::{query, SqlitePool};
#[tokio::main]
async fn main() -> Result<(), sqlx::Error> {
let pool = SqlitePool::connect("sqlite:mydatabase.db").await?;
let users = vec![
("Alice", "alice@example.com"),
("Bob", "bob@example.com"),
("Charlie", "charlie@example.com"),
];
query("INSERT INTO users (name, email) VALUES (?, ?)")
.bind_all(users)
.execute(&pool)
.await?;
Ok(())
}
这里使用了bind_all方法绑定多个参数，可以将多个参数一次性绑定到SQL语句中，避免了多次执行SQL语句的开销。
总结
本教程介绍了SQLx的基础用法和进阶用法，包括连接数据库、创建表、插入数据、查询数据、更新数据、删除数据、事务和批量插入。SQLx是一个非常方便的异步SQL数据库访问库，可以大大提高开发效率。
本作品采用 知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议 进行许可。欢迎转载、使用、重新发布，但务必保留文章署名 TinyZ Zzh （包含链接： https://tinyzzh.github.io ），不得用于商业目的，基于本文修改后的作品务必以相同的许可发布。 如有任何疑问，请 与我联系 (tinyzzh815@gmail.com) 。
评论