notfire/Iceshrimp.NET
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

[backend/drive] Improve media fixup algorithm

Browse source
This commit is contained in:
Laura Hausmann 2024-10-19 00:21:01 +02:00
parent d2b5cfdadc
commit b21d9691e5
No known key found for this signature in database
GPG key ID: D044E84C5BE01605
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
Iceshrimp.Backend/Core/Services/StorageMaintenanceService.cs
View file

@ -179,6 +179,10 @@ public class StorageMaintenanceService(
} }
await driveSvc.ExpireFile(file); await driveSvc.ExpireFile(file);
await db.Users.Where(p => p.AvatarId == file.Id)
.ExecuteUpdateAsync(p => p.SetProperty(u => u.AvatarUrl, file.Uri));
await db.Users.Where(p => p.BannerId == file.Id)
.ExecuteUpdateAsync(p => p.SetProperty(u => u.BannerUrl, file.Uri));
continue; continue;
} }
@ -193,7 +197,6 @@ public class StorageMaintenanceService(
file.ThumbnailAccessKey = null; file.ThumbnailAccessKey = null;
file.ThumbnailUrl = null; file.ThumbnailUrl = null;
file.ThumbnailMimeType = null; file.ThumbnailMimeType = null;
await db.SaveChangesAsync();
} }
} }
@ -209,9 +212,16 @@ public class StorageMaintenanceService(
file.PublicAccessKey = null; file.PublicAccessKey = null;
file.PublicUrl = null; file.PublicUrl = null;
file.PublicMimeType = null; file.PublicMimeType = null;
await db.SaveChangesAsync();
} }
} }
if (dryRun) continue;
await db.SaveChangesAsync();
await db.Users.Where(p => p.AvatarId == file.Id)
.ExecuteUpdateAsync(p => p.SetProperty(u => u.AvatarUrl, file.AccessUrl));
await db.Users.Where(p => p.BannerId == file.Id)
.ExecuteUpdateAsync(p => p.SetProperty(u => u.BannerUrl, file.AccessUrl));
} }
if (dryRun) if (dryRun)