SQLx在Rust語言中的基礎(chǔ)用法和進(jìn)階用法

SQLx是一個(gè)Rust語言的異步SQL執(zhí)行庫，它支持多種數(shù)據(jù)庫，包括MySQL、PostgreSQL、SQLite等。本教程將以MySQL數(shù)據(jù)庫為例，介紹SQLx在Rust語言中的基礎(chǔ)用法和進(jìn)階用法。

基礎(chǔ)用法

要使用SQLx，需要在复制Cargo.toml文件中添加以下依賴：

复制[dependencies]
sqlx = { version = "0.6", features = ["mysql", "runtime-tokio-rustls"] }
tokio = { version = "1", features = ["full"] }

連接數(shù)據(jù)庫

在使用SQLx之前，需要先連接數(shù)據(jù)庫。SQLx提供了兩種方式連接MySQL數(shù)據(jù)庫：使用URL連接和使用配置文件連接。

URL連接

使用URL連接時(shí)，需要在代碼中指定連接字符串，例如：

复制use sqlx::mysql::MySqlPoolOptions;

#[tokio::main]
async fn main() - > Result< (), sqlx::Error > {
    let pool = MySqlPoolOptions::new()
        .max_connections(5)
        .connect("mysql://username:password@hostname:port/database")
        .await?;
    // ...
    Ok(())
}

其中，复制username和复制password是數(shù)據(jù)庫用戶名和密碼，复制hostname是數(shù)據(jù)庫主機(jī)名，复制port是數(shù)據(jù)庫端口號(hào)，复制database是要連接的數(shù)據(jù)庫名。

配置文件連接

使用配置文件連接時(shí)，需要在項(xiàng)目根目錄下創(chuàng)建一個(gè)名為复制.env的文件，并在其中指定連接信息，例如：

复制DATABASE_URL=mysql://username:password@hostname:port/database

然后在代碼中使用复制dotenv庫加載复制.env文件，并使用复制sqlx::MySqlPool::connect_dotenv()方法連接數(shù)據(jù)庫，例如：

复制use sqlx::mysql::MySqlPoolOptions;

#[tokio::main]
async fn main() - > Result< (), sqlx::Error > {
    dotenv::dotenv().ok();
    let pool = MySqlPoolOptions::new()
        .max_connections(5)
        .connect_dotenv()
        .await?;
    // ...
    Ok(())
}

查詢數(shù)據(jù)

連接成功后，就可以使用SQLx執(zhí)行SQL查詢語句了。SQLx提供了兩種方式執(zhí)行SQL查詢語句：使用复制query()方法和使用复制query_as()方法。

使用query()方法

使用复制query()方法執(zhí)行SQL查詢語句時(shí)，需要手動(dòng)指定返回結(jié)果的類型，例如：

复制use sqlx::{MySqlPool, Row};

#[derive(Debug)]
struct User {
    id: i32,
    name: String,
}

#[tokio::main]
async fn main() - > Result< (), sqlx::Error > {
    let pool = MySqlPool::connect("mysql://username:password@hostname:port/database").await?;
    let mut conn = pool.acquire().await?;

    let mut rows = sqlx::query("SELECT id, name FROM users")
        .map(|row: sqlx::mysql::MySqlRow| {
            User {
                id: row.get(0),
                name: row.get(1),
            }
        })
        .fetch_all(&mut conn)
        .await?;

    for row in rows.iter() {
        println!("{:?}", row);
    }

    Ok(())
}

使用query_as()方法

使用复制query_as()方法執(zhí)行SQL查詢語句時(shí)，可以自動(dòng)將返回結(jié)果轉(zhuǎn)換為指定類型的結(jié)構(gòu)體，例如：

复制use sqlx::{MySqlPool, FromRow};

#[derive(Debug, FromRow)]
struct User {
    id: i32,
    name: String,
}

#[tokio::main]
async fn main() - > Result< (), sqlx::Error > {
    let pool = MySqlPool::connect("mysql://username:password@hostname:port/database").await?;
    let mut conn = pool.acquire().await?;

    let mut rows = sqlx::query_as::< _, User >("SELECT id, name FROM users")
        .fetch_all(&mut conn)
        .await?;

    for row in rows.iter() {
        println!("{:?}", row);
    }

    Ok(())
}

插入數(shù)據(jù)

使用SQLx插入數(shù)據(jù)時(shí)，可以使用复制execute()方法或复制execute_with()方法。

使用execute()方法

使用复制execute()方法插入數(shù)據(jù)時(shí)，需要手動(dòng)指定插入的數(shù)據(jù)，例如：

复制use sqlx::{MySqlPool, Row};

#[derive(Debug)]
struct User {
    name: String,
}

#[tokio::main]
async fn main() - > Result< (), sqlx::Error > {
    let pool = MySqlPool::connect("mysql://username:password@hostname:port/database").await?;
    let mut conn = pool.acquire().await?;

    let user = User {
        name: "John".to_string(),
    };

    let result = sqlx::query("INSERT INTO users (name) VALUES (?)")
        .bind(user.name)
        .execute(&mut conn)
        .await?;

    println!("{:?}", result);

    Ok(())
}

使用execute_with()方法

使用复制execute_with()方法插入數(shù)據(jù)時(shí)，可以使用結(jié)構(gòu)體自動(dòng)映射的特性，例如：

复制use sqlx::{MySqlPool, FromRow};

#[derive(Debug, FromRow)]
struct User {
    name: String,
}

