feat: add retry after policy to api exception

This commit solves task inter-860
fingerprintjs · Aug 28, 2024 · 64e0510 · 64e0510
1 parent 2a8be2e
commit 64e0510
Show file tree

Hide file tree

Showing 4 changed files with 66 additions and 0 deletions.
diff --git a/src/Api/FingerprintApi.php b/src/Api/FingerprintApi.php
@@ -149,6 +149,13 @@ public function getEvent(string $request_id): array
                     break;
             }
 
+            if (429 === $e->getCode()) {
+                $e->setRetryAfter(1);
+                if ($response->hasHeader('retry-after')) {
+                    $e->setRetryAfter((int) $response->getHeader('retry-after'));
+                }
+            }
+
             throw $e;
         }
     }
@@ -219,6 +226,13 @@ function ($e) {
                             break;
                     }
 
+                    if (429 === $e->getCode()) {
+                        $e->setRetryAfter(1);
+                        if ($response->hasHeader('retry-after')) {
+                            $e->setRetryAfter((int) $response->getHeader('retry-after'));
+                        }
+                    }
+
                     throw $e;
                 }
             );
@@ -308,6 +322,13 @@ public function getVisits(string $visitor_id, ?string $request_id = null, ?strin
                     break;
             }
 
+            if (429 === $e->getCode()) {
+                $e->setRetryAfter(1);
+                if ($response->hasHeader('retry-after')) {
+                    $e->setRetryAfter((int) $response->getHeader('retry-after'));
+                }
+            }
+
             throw $e;
         }
     }
@@ -383,6 +404,13 @@ function ($e) {
                             break;
                     }
 
+                    if (429 === $e->getCode()) {
+                        $e->setRetryAfter(1);
+                        if ($response->hasHeader('retry-after')) {
+                            $e->setRetryAfter((int) $response->getHeader('retry-after'));
+                        }
+                    }
+
                     throw $e;
                 }
             );

diff --git a/src/ApiException.php b/src/ApiException.php
@@ -44,6 +44,7 @@ class ApiException extends \Exception
 {
     protected ResponseInterface $responseObject;
     protected ModelInterface $errorDetails;
+    protected ?int $retryAfter = null;
 
     public function __construct(?string $message = '', ?int $code = 0)
     {
@@ -72,4 +73,14 @@ public function setErrorDetails(ModelInterface $errorDetails): void
     {
         $this->errorDetails = $errorDetails;
     }
+
+    public function getRetryAfter(): ?int
+    {
+        return $this->retryAfter;
+    }
+
+    public function setRetryAfter(?int $retryAfter): void
+    {
+        $this->retryAfter = $retryAfter;
+    }
 }
diff --git a/template/ApiException.mustache b/template/ApiException.mustache
@@ -34,6 +34,7 @@ class ApiException extends Exception
 {
     protected ResponseInterface $responseObject;
     protected ModelInterface $errorDetails;
+    protected ?int $retryAfter = null;
 
     public function __construct(?string $message = "", ?int $code = 0)
     {
@@ -63,4 +64,14 @@ class ApiException extends Exception
     {
         $this->errorDetails = $errorDetails;
     }
+
+    public function getRetryAfter(): ?int
+    {
+        return $this->retryAfter;
+    }
+
+    public function setRetryAfter(?int $retryAfter): void
+    {
+        $this->retryAfter = $retryAfter;
+    }
 }
diff --git a/template/api.mustache b/template/api.mustache
@@ -137,6 +137,14 @@ use \GuzzleHttp\Exception\GuzzleException;
                     {{/dataType}}
                 {{/responses}}
                 }
+
+                if ($e->getCode() === 429) {
+                    $e->setRetryAfter(1);
+                    if ($response->hasHeader('retry-after')) {
+                        $e->setRetryAfter((int)$response->getHeader('retry-after'));
+                    }
+                }
+
                 throw $e;
             }
     }
@@ -203,6 +211,14 @@ use \GuzzleHttp\Exception\GuzzleException;
                         {{/dataType}}
                     {{/responses}}
                     }
+
+                    if ($e->getCode() === 429) {
+                        $e->setRetryAfter(1);
+                        if ($response->hasHeader('retry-after')) {
+                            $e->setRetryAfter((int)$response->getHeader('retry-after'));
+                        }
+                    }
+
                     throw $e;
                 }
             );