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多平台发布

相关推荐
G探险者1 分钟前
为什么 Zookeeper 越扩越慢，而 Nacos 却越扩越快？
分布式·后端
不太厉害的程序员16 分钟前
NC65配置xml找不到Bean
xml·java·后端·eclipse
不被定义的程序猿17 分钟前
Golang 在 Linux 平台上的并发控制
开发语言·后端·golang
AntBlack37 分钟前
Python : AI 太牛了 ，撸了两个 Markdown 阅读器 ，谈谈使用感受
前端·人工智能·后端
mikes zhang1 小时前
Flask文件上传与异常处理完全指南
后端·python·flask
Pitayafruit2 小时前
跟着大厂学架构01：如何利用开源方案，复刻B站那套“永不崩溃”的评论系统？
spring boot·分布式·后端
方圆想当图灵2 小时前
深入理解软件设计：领域驱动设计 DDD
后端·架构
excel2 小时前
MySQL 9 在 Windows 上使用 mysqld --initialize-insecure 无响应的排查与解决方案
后端
你怎么知道我是队长2 小时前
GO语言---defer关键字
开发语言·后端·golang
方圆想当图灵2 小时前
深入理解软件设计：什么是好的架构？
后端·架构·代码规范
热门推荐
01Coze扣子平台完整体验和实践（附国内和国际版对比）02【图像处理与机器视觉】XJTU期末考点03KGG转MP3工具|非KGM文件|解密音频04扣子（coze）实战|我用扣子搭建了一个自动分析小红薯笔记内容的AI应用|详细步骤拆解05海康Visionmaster-常见问题排查方法-启动阶段06YOLOv8入门 | 重要性能衡量指标、训练结果评价及分析及影响mAP的因素【发论文关注的指标】07DeepSeek各版本说明与优缺点分析08从零安装 LLaMA-Factory 微调 Qwen 大模型成功及所有的坑09【SpeedAI科研小助手】2分钟极速解决知网维普重复率、AIGC率过高，一键全文降！文件格式不变，公式都保留的！10零代码入门 | Coze——让大模型接入自己的数据库