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

[backend/core] Add AsyncKeyedLocker to SystemUserService

Browse source
This commit is contained in:
Laura Hausmann 2024-06-21 17:01:49 +02:00
parent c49973becf
commit b23348cde9
No known key found for this signature in database
GPG key ID: D044E84C5BE01605
1 changed files with 14 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

18
Iceshrimp.Backend/Core/Services/SystemUserService.cs
View file

@ -1,4 +1,5 @@
using System.Security.Cryptography; using System.Security.Cryptography;
using AsyncKeyedLock;
using Iceshrimp.Backend.Core.Database; using Iceshrimp.Backend.Core.Database;
using Iceshrimp.Backend.Core.Database.Tables; using Iceshrimp.Backend.Core.Database.Tables;
using Iceshrimp.Backend.Core.Helpers; using Iceshrimp.Backend.Core.Helpers;
@ -9,6 +10,12 @@ namespace Iceshrimp.Backend.Core.Services;
public class SystemUserService(ILogger<SystemUserService> logger, DatabaseContext db, CacheService cache) public class SystemUserService(ILogger<SystemUserService> logger, DatabaseContext db, CacheService cache)
{ {
private static readonly AsyncKeyedLocker<string> KeyedLocker = new(o =>
{
o.PoolSize = 10;
o.PoolInitialFill = 2;
});
public async Task<User> GetInstanceActorAsync() public async Task<User> GetInstanceActorAsync()
{ {
return await GetOrCreateSystemUserAsync("instance.actor"); return await GetOrCreateSystemUserAsync("instance.actor");
@ -41,10 +48,13 @@ public class SystemUserService(ILogger<SystemUserService> logger, DatabaseContex
{ {
return await cache.FetchAsync($"systemUser:{username}", TimeSpan.FromHours(24), async () => return await cache.FetchAsync($"systemUser:{username}", TimeSpan.FromHours(24), async () =>
{ {
logger.LogTrace("GetOrCreateSystemUser delegate method called for user {username}", username); using (await KeyedLocker.LockAsync(username.ToLowerInvariant()))
return await db.Users.FirstOrDefaultAsync(p => p.UsernameLower == username.ToLowerInvariant() && {
p.IsLocalUser) ?? logger.LogTrace("GetOrCreateSystemUser delegate method called for user {username}", username);
await CreateSystemUserAsync(username); return await db.Users.FirstOrDefaultAsync(p => p.UsernameLower == username.ToLowerInvariant() &&
p.IsLocalUser) ??
await CreateSystemUserAsync(username);
}
}); });
} }