使用Rusqlite操作sqlite数据库

最新推荐文章于 2025-03-02 21:51:19 发布

勾吴江南

最新推荐文章于 2025-03-02 21:51:19 发布

阅读量3.9k

点赞数 1

分类专栏： rust 文章标签： rust

本文链接：https://blog.youkuaiyun.com/TX_OfficeDev/article/details/105497384

版权

rust 专栏收录该内容

8 篇文章

订阅专栏

本文介绍了一个使用Rust编程语言和rusqlite库实现的简单SQLite数据库操作示例。包括创建数据库表、检查表是否存在、插入数据、查询数据等功能。通过这个示例，读者可以了解到如何在Rust中进行基本的数据库操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

[dependencies]
rusqlite = "0.22.0"

use rusqlite::{params, Connection, Result};

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

fn main() {
    let con = Connection::open_in_memory().unwrap();

    let flag1 = check_table_existed("table_1", &con);

    if !flag1 {
        create_table(&con);
    }

    if !check_person_existed("Tester1", &con) {
        insert_person("Tester1", &con);
    }

    if !check_person_existed("Tester2", &con) {
        insert_person("Tester2", &con);
    }

    list_persons(&con);

    match con.close(){
        Ok(v) => v,
        Err(e) => panic!("{:?}", e),
    };
}

fn check_table_existed(table_name : &str, con : &Connection) -> bool {
    let sql: &str = "SELECT COUNT(`name`) FROM `sqlite_master` WHERE `type` = 'table' AND `name` = ?";
    let mut stmt = con.prepare(sql).unwrap();
    let rs = stmt.query_row(params![table_name], |row | {
        return row.get(0) as Result<i32>;
    });

    let count = rs.unwrap();

    return count > 0;
}

fn create_table(con : &Connection) {
    let sql : &str = "CREATE TABLE IF NOT EXISTS `table_1`(`Id` INTEGER PRIMARY KEY AUTOINCREMENT NOT NULL, `Name` TEXT NOT NULL)";
    match con.execute(sql, params![]) {
        Ok(o) => o,
        Err(e) => panic!("{:?}", e),
    };
}

fn check_person_existed(person: &str, con: &Connection) -> bool {
    let sql: &str = "SELECT COUNT(`Id`) FROM `table_1` WHERE `Name` = ?";
    let mut stmt = con.prepare(sql).unwrap();
    let rs = stmt.query_row(params![person], |row|{
        row.get(0) as Result<i32>
    });
    let count = rs.unwrap();
    return count > 0;
}

fn insert_person(person: &str, con: &Connection) {
    let sql: &str = "INSERT INTO `table_1`(`Name`) VALUES (?)";
    match con.execute(sql, params![person]) {
        Ok(o) => o,
        Err(e) => panic!("{:?}", e),
    };
}

fn list_persons(con : &Connection){
    let sql:&str = "SELECT * FROM `table_1`";
    let mut stmt = con.prepare(sql).unwrap();
    let person_iterator = stmt.query_map(params![], |row| {
        Ok(Person{
            id: row.get(0).unwrap(),
            name: row.get(1).unwrap(),
        })
    }).unwrap();

    for p in person_iterator {
        println!("{:?}", p.unwrap());
    }
}