sqlx概览

开始之前,我们需要一些准备工作,包括:创建用于演示的数据库及数据、创建一个 Rust 项目以及为项目编写一些基础性代码。

示例数据库及数据

本专题我们使用 MySQL 8 数据库,并只使用一张会员(memeber)表:

你可以在filess.io里创建一个免费的 MySQL 8 实例

表结构如下:

初始数据:

INSERT INTO member (name, balance, types) VALUES
    ('axum.rs', 12345, 3),
    ('张三', 33333, 0),
    ('李四', 44444, 2),
    ('王五', 55555, 1),
    ('赵六', 66666, 3);

创建项目及依赖

使用 cargo 创建名为 axum-sqlx 的项目:

cargo new axum-sqlx

并加上依赖:

[dependencies]
tokio = {version="1",features=["full"]}
axum = "0.6"
serde = {version="1", features=["derive"]}
chrono = {version="0.4", features=["serde"]}
sqlx = { version = "0.6", features = [ "runtime-tokio-native-tls", "mysql","chrono", "macros" ] }
dotenv = "0.15"
config = "0.13"
tracing = "0.1"
tracing-subscriber = "0.3"

基础性代码

基础性代码无非就是配置文件 、错误处理、状态共享、日志等,在和其它专题的相差不大,这里不再重复。我们重点看一下数据模型的定义:

// src/model/member.rs

use serde::{Deserialize, Serialize};

#[derive(Debug, Default, Deserialize, Serialize, sqlx::Type, Clone)]
#[repr(u8)]
pub enum MemberTypes {
    #[default]
    /// 普通会员
    Normal,
    /// 白银会员
    Silver,
    /// 黄金会员
    Gold,
    /// 钻石会员
    Diamond,
}

#[derive(Debug, Default, Deserialize, Serialize, sqlx::FromRow)]
pub struct Member {
    pub id: u32,
    pub name: String,
    pub dateline: chrono::DateTime<chrono::Local>,
    pub balance: u32,
    pub types: MemberTypes,
    pub is_del: bool,
}

枚举 MemberTypes

#[derive(Debug, Default, Deserialize, Serialize, sqlx::Type, Clone)]
#[repr(u8)]
pub enum MemberTypes {
    #[default]
    /// 普通会员
    Normal,
    /// 白银会员
    Silver,
    /// 黄金会员
    Gold,
    /// 钻石会员
    Diamond,
}

这是一个非常简单的枚举类型,但是:

  • #[derive(sqlx::Type)]:用于将这个枚举声明为 sqlx 的类型
  • #[repr(u8)]:将枚举值的类型替换为 u8

想一想:PostgreSQL 并没有 UNSIGNED,所以它只能映射为 i8,因此例中的 #[repr(u8)] 应该改为 #[repr(i8)]

结构体 Member

#[derive(Debug, Default, Deserialize, Serialize, sqlx::FromRow)]
pub struct Member {
    pub id: u32,
    pub name: String,
    pub dateline: chrono::DateTime<chrono::Local>,
    pub balance: u32,
    pub types: MemberTypes,
    pub is_del: bool,
}
  • #[derive(sqlx::FromRow)]:使 sqlx 具有自动将查询结果映射到结构体的功能(比如 query_as()
  • pub dateline: chrono::DateTime<chrono::Local>:将数据库中的 DATETIME 类型映射到 rust 中的 chrono::DateTime<chrono::Local>类型
  • pub types: MemberTypes:将数据库中的 TINYINT UNSIGNED 映射到 rust 中的枚举

想一想:和 MySQL 的 DATETIME 类似,PostgreSQL 中的 TIMESTAMPTZ 也可以映射到 chrono::DateTime<T>

本章代码位于01/准备工作分支。

要查看完整内容,请先登录