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

[backend/core] Also use the NoteService KeyedLocker when it's called from ActivityHandlerService

Browse source 
This fixes sporadic "Unique constraint violation" exceptions on the "IX_note_uri" index.
This commit is contained in:
Laura Hausmann 2024-08-18 00:22:09 +02:00
parent 4be51a0cad
commit 093abe82b8
No known key found for this signature in database
GPG key ID: D044E84C5BE01605
2 changed files with 6 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
Iceshrimp.Backend/Core/Federation/ActivityPub/ActivityHandlerService.cs
View file

@ -105,6 +105,7 @@ public class ActivityHandlerService(
activity.Object = await objectResolver.ResolveObject(activity.Object, actor.Uri) as ASNote ?? activity.Object = await objectResolver.ResolveObject(activity.Object, actor.Uri) as ASNote ??
throw GracefulException.UnprocessableEntity("Failed to resolve create object"); throw GracefulException.UnprocessableEntity("Failed to resolve create object");
using (await NoteService.GetNoteProcessLockAsync(activity.Object.Id))
await noteSvc.ProcessNoteAsync(activity.Object, actor, inboxUser); await noteSvc.ProcessNoteAsync(activity.Object, actor, inboxUser);
} }
@ -335,6 +336,7 @@ public class ActivityHandlerService(
await userSvc.UpdateUserAsync(resolvedActor, actor); await userSvc.UpdateUserAsync(resolvedActor, actor);
break; break;
case ASNote note: case ASNote note:
using (await NoteService.GetNoteProcessLockAsync(note.Id))
await noteSvc.ProcessNoteUpdateAsync(note, resolvedActor); await noteSvc.ProcessNoteUpdateAsync(note, resolvedActor);
break; break;
default: default:

2
Iceshrimp.Backend/Core/Services/NoteService.cs
View file

@ -715,6 +715,8 @@ public class NoteService(
eventSvc.RaiseNoteDeleted(this, hit); eventSvc.RaiseNoteDeleted(this, hit);
} }
public static ValueTask<IDisposable> GetNoteProcessLockAsync(string uri) => KeyedLocker.LockAsync(uri);
public async Task<Note?> ProcessNoteAsync(ASNote note, User actor, User? user = null) public async Task<Note?> ProcessNoteAsync(ASNote note, User actor, User? user = null)
{ {
var dbHit = await db.Notes.IncludeCommonProperties().FirstOrDefaultAsync(p => p.Uri == note.Id); var dbHit = await db.Notes.IncludeCommonProperties().FirstOrDefaultAsync(p => p.Uri == note.Id);