update postgres transaction to display using references in transaction callbacks

Author: tlowerison

examples/postgres/transaction/Cargo.toml

@@ -7,4 +7,4 @@ workspace = "../../../"
 [dependencies]
 sqlx = { path = "../../../", features = [ "postgres", "runtime-tokio-native-tls" ] }
 futures = "0.3.1"
-tokio = { version = "1.20.0", features = ["macros"]}
+tokio = { version = "1.20.0", features = ["macros", "rt-multi-thread"]}

examples/postgres/transaction/src/main.rs

@@ -1,22 +1,25 @@
 use sqlx::query;
+use sqlx::Acquire;
+use sqlx::Connection;
 
 async fn insert_and_verify(
     transaction: &mut sqlx::Transaction<'_, sqlx::Postgres>,
     test_id: i64,
 ) -> Result<(), Box<dyn std::error::Error>> {
+    let connection = transaction.acquire().await?;
     query!(
         r#"INSERT INTO todos (id, description)
         VALUES ( $1, $2 )
         "#,
         test_id,
         "test todo"
     )
-    .execute(&mut *transaction)
+    .execute(&mut *connection)
     .await?;
 
     // check that inserted todo can be fetched inside the uncommitted transaction
     let _ = query!(r#"SELECT FROM todos WHERE id = $1"#, test_id)
-        .fetch_one(transaction)
+        .fetch_one(&mut *connection)
         .await?;
 
     Ok(())
@@ -60,6 +63,33 @@ async fn commit_example(
     Ok(())
 }
 
+async fn insert_and_update_description(
+    pool: &sqlx::PgPool,
+    test_id: i64,
+    new_description: &str,
+) -> Result<(), Box<dyn std::error::Error>> {
+    let mut connection = pool.acquire().await?;
+
+    connection
+        .transaction(|tx| {
+            Box::pin(async {
+                insert_and_verify(tx, test_id).await.unwrap();
+                let conn = tx.acquire().await?;
+                query!(
+                    r#"UPDATE todos set description = $1 where id = $2"#,
+                    new_description,
+                    test_id,
+                )
+                .execute(conn)
+                .await?;
+                Ok::<_, sqlx::Error>(())
+            })
+        })
+        .await?;
+
+    Ok(())
+}
+
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error>> {
     let conn_str =
@@ -100,5 +130,18 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
 
     assert!(inserted_todo.is_ok());
 
+    let test_id = 2;
+
+    let description = String::from("hello world");
+    insert_and_update_description(&pool, test_id, &description).await?;
+
+    // check that inserted todo is visible outside the transaction after commit
+    let inserted_todo: Result<_, _> = query!(r#"SELECT * FROM todos WHERE id = $1"#, test_id)
+        .fetch_one(&pool)
+        .await;
+
+    assert!(inserted_todo.is_ok());
+    assert_eq!(inserted_todo.unwrap().description, description);
+
     Ok(())
 }