create MongodbStorage::from_db initializer
This commit is contained in:
parent
9e99064bc5
commit
a136558681
@ -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<S> {
|
||||
database: Database,
|
||||
serializer: S,
|
||||
@ -28,6 +30,13 @@ impl<S> MongodbStorage<S> {
|
||||
serializer,
|
||||
}))
|
||||
}
|
||||
|
||||
pub async fn from_db(db: &mut DB, serializer: S) -> Result<Arc<Self>, mongodb::error::Error> {
|
||||
Ok(Arc::new(Self {
|
||||
database: CallDB::get_database(db).await,
|
||||
serializer,
|
||||
}))
|
||||
}
|
||||
}
|
||||
|
||||
#[derive(Serialize, Deserialize)]
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user