From 9c1b13bfb7f419e41ffab7298c5a90758fde10de Mon Sep 17 00:00:00 2001 From: eraden Date: Fri, 11 Nov 2022 09:53:16 +0100 Subject: [PATCH] Rewrite multi loader query use --- crates/stock_manager/src/db/products.rs | 26 ++++++++++++------------- 1 file changed, 12 insertions(+), 14 deletions(-) diff --git a/crates/stock_manager/src/db/products.rs b/crates/stock_manager/src/db/products.rs index fd2c5e1..a76c142 100644 --- a/crates/stock_manager/src/db/products.rs +++ b/crates/stock_manager/src/db/products.rs @@ -236,7 +236,7 @@ impl FindProducts { self, pool: &mut sqlx::Transaction<'_, sqlx::Postgres>, ) -> Result> { - db_utils::MultiLoad::new( + let mut loader = db_utils::MultiLoad::new( pool, r#" SELECT id, @@ -248,16 +248,17 @@ WHERE "#, "products.id =", ) - .with_size(200) - .load( - self.product_ids.len(), - self.product_ids.into_iter().map(|id| *id), - |e| { - tracing::error!("{e:?}"); - Error::FindProducts - }, - ) - .await + .with_size(200); + loader + .load( + self.product_ids.len(), + self.product_ids.into_iter().map(|id| *id), + |e| { + tracing::error!("{e:?}"); + Error::FindProducts + }, + ) + .await } } @@ -372,10 +373,7 @@ mod tests { Product { id: original.id, name: ProductName::new("a9s0dja0sjd0jas09dj"), - short_description: ProductShortDesc::new("ajs9d8ua9sdu9ahsd98has"), - long_description: ProductLongDesc::new("hja89sdy9yha9sdy98ayusd9hya9sy8dh"), category: None, - price: Price::from_u32(823794), deliver_days_flag: Day::Tuesday | Day::Saturday, } );