akulij/gongbotrs
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

make CallbackInfo::store to consume self and return

Browse Source
This commit is contained in:
Akulij 2025-04-29 19:51:54 +03:00
parent a00c0017bb
commit 58a16a5927
2 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/db/callback_info.rs
View File

@ -1,4 +1,4 @@
use crate::query_call; use crate::query_call_consume;
use crate::CallDB; use crate::CallDB;
use bson::oid::ObjectId; use bson::oid::ObjectId;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
@ -31,13 +31,13 @@ where
self._id.to_hex() self._id.to_hex()
} }
query_call!(store, self, db, (), { query_call_consume!(store, self, db, Self, {
let db = db.get_database().await; let db = db.get_database().await;
let ci = db.collection::<Self>("callback_info"); let ci = db.collection::<Self>("callback_info");
ci.insert_one(self).await?; ci.insert_one(&self).await?;
Ok(()) Ok(self)
}); });
pub async fn get<D: CallDB>(db: &mut D, id: &str) -> DbResult<Option<Self>> { pub async fn get<D: CallDB>(db: &mut D, id: &str) -> DbResult<Option<Self>> {

4
src/db/tests/callback_info_tests.rs
View File

@ -1,4 +1,4 @@
use super::super::{callback_info::CallbackInfo, CallDB, DB}; use super::super::callback_info::CallbackInfo;
use super::setup_db; use super::setup_db;
use serde::{Deserialize, Serialize}; use serde::{Deserialize, Serialize};
@ -20,7 +20,7 @@ async fn test_store() {
let ci = CI::new(Default::default()); let ci = CI::new(Default::default());
ci.store(&mut db).await.unwrap(); let ci = ci.store(&mut db).await.unwrap();
let ci = CI::get(&mut db, &ci.get_id()).await.unwrap(); let ci = CI::get(&mut db, &ci.get_id()).await.unwrap();