DysonNetwork/Swarm
1
1
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

⏪ The post scanner now migrate outdated attachments again

Browse Source
This commit is contained in:
LittleSheep 2025-06-06 00:21:19 +08:00
parent f62d86d4a7
commit b6aa0e83a3
1 changed files with 38 additions and 24 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

58
DysonNetwork.Sphere/Storage/FileService.ReferenceMigration.cs
View File

@ -33,29 +33,43 @@ public class FileReferenceMigrationService(AppDatabase db)
private async Task ScanPosts() private async Task ScanPosts()
{ {
var posts = await db.Posts var posts = await db.Posts
.Include(p => p.OutdatedAttachments)
.Where(p => p.OutdatedAttachments.Any())
.ToListAsync(); .ToListAsync();
var attachmentsId = posts.SelectMany(p => p.Attachments.Select(a => a.Id)).ToList(); foreach (var post in posts)
var attachments = {
await db.Files.Where(f => attachmentsId.Contains(f.Id)).ToDictionaryAsync(x => x.Id); var updatedAttachments = new List<CloudFileReferenceObject>();
var fileReferences = posts.SelectMany(post => post.Attachments.Select(attachment => foreach (var attachment in post.OutdatedAttachments)
{ {
var value = attachments.TryGetValue(attachment.Id, out var file) ? file : null; var file = await db.Files.FirstOrDefaultAsync(f => f.Id == attachment.Id);
if (value is null) return null; if (file != null)
return new CloudFileReference
{ {
FileId = value.Id, // Create a reference for the file
File = value, var reference = new CloudFileReference
{
FileId = file.Id,
File = file,
Usage = "post", Usage = "post",
ResourceId = post.Id.ToString(), ResourceId = post.ResourceIdentifier
CreatedAt = SystemClock.Instance.GetCurrentInstant(),
UpdatedAt = SystemClock.Instance.GetCurrentInstant()
}; };
}).Where(x => x != null).Select(x => x!)
).ToList();
await db.BulkInsertAsync(fileReferences); await db.FileReferences.AddAsync(reference);
updatedAttachments.Add(file.ToReferenceObject());
}
else
{
// Keep the existing reference object if file not found
updatedAttachments.Add(attachment.ToReferenceObject());
}
}
post.Attachments = updatedAttachments;
db.Posts.Update(post);
}
await db.SaveChangesAsync();
} }
private async Task ScanMessages() private async Task ScanMessages()
@ -71,7 +85,7 @@ public class FileReferenceMigrationService(AppDatabase db)
FileId = attachment.Id, FileId = attachment.Id,
File = attachment, File = attachment,
Usage = "chat", Usage = "chat",
ResourceId = message.Id.ToString(), ResourceId = message.ResourceIdentifier,
CreatedAt = SystemClock.Instance.GetCurrentInstant(), CreatedAt = SystemClock.Instance.GetCurrentInstant(),
UpdatedAt = SystemClock.Instance.GetCurrentInstant() UpdatedAt = SystemClock.Instance.GetCurrentInstant()
}) })
@ -157,7 +171,7 @@ public class FileReferenceMigrationService(AppDatabase db)
FileId = avatarFile.Id, FileId = avatarFile.Id,
File = avatarFile, File = avatarFile,
Usage = "chatroom.picture", Usage = "chatroom.picture",
ResourceId = chatRoom.Id.ToString() ResourceId = chatRoom.ResourceIdentifier
}; };
await db.FileReferences.AddAsync(reference); await db.FileReferences.AddAsync(reference);
@ -176,7 +190,7 @@ public class FileReferenceMigrationService(AppDatabase db)
FileId = bannerFile.Id, FileId = bannerFile.Id,
File = bannerFile, File = bannerFile,
Usage = "chatroom.background", Usage = "chatroom.background",
ResourceId = chatRoom.Id.ToString() ResourceId = chatRoom.ResourceIdentifier
}; };
await db.FileReferences.AddAsync(reference); await db.FileReferences.AddAsync(reference);
@ -208,7 +222,7 @@ public class FileReferenceMigrationService(AppDatabase db)
FileId = avatarFile.Id, FileId = avatarFile.Id,
File = avatarFile, File = avatarFile,
Usage = "realm.picture", Usage = "realm.picture",
ResourceId = realm.Id.ToString() ResourceId = realm.ResourceIdentifier
}; };
await db.FileReferences.AddAsync(reference); await db.FileReferences.AddAsync(reference);
@ -228,7 +242,7 @@ public class FileReferenceMigrationService(AppDatabase db)
FileId = bannerFile.Id, FileId = bannerFile.Id,
File = bannerFile, File = bannerFile,
Usage = "realm.background", Usage = "realm.background",
ResourceId = realm.Id.ToString() ResourceId = realm.ResourceIdentifier
}; };
await db.FileReferences.AddAsync(reference); await db.FileReferences.AddAsync(reference);
@ -280,7 +294,7 @@ public class FileReferenceMigrationService(AppDatabase db)
FileId = backgroundFile.Id, FileId = backgroundFile.Id,
File = backgroundFile, File = backgroundFile,
Usage = "publisher.background", Usage = "publisher.background",
ResourceId = publisher.Id.ToString() ResourceId = publisher.ResourceIdentifier
}; };
await db.FileReferences.AddAsync(reference); await db.FileReferences.AddAsync(reference);
@ -311,7 +325,7 @@ public class FileReferenceMigrationService(AppDatabase db)
FileId = imageFile.Id, FileId = imageFile.Id,
File = imageFile, File = imageFile,
Usage = "sticker.image", Usage = "sticker.image",
ResourceId = sticker.Id.ToString() ResourceId = sticker.ResourceIdentifier
}; };
await db.FileReferences.AddAsync(reference); await db.FileReferences.AddAsync(reference);