如何在Rust的Diesel ORM中指定多个数据库schema？

和你在Hibernate中使用的「全局默认Schema+实体单独指定非默认Schema」逻辑完全对齐，具体实现步骤如下： #

1. 调整diesel.toml配置导出多Schema #

移除原来单个schema配置项，改用with-schemas指定所有需要用到的Schema，Diesel CLI生成schema.rs时会自动按Schema分模块导出：

[print_schema]
file = "src/schema.rs"
# 按你的业务领域填写所有需要用到的Schema名称，包含默认Schema
with-schemas = ["user_center", "order_center", "pay_center", "public"]

2. 自动生成的多Schema结构 #

执行diesel print-schema或者diesel migration run后，生成的src/schema.rs会自动按Schema划分模块，示例结构如下：

// 自动生成，无需手动修改
pub mod user_center {
    pub mod users {
        // 表字段、主键等元信息定义
    }
}
pub mod order_center {
    pub mod orders {
        // 表字段、主键等元信息定义
    }
}
// 默认Schema public模块
pub mod public {
    pub mod system_config {
        // 表字段、主键等元信息定义
    }
}

3. 配置全局默认Schema #

两种方式实现全局默认Schema效果：

方式1：代码别名适配（全数据库支持） #

单独创建一个别名文件，将默认Schema的所有表导出到根路径，后续使用默认Schema的表无需加前缀：

// src/schema_default.rs
pub use crate::schema::public::*;

方式2：数据库连接层设置（仅PostgreSQL支持） #

初始化数据库连接时设置search_path，默认查询会优先命中指定的Schema，无需额外代码适配：

use diesel::prelude::*;
use diesel::pg::PgConnection;

let mut conn = PgConnection::establish("postgres://账号:密码@地址/数据库名")?;
// 设置全局默认Schema，优先级最高
diesel::sql_query("SET search_path TO public")
    .execute(&mut conn)?;

4. 实体模型指定Schema #

默认Schema实体 #

直接使用提前定义好的别名即可，无需额外指定Schema：

#[derive(Queryable, Selectable)]
#[diesel(table_name = crate::schema_default::system_config)]
#[diesel(check_for_backend(diesel::pg::Pg))]
pub struct SystemConfig {
    pub id: i32,
    pub config_key: String,
    pub config_value: String,
}

非默认Schema实体 #

在table_name属性中指定对应Schema模块下的表即可，和Hibernate在实体上单独加Schema注解的逻辑完全一致：

// 订单领域模型，归属order_center Schema
#[derive(Queryable, Selectable)]
#[diesel(table_name = crate::schema::order_center::orders)]
#[diesel(check_for_backend(diesel::pg::Pg))]
pub struct Order {
    pub id: i32,
    pub order_no: String,
    pub amount: i32,
}

额外注意事项 #

• 编写迁移SQL时，非默认Schema的表需要加上Schema前缀，示例：

  CREATE TABLE order_center.orders (
      id SERIAL PRIMARY KEY,
      order_no VARCHAR NOT NULL,
      amount INT NOT NULL
  );
  

• 如果使用MySQL，没有search_path配置，直接用方式1的别名方案即可实现全局默认Schema效果。

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