From a0d972633a9b41a4b417a59bffdf9c5d36ca4040 Mon Sep 17 00:00:00 2001
From: Billy Chan <ccw.billy.123@gmail.com>
Date: Wed, 11 Jan 2023 19:12:45 +0800
Subject: [PATCH] Implement LoaderTrait for slices (#1368)

---
 src/query/loader.rs | 31 +++++++++++++++++++++++++++++--
 1 file changed, 29 insertions(+), 2 deletions(-)

diff --git a/src/query/loader.rs b/src/query/loader.rs
index 45a5abd6..0e75d5f8 100644
--- a/src/query/loader.rs
+++ b/src/query/loader.rs
@@ -32,8 +32,35 @@ pub trait LoaderTrait {
 #[async_trait]
 impl<M> LoaderTrait for Vec<M>
 where
-    M: ModelTrait,
-    Vec<M>: Sync,
+    M: ModelTrait + Sync,
+{
+    type Model = M;
+
+    async fn load_one<R, C>(&self, stmt: Select<R>, db: &C) -> Result<Vec<Option<R::Model>>, DbErr>
+    where
+        C: ConnectionTrait,
+        R: EntityTrait,
+        R::Model: Send + Sync,
+        <<Self as LoaderTrait>::Model as ModelTrait>::Entity: Related<R>,
+    {
+        self.as_slice().load_one(stmt, db).await
+    }
+
+    async fn load_many<R, C>(&self, stmt: Select<R>, db: &C) -> Result<Vec<Vec<R::Model>>, DbErr>
+    where
+        C: ConnectionTrait,
+        R: EntityTrait,
+        R::Model: Send + Sync,
+        <<Self as LoaderTrait>::Model as ModelTrait>::Entity: Related<R>,
+    {
+        self.as_slice().load_many(stmt, db).await
+    }
+}
+
+#[async_trait]
+impl<M> LoaderTrait for &[M]
+where
+    M: ModelTrait + Sync,
 {
     type Model = M;