如何在Diesel中实现适配多结构体的通用更新查询？

核心思路 #

要实现适配多个结构体的通用更新，关键是利用Diesel提供的泛型Trait（如Identifiable、AsChangeset、IntoUpdateTarget）抽象模型和表的共性，避免重复编写单模型的更新代码。

步骤1：确保模型与Schema符合约束 #

首先，你的Book和BookComment模型需要实现Identifiable（保证存在主键），同时对应的Schema表要正确关联。以下是基础定义示例：

// schema.rs（Diesel自动生成或手动定义）
table! {
    books (id) {
        id -> Int4,
        title -> Varchar,
        author -> Varchar,
    }
}

table! {
    book_comments (id) {
        id -> Int4,
        book_id -> Int4,
        content -> Text,
    }
}

// models.rs
use diesel::Identifiable;
use schema::{books, book_comments};

#[derive(Queryable, Identifiable, AsChangeset)]
#[diesel(table_name = books)]
pub struct Book {
    pub id: i32,
    pub title: String,
    pub author: String,
}

#[derive(Queryable, Identifiable, AsChangeset)]
#[diesel(table_name = book_comments)]
pub struct BookComment {
    pub id: i32,
    pub book_id: i32,
    pub content: String,
}

// 用于接收部分更新数据的结构体（可选）
#[derive(AsChangeset)]
#[diesel(table_name = books)]
pub struct UpdateBook {
    pub title: Option<String>,
    pub author: Option<String>,
}

#[derive(AsChangeset)]
#[diesel(table_name = book_comments)]
pub struct UpdateBookComment {
    pub content: Option<String>,
}

步骤2：实现通用更新函数 #

编写泛型函数，通过约束泛型参数适配不同模型：

use diesel::{PgConnection, RunQueryDsl, ExpressionMethods, QueryDsl};
use diesel::associations::Identifiable;
use diesel::changeset::AsChangeset;
use diesel::query_builder::IntoUpdateTarget;

/// 通用更新函数
/// - `Model`: 实现Identifiable的模型类型
/// - `Changes`: 实现AsChangeset的更新数据类型，需与Model对应同一张表
fn generic_update<Model, Changes>(
    conn: &mut PgConnection,
    model_id: <Model as Identifiable>::Id,
    changes: Changes,
) -> Result<Model, diesel::result::Error>
where
    Model: Identifiable + Queryable<<Model as Identifiable>::Id::SqlType, diesel::pg::Pg>,
    Changes: AsChangeset<Target = <Model as Identifiable>::Table> + IntoUpdateTarget,
    <Model as Identifiable>::Table: QueryDsl + ExpressionMethods,
    <<Model as Identifiable>::Table as QueryDsl>::Query: RunQueryDsl<PgConnection>,
{
    use diesel::update;

    update(Model::table().find(model_id))
        .set(changes)
        .get_result(conn)
}

步骤3：调用通用函数 #

现在可以用同一个函数更新Book或BookComment：

// 更新Book标题
let update_book = UpdateBook {
    title: Some("New Title".to_string()),
    author: None,
};
let updated_book = generic_update(&mut conn, 1, update_book)?;

// 更新BookComment内容
let update_comment = UpdateBookComment {
    content: Some("Updated comment content".to_string()),
};
let updated_comment = generic_update(&mut conn, 5, update_comment)?;

关键说明 #

• Identifiable：让函数能自动获取模型的主键和对应数据表
• AsChangeset：确保更新数据结构体可以被转换为Diesel的合法更新语句
• 泛型约束中的IntoUpdateTarget和RunQueryDsl保证查询能被正确执行

如果不需要单独的更新结构体，也可以直接修改模型实例并传入（只要模型实现了AsChangeset）：

let mut book = Book::get_by_id(&mut conn, 1)?;
book.title = "Updated Title".to_string();
let updated_book = generic_update(&mut conn, book.id, book)?;

内容的提问来源于stack exchange，提问作者RfDzDeveloper