SQLx是一個Rust語言的異步SQL數據庫訪問庫，支持多種數據庫，包括PostgreSQL、MySQL、SQLite等。本教程將以SQLite為例，介紹SQLx的基礎用法和進階用法。

基礎用法

連接數據庫

首先，需要在Rust項目中添加SQLx庫的依賴：

[dependencies]
sqlx = "0.6"
sqlx-core = "0.6"
sqlx-derive = "0.6"
sqlx-macros = "0.6"

然后，可以使用以下代碼連接SQLite數據庫：

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的表：

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表中插入一條數據：

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("[email protected]")
        .execute(&pool)
        .await?;
    Ok(())
}

這里使用了bind方法綁定參數，可以避免SQL注入攻擊。

查詢數據

下面的代碼演示了如何使用SQLx查詢users表中所有數據：

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表中的數據：

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("[email protected]")
        .bind("Alice")
        .execute(&pool)
        .await?;
    Ok(())
}

這里使用了UPDATE語句更新數據。

刪除數據

下面的代碼演示了如何使用SQLx刪除users表中的數據：

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進行事務操作：

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("[email protected]")
        .execute(&mut tx)
        .await?;
    query("INSERT INTO users (name, email) VALUES (?, ?)")
        .bind("Bob")
        .bind("[email protected]")
        .execute(&mut tx)
        .await?;
    tx.commit().await?;
    Ok(())
}

這里使用了begin方法開啟一個事務，可以在事務中執行多條SQL語句。如果所有語句執行成功，可以使用commit方法提交事務，否則可以使用rollback方法回滾事務。

批量插入

下面的代碼演示了如何使用SQLx進行批量插入操作：

use sqlx::{query, SqlitePool};

#[tokio::main]
async fn main() - > Result< (), sqlx::Error > {
    let pool = SqlitePool::connect("sqlite:mydatabase.db").await?;
    let users = vec![
        ("Alice", "[email protected]"),
        ("Bob", "[email protected]"),
        ("Charlie", "[email protected]"),
    ];
    query("INSERT INTO users (name, email) VALUES (?, ?)")
        .bind_all(users)
        .execute(&pool)
        .await?;
    Ok(())
}

這里使用了bind_all方法綁定多個參數，可以將多個參數一次性綁定到SQL語句中，避免了多次執行SQL語句的開銷。

總結

本教程介紹了SQLx的基礎用法和進階用法，包括連接數據庫、創建表、插入數據、查詢數據、更新數據、刪除數據、事務和批量插入。SQLx是一個非常方便的異步SQL數據庫訪問庫，可以大大提高開發效率。