Struct TenantContext 

pub struct TenantContext;

全局租户上下文 — 持有当前请求的 tenant_id

线程安全

内部使用线程局部存储（thread_local!），每个线程持有独立的租户上下文。 在 axum 中间件（运行于请求线程）中设置后，同线程的业务代码通过 Self::current() 读取。 多线程并发场景下各线程互不干扰，适合测试并行执行。

Implementations

impl TenantContext

pub fn set_current(tenant_id: i64)

设置当前租户 ID

pub fn clear()

清除当前租户 ID（请求结束后调用）

pub fn current() -> Option<i64>

获取当前租户 ID

未设置时返回 None。业务代码应在此情况下拒绝数据操作。

pub fn require_current() -> Result<i64, TenantError>

获取当前租户 ID，未设置时返回错误

pub fn is_set() -> bool

判断是否已设置租户上下文

pub fn guard() -> Result<TenantGuard, TenantError>

创建 TenantGuard — 在 await 前捕获当前租户 ID

为什么需要 Guard

TenantContext 基于 thread_local! 存储。在 tokio 异步运行时中， .await 可能导致任务切换到不同线程，使 thread_local 值静默改变。

在业务代码需要跨 await 使用租户 ID 时，应先在 await 前调用本方法 创建 TenantGuard，之后通过 guard.tenant_id() 访问（而非 TenantContext::current()）。

使用示例

async fn handle_request() -> Result<(), TenantError> {
    // 在第一个 await 之前捕获租户 ID
    let guard = TenantContext::guard()?;

    // 以下操作可能跨越 await，但 guard 持有正确的 tenant_id
    let data = fetch_from_db(guard.tenant_id()).await?;
    process(data, guard.tenant_id()).await?;

    Ok(())
}

安全保证

• guard.tenant_id() 返回创建时捕获的值，不受线程切换影响
• guard.assert_current() 可验证当前 thread_local 是否与捕获值一致 （用于检测中间件是否正确设置了上下文）