added semaphore to writer

mizrael · Jun 7, 2024 · e164d9a · e164d9a
1 parent ef5093e
commit e164d9a
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/src/EvenireDB/Persistence/EventsProvider.cs b/src/EvenireDB/Persistence/EventsProvider.cs
@@ -41,7 +41,17 @@ public async ValueTask AppendAsync(
     {
         var extentInfo = _extentInfoProvider.Get(streamId);
 
-        var inHeaders = _dataRepo.AppendAsync(extentInfo, events, cancellationToken);
-        await _headersRepo.AppendAsync(extentInfo, inHeaders, cancellationToken);
+        var semaphore = _streamLocks.GetOrAdd(streamId, _ => new SemaphoreSlim(1, 1));
+        await semaphore.WaitAsync(cancellationToken);
+
+        try
+        {
+            var inHeaders = _dataRepo.AppendAsync(extentInfo, events, cancellationToken);
+            await _headersRepo.AppendAsync(extentInfo, inHeaders, cancellationToken);
+        }
+        finally
+        {
+            semaphore.Release();
+        }   
     }
 }