Add triggering code

Author: momobel

examples/sync-wrapper/src/main.rs

@@ -4,6 +4,7 @@ use diesel_async::async_connection_wrapper::AsyncConnectionWrapper;
 use diesel_async::sync_connection_wrapper::SyncConnectionWrapper;
 use diesel_async::{AsyncConnection, RunQueryDsl, SimpleAsyncConnection};
 use diesel_migrations::{embed_migrations, EmbeddedMigrations, MigrationHarness};
+use futures_util::FutureExt;
 
 // ordinary diesel model setup
 
@@ -15,7 +16,7 @@ table! {
 }
 
 #[allow(dead_code)]
-#[derive(Debug, Queryable, Selectable)]
+#[derive(Debug, Queryable, QueryableByName, Selectable)]
 #[diesel(table_name = users)]
 struct User {
     id: i32,
@@ -47,6 +48,38 @@ where
     .map_err(|e| Box::new(e) as Box<dyn std::error::Error>)
 }
 
+async fn transaction(
+    async_conn: &mut SyncConnectionWrapper<InnerConnection>,
+    old_name: &str,
+    new_name: &str,
+) -> Result<Vec<User>, diesel::result::Error> {
+    async_conn
+        .transaction::<Vec<User>, diesel::result::Error, _>(|c| {
+            Box::pin(async {
+                if old_name.is_empty() {
+                    Ok(Vec::new())
+                } else {
+                    diesel::sql_query(
+                        r#"
+                    update
+                        users
+                    set
+                        name = ?2
+                    where
+                        name == ?1
+                    returning *
+                "#,
+                    )
+                    .bind::<diesel::sql_types::Text, _>(old_name)
+                    .bind::<diesel::sql_types::Text, _>(new_name)
+                    .load(c)
+                    .await
+                }
+            })
+        })
+        .await
+}
+
 #[tokio::main]
 async fn main() -> Result<(), Box<dyn std::error::Error>> {
     let db_url = std::env::var("DATABASE_URL").expect("Env var `DATABASE_URL` not set");
@@ -86,5 +119,32 @@ async fn main() -> Result<(), Box<dyn std::error::Error>> {
         .await?;
     println!("{data:?}");
 
+    // let changed = transaction(&mut sync_wrapper, "iLuke", "JustLuke").await?;
+    // println!("Changed {changed:?}");
+
+    // create an async connection for the migrations
+    let mut conn_a: SyncConnectionWrapper<InnerConnection> = establish(&db_url).await?;
+    let mut conn_b: SyncConnectionWrapper<InnerConnection> = establish(&db_url).await?;
+
+    tokio::spawn(async move {
+        loop {
+            let changed = transaction(&mut conn_a, "iLuke", "JustLuke").await;
+            println!("Changed {changed:?}");
+            std::thread::sleep(std::time::Duration::from_secs(1));
+        }
+    });
+
+    tokio::spawn(async move {
+        loop {
+            let changed = transaction(&mut conn_b, "JustLuke", "iLuke").await;
+            println!("Changed {changed:?}");
+            std::thread::sleep(std::time::Duration::from_secs(1));
+        }
+    });
+
+    loop {
+        std::thread::sleep(std::time::Duration::from_secs(1));
+    }
+
     Ok(())
 }