Rust操作MySQL

techdashen2023-07-25 14:32

查询

本部分是对 「Rust入门系列」Rust 中使用 MySQL[1]的学习与记录

  • 经常使用的时间处理库： chrono

  • 流式查询使用： query_iter

  • 输出到Vec使用： query

  • 映射到结构体使用： query_map

  • 获取单条数据使用： query_first

  • 命名参数查询使用： exec_first

    CREATE TABLE student (
    id int(11) NOT NULL AUTO_INCREMENT,
    name varchar(128) NOT NULL,
    age int(11) NOT NULL,
    id_card varchar(128) NOT NULL,
    last_update date NOT NULL,
    PRIMARY KEY (id)
    ) ENGINE=InnoDB DEFAULT CHARSET=utf8mb4;

    -- 插入测试数据
    insert into student (name, age, id_card, last_update) values ('张三', 23, '123456789X', CURRENT_DATE());
    insert into student (name, age, id_card, last_update) values ('李四', 24, '8382353902', CURRENT_DATE())

注意，mysql[2]这个crate新版本demo有问题，文档的更新速度跟不上代码的修改脚步..

需要指定版本：

复制代码 
[dependencies]
mysql = "20.0.0" #通配符*表示可以使用任何版本，通常会拉取最新版本；此处需要指定，不使用最新版本

流式查询

复制代码 
use chrono::prelude::*;// 用来处理日期
use mysql::*;
use mysql::prelude::*;

fn main() {
    let url = "mysql://root:12345678@localhost:3306/shuang";
    let pool = Pool::new(url).unwrap(); // 获取连接池


    let mut conn = pool.get_conn().unwrap();// 获取链接

    conn.query_iter("select * from student")
        .unwrap()
        .for_each(|row| {
            let r: (i32, String, i32, String, NaiveDate) = from_row(row.unwrap());
            println!("{}, {},{},{}, {:?}", r.0, r.1, r.2, r.3, r.4);
        });
}

row 的类型是mysql_common::row::Row，其把数据以字节的形式存储。

所以需将低级的字节转换成想要的类型 如i32,String，这里使用了from_row。注意，转换后的数据以元组的形式返回，其中每一项和选择列的顺序相同。

输出：

复制代码 
1, 张三,23,123456789X, 2022-04-26
2, 李四,24,8382353902, 2022-04-26

流式查询

其实还可以将查询结果收集到Vec中。 Vec中的每个元素都是一个元组。

query函数已经将字节转换为选择的数据类型，因此不需要再转换了。 需要注意的是，这里必须明确元组的数据类型(如此处是 Vec<(i32, String, i32, String, NaiveDate)>)。 否则，编译器没办法做转换。

复制代码 
use chrono::prelude::*;// 用来处理日期
use mysql::*;
use mysql::prelude::*;

fn main() {
    let url = "mysql://root:12345678@localhost:3306/shuang";
    let pool = Pool::new(url).unwrap(); // 获取连接池


    let mut conn = pool.get_conn().unwrap();// 获取链接

// 输出到Vec
    let res: Vec<(i32, String, i32, String, NaiveDate)> =
        conn.query("select * from student").unwrap();
    for r in res {
        println!("{}, {},{},{}, {:?}", r.0, r.1, r.2, r.3, r.4);
    }
}

映射结果到结构体

如果表的列数很多，使用元组容易混淆，更普遍的做法是定义一个结构体。

如下定义一个Student 结构体, 然后可以用query_map将查询结果映射到Student中。

不需要指定数据类型，编译器会根据Student类型自动推导

复制代码 
use chrono::prelude::*;// 用来处理日期
use mysql::*;
use mysql::prelude::*;

fn main() {
    let url = "mysql://root:12345678@localhost:3306/shuang";
    let pool = Pool::new(url).unwrap(); // 获取连接池


    let mut conn = pool.get_conn().unwrap();// 获取链接

    // 将结果映射到提前定义好的结构体
    struct Student {
        id: u64,
        name: String,
        age: u16,
        id_card: String,
        last_changed_on: NaiveDate,
    }

    let res = conn.query_map(
        "select * from student",
        |(id, name, age, id_card, update)| Student {
            id: id,
            name: name,
            age: age,
            id_card: id_card,
            last_changed_on: update,
        },
    ).expect("Query failed.");

    for i in res {
        println!(
            "{}, {},{},{}, {:?}",
            i.id, i.name, i.age, i.id_card, i.last_changed_on
        )
    }
}

