azaion/annotations
1
0
Fork 0
mirror of https://github.com/azaion/annotations.git synced 2026-04-22 22:26:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix throttle ext

Browse Source 
fix configs
fix build scripts
This commit is contained in:
Alex Bezdieniezhnykh
2025-04-17 19:40:09 +03:00
parent 277aaf09b0
commit d42409de7d
9 changed files with 21 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files
Azaion.Common/Extensions/ThrottleExtensions.cs
+5 -3
View File
@@ -6,7 +6,7 @@ public static class ThrottleExt
{
private class ThrottleState(Func<Task> action)
{
public Func<Task> Action { get; } = action ?? throw new ArgumentNullException(nameof(action));
public Func<Task> Action { get; set; } = action ?? throw new ArgumentNullException(nameof(action));
public bool IsCoolingDown = false;
public bool CallScheduledDuringCooldown = false;
public Task CooldownTask = Task.CompletedTask;
@@ -15,7 +15,7 @@ public static class ThrottleExt
private static readonly ConcurrentDictionary<Guid, ThrottleState> ThrottlerStates = new();
public static void Throttle(Func<Task> action, Guid actionId, TimeSpan interval)
public static void Throttle(Func<Task> action, Guid actionId, TimeSpan interval, bool scheduleCallAfterCooldown = false)
{
ArgumentNullException.ThrowIfNull(action);
if (actionId == Guid.Empty)
@@ -24,6 +24,7 @@ public static class ThrottleExt
throw new ArgumentOutOfRangeException(nameof(interval), "Interval must be positive.");
var state = ThrottlerStates.GetOrAdd(actionId, new ThrottleState(action));
state.Action = action;
lock (state.StateLock)
{
@@ -34,7 +35,8 @@ public static class ThrottleExt
}
else
{
state.CallScheduledDuringCooldown = true;
if (scheduleCallAfterCooldown)
state.CallScheduledDuringCooldown = true;
}
}
}