-
Notifications
You must be signed in to change notification settings - Fork 1
Expand file tree
/
Copy pathTokenBase.cs
More file actions
69 lines (57 loc) · 2.03 KB
/
TokenBase.cs
File metadata and controls
69 lines (57 loc) · 2.03 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
// <auto-generated>
using System;
namespace DocSpring.Client.Client
{
/// <summary>
/// The base for all tokens.
/// </summary>
public abstract class TokenBase
{
private DateTime _nextAvailable = DateTime.UtcNow;
private object _nextAvailableLock = new object();
private readonly System.Timers.Timer _timer = new System.Timers.Timer();
internal TimeSpan? Timeout { get; set; }
internal delegate void TokenBecameAvailableEventHandler(object sender);
internal event TokenBecameAvailableEventHandler TokenBecameAvailable;
/// <summary>
/// Initialize a TokenBase object.
/// </summary>
/// <param name="timeout"></param>
internal TokenBase(TimeSpan? timeout = null)
{
Timeout = timeout;
if (Timeout != null)
StartTimer(Timeout.Value);
}
/// <summary>
/// Starts the token's timer
/// </summary>
/// <param name="timeout"></param>
internal void StartTimer(TimeSpan timeout)
{
Timeout = timeout;
_timer.Interval = Timeout.Value.TotalMilliseconds;
_timer.Elapsed += OnTimer;
_timer.AutoReset = true;
_timer.Start();
}
/// <summary>
/// Returns true while the token is rate limited.
/// </summary>
public bool IsRateLimited => _nextAvailable > DateTime.UtcNow;
/// <summary>
/// Triggered when the server returns status code TooManyRequests
/// Once triggered the local timeout will be extended an arbitrary length of time.
/// </summary>
public void BeginRateLimit()
{
lock(_nextAvailableLock)
_nextAvailable = DateTime.UtcNow.AddSeconds(5);
}
private void OnTimer(object sender, System.Timers.ElapsedEventArgs e)
{
if (TokenBecameAvailable != null && !IsRateLimited)
TokenBecameAvailable.Invoke(this);
}
}
}