Avoid constantly attempting to acquire a lock

- ignore expiration timeout on WaitForChangesAsync when there are queued writer locks
sillsdev · Sep 13, 2024 · 70b2eea · 70b2eea
1 parent 2da4beb
commit 70b2eea
Showing 1 changed file with 13 additions and 10 deletions.
diff --git a/src/Machine/src/Serval.Machine.Shared/Services/DistributedReaderWriterLock.cs b/src/Machine/src/Serval.Machine.Shared/Services/DistributedReaderWriterLock.cs
@@ -73,8 +73,8 @@ public async Task<T> ReaderLockAsync<T>(
                 RWLock? rwLock = sub.Change.Entity;
                 if (rwLock is not null && !rwLock.IsAvailableForReading())
                 {
-                    TimeSpan? timeout = default;
-                    if (rwLock.WriterLock?.ExpiresAt is not null)
+                    TimeSpan? timeout = null;
+                    if (rwLock.WriterQueue.Count == 0 && rwLock.WriterLock?.ExpiresAt is not null)
                     {
                         timeout = rwLock.WriterLock.ExpiresAt - DateTime.UtcNow;
                         if (timeout < TimeSpan.Zero)
@@ -138,15 +138,18 @@ await _locks.UpdateAsync(
                     RWLock? rwLock = sub.Change.Entity;
                     if (rwLock is not null && !rwLock.IsAvailableForWriting(lockId))
                     {
-                        var dateTimes = rwLock.ReaderLocks.Select(l => l.ExpiresAt).ToList();
-                        if (rwLock.WriterLock?.ExpiresAt is not null)
-                            dateTimes.Add(rwLock.WriterLock.ExpiresAt);
-                        TimeSpan? timeout = default;
-                        if (dateTimes.Count > 0)
+                        TimeSpan? timeout = null;
+                        if (rwLock.WriterQueue[0].Id == lockId)
                         {
-                            timeout = dateTimes.Max() - DateTime.UtcNow;
-                            if (timeout < TimeSpan.Zero)
-                                timeout = TimeSpan.Zero;
+                            var dateTimes = rwLock.ReaderLocks.Select(l => l.ExpiresAt).ToList();
+                            if (rwLock.WriterLock?.ExpiresAt is not null)
+                                dateTimes.Add(rwLock.WriterLock.ExpiresAt);
+                            if (dateTimes.Count > 0)
+                            {
+                                timeout = dateTimes.Max() - DateTime.UtcNow;
+                                if (timeout < TimeSpan.Zero)
+                                    timeout = TimeSpan.Zero;
+                            }
                         }
                         if (timeout != TimeSpan.Zero)
                             await sub.WaitForChangeAsync(timeout, cancellationToken);