Rust 使用SQLite

在本章中，我们将探讨如何在 Rust 中使用 SQLite 数据库。SQLite 是一个轻量级的数据库引擎，非常适合用于小型项目或需要嵌入式数据库的应用程序。我们将学习如何设置 SQLite 数据库，并在 Rust 应用程序中执行基本的 CRUD 操作。

安装依赖

首先，我们需要添加对 SQLite 的支持。在 Cargo.toml 文件中，添加以下依赖项：

[dependencies]
rusqlite = "0.27"

rusqlite 是一个流行的 Rust 库，它提供了一个方便的接口来与 SQLite 数据库进行交互。

创建数据库连接

在 Rust 中使用 SQLite 数据库的第一步是创建一个数据库连接。我们可以通过以下方式实现：

// www.javascriptcn.com code example
use rusqlite::{Connection, Result};

fn main() -> Result<()> {
    // 创建一个新的数据库连接
    let conn = Connection::open("test.db")?;

    println!("成功打开数据库!");
    Ok(())
}

上述代码尝试打开一个名为 test.db 的 SQLite 数据库文件。如果该文件不存在，则会创建一个新的数据库文件。

创建表

为了存储数据，我们需要创建一个表。下面是一个简单的示例，演示如何创建一个用户表：

// www.javascriptcn.com code example
use rusqlite::{Connection, Result, ToSql};

fn create_table(conn: &Connection) -> Result<()> {
    // 执行 SQL 语句来创建一个用户表
    conn.execute(
        "CREATE TABLE IF NOT EXISTS users (
                  id              INTEGER PRIMARY KEY,
                  name            TEXT NOT NULL,
                  email           TEXT NOT NULL
                )",
        [],
    )?;
    println!("用户表已创建!");
    Ok(())
}

fn main() -> Result<()> {
    let conn = Connection::open("test.db")?;

    create_table(&conn)?;

    Ok(())
}

插入数据

插入数据到表中非常简单。我们可以使用 insert_row 方法来完成这个操作：

// www.javascriptcn.com code example
use rusqlite::{Connection, Result, params};

fn insert_user(conn: &Connection, name: &str, email: &str) -> Result<()> {
    // 插入一条新记录
    conn.execute(
        "INSERT INTO users (name, email) VALUES (?1, ?2)",
        params![name, email],
    )?;
    println!("用户已插入!");
    Ok(())
}

fn main() -> Result<()> {
    let conn = Connection::open("test.db")?;

    create_table(&conn)?;
    insert_user(&conn, "Alice", "alice@example.com")?;

    Ok(())
}

在这个例子中，我们定义了一个函数 insert_user 来向 users 表中插入新的用户记录。

查询数据

查询数据同样简单，我们可以使用 query_row 或 prepare 方法来执行查询：

// www.javascriptcn.com code example
use rusqlite::{Connection, Result, Row};
use std::fmt;

fn print_user(row: &Row) -> fmt::Result {
    let id: i32 = row.get(0)?;
    let name: String = row.get(1)?;
    let email: String = row.get(2)?;

    println!("ID: {}, 名字: {}, 邮箱: {}", id, name, email);
    Ok(())
}

fn query_users(conn: &Connection) -> Result<()> {
    // 准备并执行查询
    let mut stmt = conn.prepare("SELECT id, name, email FROM users")?;
    let user_iter = stmt.query_map([], |row| Ok((row,)))?;

    for row in user_iter {
        if let Ok((row,)) = row {
            print_user(&row)?;
        }
    }

    Ok(())
}

fn main() -> Result<()> {
    let conn = Connection::open("test.db")?;

    create_table(&conn)?;
    insert_user(&conn, "Bob", "bob@example.com")?;
    query_users(&conn)?;

    Ok(())
}

在这个例子中，我们定义了一个 print_user 函数来格式化输出每一行的数据。然后我们通过 query_users 函数执行一个查询，并遍历结果集打印出所有用户的信息。

更新和删除数据

更新和删除数据的操作也很直观。这里给出一个简单的示例：

// www.javascriptcn.com code example
fn update_user_email(conn: &Connection, id: i32, new_email: &str) -> Result<()> {
    // 更新用户的邮箱地址
    conn.execute(
        "UPDATE users SET email = ?1 WHERE id = ?2",
        params![new_email, id],
    )?;
    println!("用户邮箱已更新!");
    Ok(())
}

fn delete_user(conn: &Connection, id: i32) -> Result<()> {
    // 删除指定 ID 的用户
    conn.execute("DELETE FROM users WHERE id = ?1", [id])?;
    println!("用户已删除!");
    Ok(())
}

fn main() -> Result<()> {
    let conn = Connection::open("test.db")?;

    create_table(&conn)?;
    insert_user(&conn, "Charlie", "charlie@example.com")?;
    update_user_email(&conn, 1, "updated@example.com")?;
    delete_user(&conn, 2)?;

    Ok(())
}

在这个例子中，我们定义了两个函数：update_user_email 和 delete_user 分别用于更新和删除用户记录。

以上就是 Rust 中使用 SQLite 数据库的基本方法。通过这些基础操作，你可以构建更复杂的功能，例如处理更复杂的数据模型、事务管理等。希望这个章节对你有所帮助！