使用 sqlx 操作 PostgreSQL 数组

PostgreSQL 原生支持数组。本章将讨论如何使用 sqlx 操作 PostgreSQL 的数组。

我们使用一个极简的“朋友圈”动态来做示例。

数据表定义

CREATE TABLE IF NOT EXISTS "posts"(
    "id" SERIAL PRIMARY KEY,
    "content" VARCHAR(255) NOT NULL,
    "images" VARCHAR(50)[] NOT NULL DEFAULT '{}'::VARCHAR(50)[]
);

• content：内容
• images：图片，数据类型为varchar(50)的数组。默认值为空数组

// src/post/model.rs

#[derive(Default, Serialize, Deserialize, FromRow)]
pub struct Post {
    pub id: i32,
    pub content: String,
    pub images: Vec<String>,
}

数据操作

pub async fn create(e: impl PgExecutor<'_>, m: &Post) -> sqlx::Result<i32> {
    let (id,): (i32,) =
        sqlx::query_as(r#"INSERT INTO posts ("content", images) VALUES ($1, $2) RETURNING id"#)
            .bind(&m.content)
            .bind(&m.images)
            .fetch_one(e)
            .await?;
    Ok(id)
}

pub async fn find(e: impl PgExecutor<'_>, id: i32) -> sqlx::Result<Option<Post>> {
    sqlx::query_as(r#"SELECT id, "content", images FROM posts WHERE id = $1"#)
        .bind(id)
        .fetch_optional(e)
        .await
}

• create()：用于新建一条记录。可以直接把 Vec 绑定到 PostgreSQL 的数组
• find()：根据ID查找记录

测试

本章代码位于 03.array分支。