如何在Rust中使用SeaORM创建全局数据库连接

针对你的项目结构和需求，这里提供两种常用实现方式，更推荐Axum状态管理方案——它贴合Web框架设计逻辑，灵活性更高。

方案一：利用Axum的State传递数据库连接（推荐） #

这种方式将数据库连接作为全局状态注入Axum路由，Handler和Services可通过State提取器获取连接，实现复用。

步骤1：在main.rs中建立数据库连接并传入路由 #

修改main.rs，先创建数据库连接，用Arc包裹后传给路由构建函数：

use std::sync::Arc;

use axum::Router;
use sea_orm::{Database, DatabaseConnection};
use tokio::net::TcpListener;
use crate::v3::config::env;

mod v3;

#[tokio::main]
async fn main() {
    // 初始化环境变量
    env::init();

    // 从环境变量取数据库URL并建立连接
    let db_url = env::get_env("DATABASE_URL");
    let db_conn = Database::connect(&db_url).await.unwrap();
    // 用Arc包裹连接，实现线程安全的共享
    let db_conn = Arc::new(db_conn);

    let host = format!("0.0.0.0:{}", env::get_env("PORT"));

    // 将连接传入路由构建函数
    let app = v3::create_v3_router(db_conn);

    let listener = tokio::net::TcpListener::bind(&host).await.unwrap();

    println!("Server running on {}", host);
    axum::serve(listener, app).await.unwrap();
}

步骤2：修改路由构建函数，绑定全局状态 #

修改v3/mod.rs，让create_v3_router接收数据库连接参数，并将其绑定为路由状态：

use std::sync::Arc;
use axum::Router;
use sea_orm::DatabaseConnection;

pub fn create_v3_router(db_conn: Arc<DatabaseConnection>) -> Router {
    Router::new()
        // 将数据库连接作为全局状态绑定到路由
        .with_state(db_conn)
        .merge(session_routes())
}

步骤3：在Handler中获取连接并传给Services #

修改v3/routes/session.rs，通过State提取器拿到数据库连接，再传递给对应的Service：

use std::sync::Arc;
use axum::{extract::State, Json};
use sea_orm::DatabaseConnection;
// 导入你的service模块
use crate::v3::services::session_service;

pub fn routes() -> Router {
    Router::new()
        .route("/api/v3/session", get(example_handler))
}

async fn example_handler(State(db_conn): State<Arc<DatabaseConnection>>) -> Json<String> {
    // 调用service，传入数据库连接
    let result = session_service::get_session(db_conn).await;
    Json(result.unwrap_or("获取会话失败".to_string()))
}

步骤4：在Services中使用连接 #

创建v3/services/session_service.rs，编写数据库操作逻辑，接收Arc<DatabaseConnection>作为参数：

use std::sync::Arc;
use sea_orm::DatabaseConnection;

pub async fn get_session(db_conn: Arc<DatabaseConnection>) -> Result<String, sea_orm::DbErr> {
    // 这里写具体的数据库操作，比如查询会话数据
    // 示例：查询某个实体的数量
    // let count = crate::v3::entities::some_entity::Entity::find().count(db_conn.as_ref()).await?;
    Ok("会话数据获取成功".to_string())
}

方案二：使用全局静态变量（不推荐，适合简单场景） #

如果一定要用全局变量，可借助once_cell的Lazy实现，但要注意异步环境下的同步开销：

修改main.rs添加全局连接 #

use std::sync::{Arc, Mutex};
use once_cell::sync::Lazy;
use sea_orm::DatabaseConnection;

// 全局静态数据库连接，用Arc+Mutex包裹（Mutex会带来性能开销）
static DB_CONN: Lazy<Arc<Mutex<Option<DatabaseConnection>>>> = Lazy::new(|| Arc::new(Mutex::new(None)));

#[tokio::main]
async fn main() {
    env::init();
    let db_url = env::get_env("DATABASE_URL");
    let db_conn = Database::connect(&db_url).await.unwrap();
    // 初始化全局连接
    *DB_CONN.lock().unwrap() = Some(db_conn);

    // 后续路由和service可直接访问DB_CONN
    // ... 其余代码不变
}

在Services中使用全局连接 #

use std::sync::Arc;
use sea_orm::DatabaseConnection;
use crate::DB_CONN;

pub async fn get_session() -> Result<String, sea_orm::DbErr> {
    let conn = DB_CONN.lock().unwrap();
    let conn = conn.as_ref().unwrap();
    // 执行数据库操作
    Ok("会话数据获取成功".to_string())
}

关键概念说明 #

• Arc：Arc<DatabaseConnection>实现线程安全的共享所有权，因为DatabaseConnection本身不是Sync的，用Arc包裹后可在多线程异步任务中安全传递。
• Axum State：Axum的状态管理专为Web请求设计，每个请求都能安全获取共享状态，避免了全局变量的同步问题和生命周期隐患。
• 错误处理：示例中用了unwrap，实际项目应替换为更健壮的错误处理，比如返回自定义错误类型或Axum的Result。

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