1

我正在尝试在 rust/diesel/rocket 应用程序中实现连接池。我不知道如何保证 establish_pooled_connection()方法的内容只被调用一次，以便准备连接池。

这是我的代码。

来自lib.rs：

pub fn establish_pooled_connection() -> PooledConnection<ConnectionManager<PgConnection>> {
    dotenv().ok();

    let database_url = env::var("DB_URL")
        .expect("DATABASE_URL must be set");

    let manager = ConnectionManager::<PgConnection>::new(&database_url);
    let pool = r2d2::Pool::builder().build(manager).expect("Failed to create pool.");
    let conn = pool.clone().get().unwrap();
    return conn;
}

这里我在main.rs中使用了上述方法：

#[get("/", format = "json")]
fn find_all() -> Json<Vec<Person>> {
    let connection = establish_pooled_connection();

    let all: QueryResult<Vec<Person>> = person::table().get_results(&connection);
    ...

这里的问题是每个get方法（例如上面的）调用establish_pooled_connection()并且一切都被重新实例化......

我来自 java 世界，依赖注入允许我们避免重新实例化。

在 rust/diesel 应用程序中实现连接池的正确方法是什么？

4

1 回答 1

2

您不会在每个处理程序中创建新连接，而是使用框架的一些状态共享机制。

有关如何在火箭中使用状态的指南，请参阅此处，这将是一个很好的起点。

于 2021-08-03T10:02:07.450 回答