fix

Author: Glubus

src/models/beatmaps/pending_beatmap/impl.rs

@@ -1,4 +1,6 @@
-use super::query::{delete_by_id, find_by_hash, find_by_id, insert, last_pending_beatmap};
+use super::query::{
+    bulk_insert, delete_by_id, find_by_hash, find_by_id, insert, last_pending_beatmap,
+};
 use super::PendingBeatmapRow;
 use sqlx::{Error as SqlxError, PgPool};
 
@@ -22,4 +24,8 @@ impl PendingBeatmapRow {
     pub async fn delete_by_id(pool: &PgPool, id: i32) -> Result<bool, SqlxError> {
         delete_by_id(pool, id).await
     }
+
+    pub async fn bulk_insert(pool: &PgPool, rows: &[PendingBeatmapRow]) -> Result<i64, SqlxError> {
+        bulk_insert(pool, rows).await
+    }
 }

src/models/beatmaps/pending_beatmap/query/bulk_insert.rs

@@ -0,0 +1,21 @@
+use crate::models::beatmaps::pending_beatmap::types::PendingBeatmapRow;
+use sqlx::{Error as SqlxError, PgPool};
+use validator::Validate;
+
+pub async fn bulk_insert(pool: &PgPool, rows: &[PendingBeatmapRow]) -> Result<i64, SqlxError> {
+    let filtered: Vec<&PendingBeatmapRow> = rows.iter().filter(|n| n.validate().is_ok()).collect();
+
+    if filtered.is_empty() {
+        return Ok(0);
+    }
+    let hashes: Vec<&str> = filtered.iter().map(|r| r.osu_hash.as_str()).collect();
+    let res = sqlx::query(
+        "INSERT INTO pending_beatmap (osu_hash)
+         SELECT DISTINCT UNNEST($1::text[])
+         ON CONFLICT (osu_hash) DO NOTHING",
+    )
+    .bind(&hashes)
+    .execute(pool)
+    .await?;
+    Ok(res.rows_affected() as i64)
+}

src/models/beatmaps/pending_beatmap/query/mod.rs

@@ -1,8 +1,10 @@
+pub mod bulk_insert;
 pub mod by_id;
 pub mod delete_by_id;
 pub mod insert;
 pub mod last;
 
+pub use bulk_insert::*;
 pub use by_id::*;
 pub use delete_by_id::*;
 pub use insert::*;

src/models/beatmaps/pending_beatmap/types.rs

@@ -30,3 +30,9 @@ pub struct PendingBeatmapRow {
     /// Timestamp when the pending beatmap was created.
     pub created_at: Option<NaiveDateTime>,
 }
+
+#[derive(Debug, Clone)]
+pub struct PendingBeatmapNew {
+    pub osu_hash: String,
+    pub osu_id: Option<i32>,
+}