如何在Rust Diesel中使用自引用外键

要在Diesel v2.x里处理自引用表，需要完成以下几个核心步骤：

1. 完善schema定义 #

首先确保你的schema.rs中正确声明自引用外键和关联关系：

diesel::table! {
    modality (id) {
        id -> Int4,
        // 替换为你的实际字段
        name -> Varchar,
        reply_to -> Int4,
        // 添加外键约束（可选，根据数据库要求）
        foreign_key (reply_to) -> modality (id),
    }
}

// 声明自引用的可关联关系
diesel::joinable!(modality -> modality (reply_to));

2. 为结构体添加关联支持 #

在你的Modality结构体上补充Associations派生宏，并指定自引用的关联规则：

#[derive(Debug, PartialEq, Identifiable, Queryable, Selectable, Associations)]
#[diesel(table_name = modality)]
// 指定当前结构体属于自身，外键为reply_to
#[diesel(belongs_to(Modality, foreign_key = reply_to))]
pub struct Modality {
    pub id: i32,
    pub name: String,
    // 建议用Option<i32>，支持无父节点的顶级记录
    pub reply_to: Option<i32>,
}

3. 执行关联查询 #

现在可以用Diesel的关联方法轻松查询父/子节点：

查询单条记录的父节点 #

use diesel::{QueryDsl, RunQueryDsl};

fn get_parent(conn: &mut PgConnection, target_id: i32) -> QueryResult<Option<Modality>> {
    let target = Modality::find(target_id).first(conn)?;
    target.belongs_to(conn)
}

查询某节点的所有子节点 #

fn get_children(conn: &mut PgConnection, parent_id: i32) -> QueryResult<Vec<Modality>> {
    // 构造父节点占位实例，仅需id字段有效
    let parent = Modality { id: parent_id, name: String::new(), reply_to: None };
    Modality::belonging_to(&parent).load(conn)
}

4. 插入自引用记录 #

插入时直接指定reply_to即可，确保父记录已存在（如果有外键约束）：

#[derive(Insertable)]
#[diesel(table_name = modality)]
pub struct NewModality {
    pub name: String,
    pub reply_to: Option<i32>,
}

fn create_modality(conn: &mut PgConnection, new_mod: NewModality) -> QueryResult<Modality> {
    use crate::schema::modality::dsl::*;

    diesel::insert_into(modality)
        .values(&new_mod)
        .returning(Modality::as_returning())
        .get_result(conn)
}

额外说明 #

如果需要处理递归树形查询（比如获取整个分支），Diesel本身不提供直接的递归CTE封装，但可以通过sql_query宏手写递归SQL实现。

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