如何在SQLx中使用带生命周期的Transaction配合Query的execute函数？

问题背景 #

想要将Rust代码中的逐行数据库插入改为事务批量插入，现有代码编译时遇到两个错误：

原代码 #

pub async fn bulk_insert_foos(self, foos: Vec<Foo>) -> Result<(), Error> {
    let chunks = foos.chunks(10_000);
    for chunk in chunks {
        &self.insert_foo_chunk(chunk).await?;
    }

    Ok(())
}

async fn insert_foo_chunk(&self, chunk: &[Foo]) -> Result<(), Error>
{
    let mut tx = self.pool.begin().await?;
    for foo in chunk {
        let _ = &self.insert_foo(&mut  tx, foo).await?;
    }
    tx.commit().await?;

    Ok(())
}

async fn insert_foo(&self, tx: &mut Transaction<Postgres>, foo: &Foo) -> Result<(), Error> {
    let _result = sqlx::query(
        r#"INSERT INTO public.foo (bar) VALUES($1)"#)
        .bind(&foo.bar)
        .execute(&mut *tx).await
        .map_err(|e| {
            println!("{e}");
            dbg!(&foo);
            e
        })?;

    Ok(())
}

编译错误1 #

implicit elided lifetime not allowed here [E0726] 
expected lifetime parameter 
Help: indicate the anonymous lifetime

编译错误2（尝试添加Transaction<'_, Postgres>后） #

the trait bound `&mut sqlx::Transaction<'_, Postgres>: Executor<'_>` is not satisfied [E0277] 
the trait `Executor<'_>` is not implemented for `&mut sqlx::Transaction<'_, Postgres>`

Help: the following other types implement trait `Executor<'c>`:
`&'c mut PgConnection` implements `Executor<'c>` 
`&'c mut PgListener` implements `Executor<'c>` 
`&Pool<DB>` implements `Executor<'p>` 

Note: required by a bound in `Query::<'q, DB, A>::execute`

解决方案 #

修改后的代码 #

pub async fn bulk_insert_foos(self, foos: Vec<Foo>) -> Result<(), Error> {
    let chunks = foos.chunks(10_000);
    for chunk in chunks {
        self.insert_foo_chunk(chunk).await?;
    }

    Ok(())
}

async fn insert_foo_chunk(&self, chunk: &[Foo]) -> Result<(), Error>
{
    let mut tx = self.pool.begin().await?;
    for foo in chunk {
        self.insert_foo(&mut tx, foo).await?;
    }
    tx.commit().await?;

    Ok(())
}

async fn insert_foo(&self, tx: &mut Transaction<'_, Postgres>, foo: &Foo) -> Result<(), Error> {
    let _result = sqlx::query(
        r#"INSERT INTO public.foo (bar) VALUES($1)"#)
        .bind(&foo.bar)
        .execute(tx).await
        .map_err(|e| {
            println!("{e}");
            dbg!(&foo);
            e
        })?;

    Ok(())
}

关键修改点 #

1. 显式标注Transaction的生命周期：将insert_foo函数的参数tx: &mut Transaction<Postgres>改为tx: &mut Transaction<'_, Postgres>，解决隐式生命周期缺失的错误。
2. 直接传递Transaction引用给execute：去掉execute(&mut *tx)中的解引用操作，直接使用execute(tx)。因为Transaction<'_, Postgres>本身实现了Executor<'_> trait，无需手动解引用为连接类型。
3. 移除多余的引用操作：在bulk_insert_foos和insert_foo_chunk中，去掉调用函数前的&和let _ = &，这些多余的引用会导致类型不匹配。

错误原因解释 #

• 第一个错误是因为Transaction结构体带有生命周期参数，Rust编译器无法自动推断，必须显式标注匿名生命周期'_。
• 第二个错误是因为手动解引用&mut *tx后得到的&mut PgConnection虽然实现了Executor，但传递的&mut Transaction<'_, Postgres>的引用层多余了——Transaction本身的可变引用直接实现了Executor，无需额外解引用。

内容的提问来源于stack exchange，提问作者Jérôme MEVEL