Add raw data to database

1 files changed, 8 insertions, 0 deletions

diff --git a/src/db.rs b/src/db.rs
index 548ca99..5135b91 100644
--- a/src/db.rs
+++ b/src/db.rs
@@ -155,3 +155,11 @@ pub fn take_task(
         Ok(task)
     })
 }
+
+pub fn insert_data(conn: &PgConnection, data: &models::RawData) -> Result<usize, Error> {
+    use crate::schema::raw_data;
+    let rows = diesel::insert_into(raw_data::table)
+        .values(data)
+        .execute(conn)?;
+    Ok(rows)
+}