Fix for jobs still considered running

Brunni · Dec 22, 2020 · 0f02264 · 0f02264
1 parent 448c5ab
commit 0f02264
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/AsyncScheduler/Scheduler.cs b/AsyncScheduler/Scheduler.cs
@@ -388,7 +388,17 @@ private async void ResolveTaskEnd(string jobKey, Task<object> task)
             }
             finally
             {
-                _runningJobs.TryRemove(jobKey, out _);
+                bool removed;
+                do
+                {
+                    removed = _runningJobs.TryRemove(jobKey, out _);
+                    if (!removed)
+                    {
+                        _logger.LogWarning("Unable to remove running job from ConcurrentDictionary: {jobKey} ... Retry...", jobKey);
+                        await Task.Delay(TimeSpan.FromMilliseconds(50));
+                    }
+                } while (removed == false);
+
                 if (task is IDisposable disposable)
                 {
                     disposable.Dispose();