Add a lock signature

sillsdev · Sep 4, 2024 · 92fbec0 · 92fbec0
1 parent 76e5ed9
commit 92fbec0
Show file tree

Hide file tree

Showing 3 changed files with 16 additions and 2 deletions.
diff --git a/src/Machine/src/Serval.Machine.Shared/Configuration/IMachineBuilderExtensions.cs b/src/Machine/src/Serval.Machine.Shared/Configuration/IMachineBuilderExtensions.cs
@@ -268,7 +268,7 @@ await c.Indexes.CreateOrUpdateAsync(
                         );
                     }
                 );
-                o.AddRepository<RWLock>("locks");
+                o.AddRepository<RWLock>("locks", mapSetup: m => m.SetIgnoreExtraElements(true));
                 o.AddRepository<TrainSegmentPair>(
                     "train_segment_pairs",
                     init: c =>

diff --git a/src/Machine/src/Serval.Machine.Shared/Models/Lock.cs b/src/Machine/src/Serval.Machine.Shared/Models/Lock.cs
@@ -5,4 +5,5 @@ public record Lock
     public required string Id { get; init; }
     public DateTime? ExpiresAt { get; init; }
     public required string HostId { get; init; }
+    public string CallingSignature { get; init; } = "";
 }
diff --git a/src/Machine/src/Serval.Machine.Shared/Services/DistributedReaderWriterLock.cs b/src/Machine/src/Serval.Machine.Shared/Services/DistributedReaderWriterLock.cs
@@ -51,11 +51,24 @@ public async Task<IAsyncDisposable> WriterLockAsync(
     {
         lifetime ??= _defaultLifetime;
         string lockId = _idGenerator.GenerateId();
+        StackTrace stackTrace = new StackTrace();
+        string callingSignature =
+            $"Filename: {stackTrace.GetFrame(1)?.GetFileName()}, Method: {stackTrace.GetFrame(1)?.GetMethod()?.Name}";
+
         if (!await TryAcquireWriterLock(lockId, lifetime, cancellationToken))
         {
             await _locks.UpdateAsync(
                 _id,
-                u => u.Add(rwl => rwl.WriterQueue, new Lock { Id = lockId, HostId = _hostId }),
+                u =>
+                    u.Add(
+                        rwl => rwl.WriterQueue,
+                        new Lock
+                        {
+                            Id = lockId,
+                            HostId = _hostId,
+                            CallingSignature = callingSignature
+                        }
+                    ),
                 cancellationToken: cancellationToken
             );
             try