单条数据查询

查询特定数据行，可能会出现下面几种情况:

  • 找到，返回实际数据
  • 没有找到行
  • 发生错误

所以，使用query_first函数返回的是Option的结果。 需要将其解包两次才可以获取实际的行数据:

复制代码 
use chrono::prelude::*;// 用来处理日期
use mysql::*;
use mysql::prelude::*;

fn main() {
    let url = "mysql://root:12345678@localhost:3306/shuang";
    let pool = Pool::new(url).unwrap(); // 获取连接池


    let mut conn = pool.get_conn().unwrap();// 获取链接


    struct Student {
        id: u64,
        name: String,
        age: u16,
        id_card: String,
        last_changed_on: NaiveDate,
    }

    // 条件查询，查询单个数据
    let res = conn.query_first("select * from student where name = '张三'")
        .map(
            // Unpack Result
            |row| {
                row.map(|(id, name, age, id_card, update)| Student {
                    id: id,
                    name: name,
                    age: age,
                    id_card: id_card,
                    last_changed_on: update,
                })
            },
        );

    match res.unwrap() {
        Some(student) => println!(
            "{}, {},{},{}, {:?}",
            student.id, student.name, student.age, student.id_card, student.last_changed_on
        ),
        None => println!("Sorry no student found."),
    }
}

命名参数的使用

复制代码 
use chrono::prelude::*;// 用来处理日期
use mysql::*;
use mysql::prelude::*;

fn main() {
    let url = "mysql://root:12345678@localhost:3306/shuang";
    let pool = Pool::new(url).unwrap(); // 获取连接池


    let mut conn = pool.get_conn().unwrap();// 获取链接


    struct Student {
        id: u64,
        name: String,
        age: u16,
        id_card: String,
        last_changed_on: NaiveDate,
    }

    let res = conn
        .exec_first(
            "select * from student where name = :name",
            params! {
                "name" => "李四"
            },
        )
        .map(
            // Unpack Result
            |row| {
                row.map(|(id, name, age, id_card, update)| Student {
                    id: id,
                    name: name,
                    age: age,
                    id_card: id_card,
                    last_changed_on: update,
                })
            },
        );

    match res.unwrap() {
        Some(student) => println!(
            "{}, {},{},{}, {:?}",
            student.id, student.name, student.age, student.id_card, student.last_changed_on
        ),
        None => println!("Sorry no student found."),
    }

}

写操作

本部分是对 Rust使用MySQL数据库02[3]的学习与记录

  • 插入数据使用 conn.exec_drop()

  • 使用预编译语句插入大量数据， conn.prep()

  • 使用 conn.last_insert_id()可以获取主键

  • 更新和删除也使用 conn.prep和 conn.exec_drop

插入新数据

复制代码 
use chrono::prelude::*;
// 用来处理日期
use mysql::*;
use mysql::prelude::*;

fn main() {
    let url = "mysql://root:12345678@localhost:3306/shuang";
    let pool = Pool::new(url).unwrap(); // 获取连接池


    let mut conn = pool.get_conn().unwrap();// 获取链接


    conn.exec_drop(
        "INSERT INTO student (name, age, id_card, last_update) VALUES (:name, :age, :id_card, :last_update)",
        params! {
        "name" => "王五",
        "age" => 28,
        "id_card" => "66666688",
        "last_update" => today(),
}).unwrap();
}

fn today() -> NaiveDate {
    let l = Local::today();

    NaiveDate::from_ymd(l.year(), l.month(), l.day())
}

和上面一节一样，命名参数在这里使用了params宏的语法

exec_drop方法中的drop表示没有返回结果

用于执行插入/更新/删除的sql

使用预编译语句

使用conn.prep将sql编译成预编译语句。

复制代码 
use chrono::prelude::*;
// 用来处理日期
use mysql::*;
use mysql::prelude::*;

