jmug/stoatchat
1
0
Fork 1
Code Issues Packages Projects Releases Wiki Activity

fix: continue fix

Browse Source
This commit is contained in:
Zomatree
2024-05-15 22:57:34 +01:00
parent 471e0f55e4
commit 878c1a83ad
1 changed files with 10 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
crates/core/database/src/models/admin_migrations/ops/mongodb/scripts.rs
View File

@@ -6,6 +6,7 @@ use crate::{
options::FindOptions, options::FindOptions,
}, Invite, MongoDb, DISCRIMINATOR_SEARCH_SPACE }, Invite, MongoDb, DISCRIMINATOR_SEARCH_SPACE
}; };
use bson::oid::ObjectId;
use futures::StreamExt; use futures::StreamExt;
use rand::seq::SliceRandom; use rand::seq::SliceRandom;
use revolt_permissions::DEFAULT_WEBHOOK_PERMISSIONS; use revolt_permissions::DEFAULT_WEBHOOK_PERMISSIONS;
@@ -1003,8 +1004,15 @@ pub async fn run_migrations(db: &MongoDb, revision: i32) -> i32 {
} }
); );
#[derive(serde::Serialize, serde::Deserialize)]
struct Outer {
_id: ObjectId,
#[serde(flatten)]
invite: OldInvite
}
let mut invites = db.db() let mut invites = db.db()
.collection::<OldInvite>("channel_invites") .collection::<Outer>("channel_invites")
.find(doc! { .find(doc! {
"type": { "$exists": false } "type": { "$exists": false }
}, None) }, None)
@@ -1012,7 +1020,7 @@ pub async fn run_migrations(db: &MongoDb, revision: i32) -> i32 {
.expect("failed to find invites"); .expect("failed to find invites");
while let Some(Ok(invite)) = invites.next().await { while let Some(Ok(invite)) = invites.next().await {
let new_invite = match invite { let new_invite = match invite.invite {
OldInvite::Server { code, server, creator, channel } => Invite::Server { code, server, creator, channel }, OldInvite::Server { code, server, creator, channel } => Invite::Server { code, server, creator, channel },
OldInvite::Group { code, creator, channel } => Invite::Group { code, creator, channel } OldInvite::Group { code, creator, channel } => Invite::Group { code, creator, channel }
}; };