impl GetCollection for CallDB

2025-05-26 20:16:12 +05:00 · 2025-05-26 20:16:12 +05:00 · a33d4b393c
commit a33d4b393c
parent 9c15b0a375
1 changed files with 8 additions and 0 deletions
--- a/src/db/mod.rs
+++ b/src/db/mod.rs
@ -208,6 +208,14 @@ impl CallDB for DB {
    }
 }

+impl<T: CallDB> GetCollection for T {
+    async fn get_collection<C: DbCollection + Send + Sync>(&mut self) -> Collection<C> {
+        self.get_database()
+            .await
+            .collection(<C as DbCollection>::COLLECTION)
+    }
+}
+
 #[derive(thiserror::Error, Debug)]
 pub enum DbError {
    #[error("error while processing mongodb query: {0}")]