fn main() {
    let url = "mysql://root:12345678@localhost:3306/shuang";
    let pool = Pool::new(url).unwrap(); // 获取连接池


    let mut conn = pool.get_conn().unwrap();// 获取链接


    let stmt = conn.prep("INSERT INTO student (name, age, id_card, last_update) VALUES (:name, :age, :id_card, :last_update)")
        .unwrap();

    for i in 1..10 {
        conn.exec_drop(&stmt, params! {
         "name" => "dashen",
         "age" => 18 + i,
         "id_card" => "1234565X",
         "last_update" => NaiveDate::from_ymd(2017, 05, 04),
     }).unwrap()
    }
}

获取生成的主键id

可以通过conn.last_insert_id()方法获取到新记录的主键id，该方法将返回的一个类型为u64 的值

复制代码 
use chrono::prelude::*;
// 用来处理日期
use mysql::*;
use mysql::prelude::*;

fn main() {
    let url = "mysql://root:12345678@localhost:3306/shuang";
    let pool = Pool::new(url).unwrap(); // 获取连接池


    let mut conn = pool.get_conn().unwrap();// 获取链接

    conn.exec_drop("INSERT INTO student (name, age, id_card, last_update) VALUES (:name, :age, :id_card, :last_update)", params! {
    "name" => "fliter",
    "age" => 29,
    "id_card" => "88888888",
    "last_update" => NaiveDate::from_ymd(2022, 05, 04),
}).unwrap();

    println!("新插入的记录的主键为: {}", conn.last_insert_id())
}

新插入的记录的主键为: 13

更新和删除

类似于插入操作

复制代码 
use chrono::prelude::*;
// 用来处理日期
use mysql::*;
use mysql::prelude::*;

fn main() {
    let url = "mysql://root:12345678@localhost:3306/shuang";
    let pool = Pool::new(url).unwrap(); // 获取连接池


    let mut conn = pool.get_conn().unwrap();// 获取链接


    let stmt = conn.prep("update student set name=:name, last_update=:last_update where id=:id")
        .unwrap();

    conn.exec_drop(&stmt, params! {
     "name" => "新名字",
     "last_update" => NaiveDate::from_ymd(2038, 12, 31),
     "id" => 10,
}).unwrap();


    let stmt = conn.prep("delete from student where id=:id").unwrap();

    conn.exec_drop(&stmt, params! {
    "id" => 12,
}).unwrap();
}

参考资料

1

「Rust入门系列」Rust 中使用 MySQL: https://rustmagazine.github.io/rust_magazine_2021/chapter_3/rust-mysql.html

2

mysql: https://crates.io/crates/mysql

3

Rust使用MySQL数据库02: https://www.modb.pro/db/179746

本文由mdnice多平台发布

相关推荐
我命由我123455 分钟前
Java 泛型 - Java 泛型通配符（上界通配符、下界通配符、无界通配符、PECS 原则）
java·开发语言·后端·java-ee·intellij-idea·idea·intellij idea
szhf785 分钟前
SpringBoot Test详解
spring boot·后端·log4j
无尽的沉默6 分钟前
SpringBoot整合Redis
spring boot·redis·后端
摸鱼的春哥13 分钟前
春哥的Agent通关秘籍07：5分钟实现文件归类助手【实战】
前端·javascript·后端
Victor35629 分钟前
MongoDB（2）MongoDB与传统关系型数据库的主要区别是什么？
后端
JaguarJack30 分钟前
PHP 应用遭遇 DDoS 攻击时会发生什么 从入门到进阶的防护指南
后端·php·服务端
BingoGo31 分钟前
PHP 应用遭遇 DDoS 攻击时会发生什么 从入门到进阶的防护指南
后端
Victor35632 分钟前
MongoDB（3）什么是文档(Document)？
后端
牛奔3 小时前
Go 如何避免频繁抢占？
开发语言·后端·golang
想用offer打牌7 小时前
MCP (Model Context Protocol) 技术理解 - 第二篇
后端·aigc·mcp
热门推荐
01GitHub 镜像站点02Claude Code + GLM4.7 避坑指南：解决 Unable to connect to Anthropic services03openclaw配置教程（linux+局域网ollama）04UV安装并设置国内源05OpenClaw Chrome扩展使用教程 - 浏览器中继控制06Linux下V2Ray安装配置指南07Claude Code Skills 实用使用手册08Vue-skills的中文文档09让 Trae IDE 智能体 “读懂”文档 Excel+PDF+DOCX ：mcp-documents-reader 工具使用指南10OpenCode 入门教程：介绍 · 安装 · 配置第三方 API (如 Claude)