alemi
fea7c1ecdf
finished upgrading inbox to new schema, there's ton of space for improvement but lets first see if it works
88 lines
2.2 KiB
Rust
88 lines
2.2 KiB
Rust
use sea_orm::EntityTrait;
|
|
|
|
|
|
pub async fn fix(ctx: crate::server::Context, likes: bool, shares: bool, replies: bool) -> crate::Result<()> {
|
|
use futures::TryStreamExt;
|
|
let db = ctx.db();
|
|
|
|
if likes {
|
|
tracing::info!("fixing likes...");
|
|
let mut store = std::collections::HashMap::new();
|
|
{
|
|
let mut stream = crate::model::like::Entity::find().stream(db).await?;
|
|
while let Some(like) = stream.try_next().await? {
|
|
store.insert(like.object, store.get(&like.object).unwrap_or(&0) + 1);
|
|
}
|
|
}
|
|
|
|
for (k, v) in store {
|
|
let m = crate::model::object::ActiveModel {
|
|
internal: sea_orm::Set(k),
|
|
likes: sea_orm::Set(v),
|
|
..Default::default()
|
|
};
|
|
if let Err(e) = crate::model::object::Entity::update(m)
|
|
.exec(db)
|
|
.await
|
|
{
|
|
tracing::warn!("record not updated ({k}): {e}");
|
|
}
|
|
}
|
|
}
|
|
|
|
if shares {
|
|
tracing::info!("fixing shares...");
|
|
let mut store = std::collections::HashMap::new();
|
|
{
|
|
let mut stream = crate::model::announce::Entity::find().stream(db).await?;
|
|
while let Some(share) = stream.try_next().await? {
|
|
store.insert(share.object, store.get(&share.object).unwrap_or(&0) + 1);
|
|
}
|
|
}
|
|
|
|
for (k, v) in store {
|
|
let m = crate::model::object::ActiveModel {
|
|
internal: sea_orm::Set(k),
|
|
announces: sea_orm::Set(v),
|
|
..Default::default()
|
|
};
|
|
if let Err(e) = crate::model::object::Entity::update(m)
|
|
.exec(db)
|
|
.await
|
|
{
|
|
tracing::warn!("record not updated ({k}): {e}");
|
|
}
|
|
}
|
|
}
|
|
|
|
if replies {
|
|
tracing::info!("fixing replies...");
|
|
let mut store = std::collections::HashMap::new();
|
|
{
|
|
let mut stream = crate::model::object::Entity::find().stream(db).await?;
|
|
while let Some(object) = stream.try_next().await? {
|
|
if let Some(reply) = object.in_reply_to {
|
|
let before = store.get(&reply).unwrap_or(&0);
|
|
store.insert(reply, before + 1);
|
|
}
|
|
}
|
|
}
|
|
|
|
for (k, v) in store {
|
|
let m = crate::model::object::ActiveModel {
|
|
id: sea_orm::Set(k.clone()),
|
|
replies: sea_orm::Set(v),
|
|
..Default::default()
|
|
};
|
|
if let Err(e) = crate::model::object::Entity::update(m)
|
|
.exec(db)
|
|
.await
|
|
{
|
|
tracing::warn!("record not updated ({k}): {e}");
|
|
}
|
|
}
|
|
}
|
|
|
|
tracing::info!("done running fix tasks");
|
|
Ok(())
|
|
}
|