Fix #464 - default lock timeout

sillsdev · Sep 3, 2024 · d1a4b1c · d1a4b1c
1 parent bb36491
commit d1a4b1c
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 2 deletions.
diff --git a/src/Machine/src/Serval.Machine.Shared/Configuration/ServiceOptions.cs b/src/Machine/src/Serval.Machine.Shared/Configuration/ServiceOptions.cs
@@ -5,4 +5,5 @@ public class ServiceOptions
     public const string Key = "Service";
 
     public string ServiceId { get; set; } = "machine_api";
+    public TimeSpan ReadWriteLockTimeout { get; set; } = TimeSpan.FromSeconds(55);
 }
diff --git a/src/Machine/src/Serval.Machine.Shared/Services/DistributedReaderWriterLock.cs b/src/Machine/src/Serval.Machine.Shared/Services/DistributedReaderWriterLock.cs
@@ -1,18 +1,25 @@
 namespace Serval.Machine.Shared.Services;
 
-public class DistributedReaderWriterLock(string hostId, IRepository<RWLock> locks, IIdGenerator idGenerator, string id)
-    : IDistributedReaderWriterLock
+public class DistributedReaderWriterLock(
+    string hostId,
+    IRepository<RWLock> locks,
+    IIdGenerator idGenerator,
+    string id,
+    TimeSpan defaultLifetime
+) : IDistributedReaderWriterLock
 {
     private readonly string _hostId = hostId;
     private readonly IRepository<RWLock> _locks = locks;
     private readonly IIdGenerator _idGenerator = idGenerator;
     private readonly string _id = id;
+    private readonly TimeSpan _defaultLifetime = defaultLifetime;
 
     public async Task<IAsyncDisposable> ReaderLockAsync(
         TimeSpan? lifetime = default,
         CancellationToken cancellationToken = default
     )
     {
+        lifetime ??= _defaultLifetime;
         string lockId = _idGenerator.GenerateId();
         if (!await TryAcquireReaderLock(lockId, lifetime, cancellationToken))
         {
@@ -42,6 +49,7 @@ public async Task<IAsyncDisposable> WriterLockAsync(
         CancellationToken cancellationToken = default
     )
     {
+        lifetime ??= _defaultLifetime;
         string lockId = _idGenerator.GenerateId();
         if (!await TryAcquireWriterLock(lockId, lifetime, cancellationToken))
         {