From a136558681f52fc5855775ec93b0f6f5a6bcd822 Mon Sep 17 00:00:00 2001 From: Akulij Date: Tue, 27 May 2025 04:09:47 +0500 Subject: [PATCH] create MongodbStorage::from_db initializer --- src/mongodb_storage.rs | 9 +++++++++ 1 file changed, 9 insertions(+) diff --git a/src/mongodb_storage.rs b/src/mongodb_storage.rs index 9289aef..6a703dc 100644 --- a/src/mongodb_storage.rs +++ b/src/mongodb_storage.rs @@ -9,6 +9,8 @@ use mongodb::Database; use serde::{de::DeserializeOwned, Deserialize, Serialize}; use teloxide::dispatching::dialogue::{Serializer, Storage}; +use crate::db::{CallDB, DB}; + pub struct MongodbStorage { database: Database, serializer: S, @@ -28,6 +30,13 @@ impl MongodbStorage { serializer, })) } + + pub async fn from_db(db: &mut DB, serializer: S) -> Result, mongodb::error::Error> { + Ok(Arc::new(Self { + database: CallDB::get_database(db).await, + serializer, + })) + } } #[derive(Serialize, Deserialize)]