Merge pull request sjdirect#90 from xavivars/fixing-ratelimiter-issues

sjdirect · sjdirect · commit 06f91c589ad0 · 2015-09-21T11:03:21.000-07:00
Fixing Timer.Change issues with negative dueTimes
diff --git a/Abot/Util/RateLimiter.cs b/Abot/Util/RateLimiter.cs
@@ -168,24 +168,22 @@ private void ExitTimerCallback(object state)
             // While there are exit times that are passed due still in the queue,
             // exit the semaphore and dequeue the exit time.
             int exitTime;
-            while (_exitTimes.TryPeek(out exitTime)
-                    && unchecked(exitTime - Environment.TickCount) <= 0)
+            bool exitTimeValid = _exitTimes.TryPeek(out exitTime);
+            while (exitTimeValid)
             {
+                if (unchecked(exitTime - Environment.TickCount) > 0)
+                {
+                    break;
+                }
                 _semaphore.Release();
                 _exitTimes.TryDequeue(out exitTime);
+                exitTimeValid = _exitTimes.TryPeek(out exitTime);
             }
 
-            // Try to get the next exit time from the queue and compute
-            // the time until the next check should take place. If the 
-            // queue is empty, then no exit times will occur until at least
-            // one time unit has passed.
-            int timeUntilNextCheck;
-            if (_exitTimes.TryPeek(out exitTime))
-                timeUntilNextCheck = unchecked(exitTime - Environment.TickCount);
-            else
-                timeUntilNextCheck = TimeUnitMilliseconds;
-
-            // Set the timer.
+            // we are already holding the next item from the queue, do not peek again
+            // although this exit time may have already pass by this stmt.
+            var timeUntilNextCheck = exitTimeValid ? Math.Min(TimeUnitMilliseconds, Math.Max(0, exitTime - Environment.TickCount)) : TimeUnitMilliseconds;
+
             _exitTimer.Change(timeUntilNextCheck, -1);
         }
 
