rate_limit

dcr/RateLimit/Middleware/RateLimitMiddleware.php

@@ -0,0 +1,38 @@
+<?php
+declare(strict_types=1);
+
+
+namespace DcrSwoole\RateLimit\Middleware;
+
+
+use App\Exception\RuntimeException;
+use App\Utils\Json;
+use Closure;
+use DcrSwoole\Log\LogBase;
+use DcrSwoole\RateLimit\RateLimitHandler;
+use DcrSwoole\Request\Request;
+use DcrSwoole\Response\Response;
+use DcrSwoole\Utils\ApplicationContext;
+
+class RateLimitMiddleware
+{
+    public function handle(): Closure
+    {
+        return static function ($request, $next) {
+            $throttler = ApplicationContext::getContainer()->get(RateLimitHandler::class);
+            // “桶”可以容纳的请求数
+            $capacity       =  60;
+            // 	“桶”完全重新装满所需的时间
+            $seconds        =  60;
+            // “桶”此操作使用的令牌数
+            $cost           =  1;
+
+            if ($throttler->handle($request->getRemoteIp(), $capacity, $seconds, $cost) === false) {
+                throw new RuntimeException('请求次数太频繁');
+            }
+
+            return $next->handle($request);
+        };
+
+    }
+}

dcr/RateLimit/RateLimitHandler.php

@@ -74,6 +74,7 @@ private function time(): int
     /**
      * Used during testing to set the current timestamp to use.
      *
+     * @param int $time
      * @return $this
      */
     public function setTestTime(int $time): self