feat: batch insert

Author: omfj

.sqlx/query-7187b571969c741edbc079028bf27ae6614e9e2ce1bd97d4d75c90b97aaf3bce.json

@@ -61,20 +61,13 @@ pub async fn bake(
         msg.build()
     };
 
-    for entry in &baked {
-        if let Err(e) = ctx
-            .data()
-            .orders
-            .record_order(&entry.user_id, &guild_id)
-            .await
-        {
-            error!(
-                user_id = %entry.user_id,
-                guild_id = %guild_id,
-                error = ?e,
-                "Failed to record order"
-            );
-        }
+    let user_ids: Vec<&str> = baked.iter().map(|e| e.user_id.as_str()).collect();
+    if let Err(e) = ctx.data().orders.record_orders(&user_ids, &guild_id).await {
+        error!(
+            guild_id = %guild_id,
+            error = ?e,
+            "Failed to record orders"
+        );
     }
 
     ctx.say(message).await?;

.sqlx/query-bfea23a5a6ba80a5e8a52607b5741437c767c0aa8928590674d8c813a97b11d3.json

@@ -1,6 +1,6 @@
 #[async_trait::async_trait]
 pub trait OrderRepository: Send + Sync {
-    async fn record_order(&self, discord_user_id: &str, guild_id: &str) -> anyhow::Result<()>;
+    async fn record_orders(&self, discord_user_ids: &[&str], guild_id: &str) -> anyhow::Result<()>;
     async fn daily_stats(&self, guild_id: &str) -> anyhow::Result<DailyStats>;
 }
 

src/adapters/discord/commands/bake.rs

@@ -15,21 +15,43 @@ impl PostgresOrderRepository {
 
 #[async_trait::async_trait]
 impl OrderRepository for PostgresOrderRepository {
-    #[instrument(skip(self), fields(discord_user_id, guild_id))]
-    async fn record_order(&self, discord_user_id: &str, guild_id: &str) -> anyhow::Result<()> {
-        debug!(discord_user_id, guild_id, "Recording order");
+    #[instrument(skip(self), fields(count = discord_user_ids.len(), guild_id))]
+    async fn record_orders(&self, discord_user_ids: &[&str], guild_id: &str) -> anyhow::Result<()> {
+        if discord_user_ids.is_empty() {
+            debug!("No orders to record");
+            return Ok(());
+        }
+
+        debug!(
+            count = discord_user_ids.len(),
+            guild_id, "Recording orders in batch"
+        );
+
+        let discord_user_ids_vec: Vec<String> =
+            discord_user_ids.iter().map(|&s| s.to_string()).collect();
+        let guild_ids: Vec<String> = vec![guild_id.to_string(); discord_user_ids.len()];
+
         sqlx::query!(
-            "INSERT INTO orders (discord_user_id, guild_id) VALUES ($1, $2)",
-            discord_user_id,
-            guild_id
+            "INSERT INTO orders (discord_user_id, guild_id) SELECT * FROM UNNEST($1::text[], $2::text[])",
+            &discord_user_ids_vec[..],
+            &guild_ids[..]
         )
         .execute(&self.pool)
         .await
         .map_err(|e| {
-            error!(discord_user_id, guild_id, error = ?e, "Failed to record order in database");
+            error!(
+                count = discord_user_ids.len(),
+                guild_id,
+                error = ?e,
+                "Failed to record orders in database"
+            );
             e
         })?;
-        info!(discord_user_id, guild_id, "Order recorded successfully");
+
+        info!(
+            count = discord_user_ids.len(),
+            guild_id, "Orders recorded successfully"
+        );
         Ok(())
     }