#[tokio::main]
async fn main() - > Result< (), sqlx::Error > {
    let pool = MySqlPool::connect("mysql://username:password@hostname:port/database").await?;
    let mut conn = pool.acquire().await?;

    let user = User {
        name: "John".to_string(),
    };

    let result = sqlx::query_with::< _, User >("INSERT INTO users (name) VALUES (?)", user)
        .execute(&mut conn)
        .await?;

    println!("{:?}", result);

    Ok(())
}

更新數(shù)據(jù)

使用SQLx更新數(shù)據(jù)時(shí)，可以使用复制execute()方法或复制execute_with()方法。

使用execute()方法

使用复制execute()方法更新數(shù)據(jù)時(shí)，需要手動(dòng)指定更新的條件和更新的數(shù)據(jù)，例如：

复制use sqlx::{MySqlPool, Row};

#[derive(Debug)]
struct User {
    id: i32,
    name: String,
}

#[tokio::main]
async fn main() - > Result< (), sqlx::Error > {
    let pool = MySqlPool::connect("mysql://username:password@hostname:port/database").await?;
    let mut conn = pool.acquire().await?;

    let user = User {
        id: 1,
        name: "John".to_string(),
    };

    let result = sqlx::query("UPDATE users SET name = ? WHERE id = ?")
        .bind(user.name)
        .bind(user.id)
        .execute(&mut conn)
        .await?;

    println!("{:?}", result);

    Ok(())
}

使用execute_with()方法

使用复制execute_with()方法更新數(shù)據(jù)時(shí)，可以使用結(jié)構(gòu)體自動(dòng)映射的特性，例如：

复制use sqlx::{MySqlPool, FromRow};

#[derive(Debug, FromRow)]
struct User {
    id: i32,
    name: String,
}

#[tokio::main]
async fn main() - > Result< (), sqlx::Error > {
    let pool = MySqlPool::connect("mysql://username:password@hostname:port/database").await?;
    let mut conn = pool.acquire().await?;

    let user = User {
        id: 1,
        name: "John".to_string(),
    };

    let result = sqlx::query_with::< _, User >("UPDATE users SET name = :name WHERE id = :id", user)
        .execute(&mut conn)
        .await?;

    println!("{:?}", result);

    Ok(())
}

刪除數(shù)據(jù)

使用SQLx刪除數(shù)據(jù)時(shí)，可以使用复制execute()方法或复制execute_with()方法。

使用execute()方法

使用复制execute()方法刪除數(shù)據(jù)時(shí)，需要手動(dòng)指定刪除的條件，例如：

复制use sqlx::{MySqlPool, Row};

#[tokio::main]
async fn main() - > Result< (), sqlx::Error > {
    let pool = MySqlPool::connect("mysql://username:password@hostname:port/database").await?;
    let mut conn = pool.acquire().await?;

    let result = sqlx::query("DELETE FROM users WHERE id = ?")
        .bind(1)
        .execute(&mut conn)
        .await?;

    println!("{:?}", result);

    Ok(())
}

使用execute_with()方法

使用复制execute_with()方法刪除數(shù)據(jù)時(shí)，可以使用結(jié)構(gòu)體自動(dòng)映射的特性，例如：

复制use sqlx::{MySqlPool, FromRow};

#[derive(Debug, FromRow)]
struct User {
    id: i32,
}

#[tokio::main]
async fn main() - > Result< (), sqlx::Error > {
    let pool = MySqlPool::connect("mysql://username:password@hostname:port/database").await?;
    let mut conn = pool.acquire().await?;

    let user = User {
        id: 1,
    };

    let result = sqlx::query_with::< _, User >("DELETE FROM users WHERE id = :id", user)
        .execute(&mut conn)
        .await?;

    println!("{:?}", result);

    Ok(())
}

進(jìn)階用法

事務(wù)

使用SQLx執(zhí)行事務(wù)時(shí)，可以使用复制begin()方法開始事務(wù)，使用复制commit()方法提交事務(wù)，使用复制rollback()方法回滾事務(wù)。

复制use sqlx::{MySqlPool, Transaction};

#[derive(Debug)]
struct User {
    name: String,
}

#[tokio::main]
async fn main() - > Result< (), sqlx::Error > {
    let pool = MySqlPool::connect("mysql://username:password@hostname:port/database").await?;
    let mut conn = pool.acquire().await?;

    let mut tx = conn.begin().await?;

    let user = User {
        name: "John".to_string(),
    };

    let result = sqlx::query("INSERT INTO users (name) VALUES (?)")
        .bind(user.name)
        .execute(&mut tx)
        .await?;

    println!("{:?}", result);

    tx.commit().await?;

    Ok(())
}

連接池

使用SQLx連接池時(shí)，可以使用复制PoolOptions::new()方法創(chuàng)建連接池，并使用复制acquire()方法獲取連接。

复制use sqlx::{MySqlPool, PoolOptions};

#[tokio::main]
async fn main() - > Result< (), sqlx::Error > {
    let pool = MySqlPool::connect("mysql://username:password@hostname:port/database").await?;
    let mut conn = pool.acquire().await?;

    let pool = MySqlPool::builder()
        .max_size(5)
        .build("mysql://username:password@hostname:port/database")
        .await?;

    let mut conn = pool.acquire().await?;

    // ...

    Ok(())
}

總結(jié)

本教程介紹了SQLx在Rust語言中的基礎(chǔ)用法和進(jìn)階用法，包括連接數(shù)據(jù)庫、查詢數(shù)據(jù)、插入數(shù)據(jù)、更新數(shù)據(jù)、刪除數(shù)據(jù)、事務(wù)和連接池等。SQLx是一個(gè)簡(jiǎn)單易用的異步SQL執(zhí)行庫，可以幫助Rust開發(fā)者快速地與多種數(shù)據(jù)庫進(jìn)行交互。