[chain] When inserting balances, do it in chunks to avoid hitting que…

…ry limits

chain/src/repository/balance.rs

@@ -18,21 +18,29 @@ pub fn insert_balance(
     transaction_conn: &mut PgConnection,
     balances: Balances,
 ) -> anyhow::Result<()> {
-    diesel::insert_into(balance_changes::table)
-        .values::<&Vec<BalanceChangesInsertDb>>(
-            &balances
-                .into_iter()
-                .map(BalanceChangesInsertDb::from_balance)
-                .collect::<Vec<_>>(),
-        )
-        .on_conflict((
-            balance_changes::columns::owner,
-            balance_changes::columns::token,
-            balance_changes::columns::height,
-        ))
-        .do_nothing()
-        .execute(transaction_conn)
-        .context("Failed to update balances in db")?;
+    tracing::info!("Inserting {} balances into db", balances.len());
+
+    // Group balances into chunks to avoid hitting the limit of the number of bind parameters in one query.
+    balances.chunks(10000).try_for_each(|balances_chunk| {
+        diesel::insert_into(balance_changes::table)
+            .values::<&Vec<BalanceChangesInsertDb>>(
+                &balances_chunk
+                    .iter()
+                    .cloned()
+                    .map(BalanceChangesInsertDb::from_balance)
+                    .collect::<Vec<_>>(),
+            )
+            .on_conflict((
+                balance_changes::columns::owner,
+                balance_changes::columns::token,
+                balance_changes::columns::height,
+            ))
+            .do_nothing()
+            .execute(transaction_conn)
+            .context("Failed to update balances in db")?;
+
+        anyhow::Ok(())
+    })?;
 
     anyhow::Ok(())
 }