From a33d4b393c1031e2c35c054d3c4fe57ffea99180 Mon Sep 17 00:00:00 2001
From: Akulij <aakulij@gmail.com>
Date: Mon, 26 May 2025 20:16:12 +0500
Subject: [PATCH] impl GetCollection for CallDB

---
 src/db/mod.rs | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/src/db/mod.rs b/src/db/mod.rs
index 1a0715b..6f3ab70 100644
--- 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}")]