From 5824c353f978a5b9ab9696451ee3014c6a614ffc Mon Sep 17 00:00:00 2001
From: Joe Carstairs <me@joeac.net>
Date: Tue, 22 Oct 2024 07:46:43 +0100
Subject: Refactors importer to use core stuff

---
 backend/core/src/queries/budget_updates/insert.rs | 28 +++++++++++++++++++++++
 1 file changed, 28 insertions(+)
 create mode 100644 backend/core/src/queries/budget_updates/insert.rs

(limited to 'backend/core/src/queries/budget_updates/insert.rs')

diff --git a/backend/core/src/queries/budget_updates/insert.rs b/backend/core/src/queries/budget_updates/insert.rs
new file mode 100644
index 0000000..e8ae271
--- /dev/null
+++ b/backend/core/src/queries/budget_updates/insert.rs
@@ -0,0 +1,28 @@
+use anyhow::{Context, Result};
+use crate::{
+    models::budget_update::BudgetUpdate,
+    schema::budget_updates::dsl::budget_updates as budget_updates_table,
+};
+use diesel::prelude::*;
+
+pub fn insert_budget_updates(
+    budget_updates: &[BudgetUpdate],
+    connection: &mut SqliteConnection,
+) -> Result<usize> {
+    let num_rows_inserted = diesel::insert_into(budget_updates_table)
+        .values(budget_updates)
+        .execute(connection)
+        .with_context(|| err_msg(budget_updates))?;
+    Ok(num_rows_inserted)
+}
+
+fn err_msg(budget_updates: &[BudgetUpdate]) -> String {
+    format!(
+        "failed to insert budget updates: [{}]",
+        budget_updates
+            .iter()
+            .map(|a| a.id.to_string())
+            .collect::<Vec<String>>()
+            .join(", ")
+    )
+}
-- 
cgit v1.2.3