只使用一个缓存键完成计数固定窗口

Author: hongweipeng

src/throttle/CounterFixed.php

@@ -14,21 +14,13 @@ class CounterFixed extends ThrottleAbstract
     public function allowRequest(string $key, float $micronow, int $max_requests, int $duration, $cache)
     {
         $cur_requests = $cache->get($key, 0);
-        $limit_flag = $cache->get($key . 'flag', null);
         $now = (int) $micronow;
         $wait_reset_seconds = $duration - $now % $duration;     // 距离下次重置还有n秒时间
-
         $this->wait_seconds = $wait_reset_seconds % $duration  + 1;
-        if ($limit_flag === null) { // 首次访问
-            $cur_requests = 1;
-            $cache->set($key, $cur_requests, $wait_reset_seconds);
-            $cache->set($key . 'flag', 1, $wait_reset_seconds);
-            $this->cur_requests = $cur_requests;
-            return true;
-        }
         $this->cur_requests = $cur_requests;
+
         if ($cur_requests < $max_requests) {   // 允许访问
-            $cache->inc($key);
+            $cache->set($key, $this->cur_requests + 1, $wait_reset_seconds);
             return true;
         }