Rust 中如何使用 Diesel 或 sqlx 进行数据库操作？

推荐答案

使用 Diesel 进行数据库操作

// www.javascriptcn.com code example
use diesel::prelude::*;
use diesel::sqlite::SqliteConnection;
use dotenv::dotenv;
use std::env;

pub fn establish_connection() -> SqliteConnection {
    dotenv().ok();

    let database_url = env::var("DATABASE_URL").expect("DATABASE_URL must be set");
    SqliteConnection::establish(&database_url)
        .unwrap_or_else(|_| panic!("Error connecting to {}", database_url))
}

#[derive(Queryable)]
pub struct Post {
    pub id: i32,
    pub title: String,
    pub body: String,
    pub published: bool,
}

pub fn load_posts(connection: &SqliteConnection) -> Vec<Post> {
    use crate::schema::posts::dsl::*;

    posts.filter(published.eq(true))
        .load::<Post>(connection)
        .expect("Error loading posts")
}

使用 sqlx 进行数据库操作

// www.javascriptcn.com code example
use sqlx::sqlite::SqlitePool;
use sqlx::FromRow;
use dotenv::dotenv;
use std::env;

#[derive(FromRow)]
pub struct Post {
    pub id: i32,
    pub title: String,
    pub body: String,
    pub published: bool,
}

pub async fn establish_connection() -> Result<SqlitePool, sqlx::Error> {
    dotenv().ok();

    let database_url = env::var("DATABASE_URL").expect("DATABASE_URL must be set");
    SqlitePool::connect(&database_url).await
}

pub async fn load_posts(pool: &SqlitePool) -> Result<Vec<Post>, sqlx::Error> {
    sqlx::query_as::<_, Post>("SELECT * FROM posts WHERE published = ?")
        .bind(true)
        .fetch_all(pool)
        .await
}

本题详细解读

Diesel 的使用

Diesel 是一个强大的 ORM（对象关系映射）工具，适用于 Rust 语言。它提供了类型安全的查询构建器，并且支持多种数据库后端（如 PostgreSQL、MySQL、SQLite 等）。

1. 建立连接：使用 establish_connection 函数连接到数据库。dotenv 用于从 .env 文件中加载环境变量，SqliteConnection::establish 用于建立与 SQLite 数据库的连接。

2. 定义模型：Post 结构体表示数据库中的 posts 表。#[derive(Queryable)] 宏用于自动生成从数据库查询结果的映射。

3. 查询数据：load_posts 函数使用 Diesel 的查询构建器来加载已发布的帖子。posts.filter(published.eq(true)) 构建了一个查询，load::<Post> 执行查询并将结果映射到 Post 结构体。

sqlx 的使用

sqlx 是一个异步的 SQL 查询工具，支持编译时检查 SQL 查询的正确性。它不依赖于 ORM，而是直接使用 SQL 语句进行操作。

1. 建立连接：使用 establish_connection 函数异步连接到数据库。SqlitePool::connect 用于建立与 SQLite 数据库的连接池。

2. 定义模型：Post 结构体表示数据库中的 posts 表。#[derive(FromRow)] 宏用于自动生成从数据库查询结果的映射。

3. 查询数据：load_posts 函数使用 sqlx::query_as 执行 SQL 查询，并将结果映射到 Post 结构体。fetch_all 方法用于获取所有查询结果。

总结

• Diesel 更适合需要复杂查询和类型安全的场景，适合使用 ORM 的开发模式。
• sqlx 更适合需要直接使用 SQL 语句的场景，并且支持异步操作，适合需要高性能的应用。

两者各有优劣，选择哪种工具取决于具体的项目需求。