Init

ilyakharev · ilyakharev · commit 54e920bc6f5e · 2023-06-12T22:30:11.000+03:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,3 +1,5 @@
+* added retry function
+
 ## 1.5.5
 
 * fixed iam auth for PHP < 8.0
diff --git a/src/Exceptions/NonRetryableException.php b/src/Exceptions/NonRetryableException.php
@@ -0,0 +1,7 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions;
+
+class NonRetryableException extends \Exception
+{
+}
diff --git a/src/Exceptions/RetryableException.php b/src/Exceptions/RetryableException.php
@@ -0,0 +1,20 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions;
+
+class RetryableException extends \Exception
+{
+    /**
+     * @var bool
+     */
+    protected $fastBackoff;
+
+    /**
+     * @return bool
+     */
+    public function isFastBackoff(): bool
+    {
+        return $this->fastBackoff;
+    }
+
+}
diff --git a/src/Exceptions/SdkException.php b/src/Exceptions/SdkException.php
@@ -0,0 +1,34 @@
+<?php
+namespace YdbPlatform\Ydb\Exceptions;
+
+class SdkException extends \YdbPlatform\Ydb\Exception
+{
+    /**
+     * @var string
+     */
+    protected $errorCode;
+
+    /**
+     * @var bool
+     */
+    protected $retryable;
+
+    /**
+     * @return string
+     */
+    public function getErrorCode(): string
+    {
+        return $this->errorCode;
+    }
+
+    /**
+     * @return bool
+     */
+    public function isRetryable()
+    {
+        return $this->retryable;
+    }
+    public function setErrorCode($errorCode){
+        $this->errorCode = $errorCode;
+    }
+}
diff --git a/src/Exceptions/Ydb/AbortedException.php b/src/Exceptions/Ydb/AbortedException.php
@@ -0,0 +1,13 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+use YdbPlatform\Ydb\Exceptions\RetryableException;
+
+class AbortedException extends RetryableException
+{
+    public function isFastBackoff(): bool
+    {
+        return true;
+    }
+}
diff --git a/src/Exceptions/Ydb/AlreadyExistsException.php b/src/Exceptions/Ydb/AlreadyExistsException.php
@@ -0,0 +1,8 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+class AlreadyExistsException extends \YdbPlatform\Ydb\Exceptions\NonRetryableException
+{
+
+}
diff --git a/src/Exceptions/Ydb/BadRequestException.php b/src/Exceptions/Ydb/BadRequestException.php
@@ -0,0 +1,10 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+use YdbPlatform\Ydb\Exceptions\NonRetryableException;
+
+class BadRequestException extends NonRetryableException
+{
+
+}
diff --git a/src/Exceptions/Ydb/BadSessionException.php b/src/Exceptions/Ydb/BadSessionException.php
@@ -0,0 +1,13 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+use Ydb\StatusIds\StatusCode;
+
+class BadSessionException extends \YdbPlatform\Ydb\Exceptions\RetryableException
+{
+    public function isFastBackoff(): bool
+    {
+        return true;
+    }
+}
diff --git a/src/Exceptions/Ydb/CanceledException.php b/src/Exceptions/Ydb/CanceledException.php
@@ -0,0 +1,8 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+class CanceledException extends \YdbPlatform\Ydb\Exceptions\NonRetryableException
+{
+
+}
diff --git a/src/Exceptions/Ydb/ClientResourceExhaustedException.php b/src/Exceptions/Ydb/ClientResourceExhaustedException.php
@@ -0,0 +1,8 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+class ClientResourceExhaustedException extends \YdbPlatform\Ydb\Exceptions\NonRetryableException
+{
+
+}
diff --git a/src/Exceptions/Ydb/GenericErrorException.php b/src/Exceptions/Ydb/GenericErrorException.php
@@ -0,0 +1,10 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+use YdbPlatform\Ydb\Exceptions\NonRetryableException;
+
+class GenericErrorException extends NonRetryableException
+{
+
+}
diff --git a/src/Exceptions/Ydb/InternalErrorException.php b/src/Exceptions/Ydb/InternalErrorException.php
@@ -0,0 +1,10 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+use YdbPlatform\Ydb\Exceptions\NonRetryableException;
+
+class InternalErrorException extends NonRetryableException
+{
+
+}
diff --git a/src/Exceptions/Ydb/NotFoundException.php b/src/Exceptions/Ydb/NotFoundException.php
@@ -0,0 +1,8 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+class NotFoundException extends \YdbPlatform\Ydb\Exceptions\NonRetryableException
+{
+
+}
diff --git a/src/Exceptions/Ydb/OverloadException.php b/src/Exceptions/Ydb/OverloadException.php
@@ -0,0 +1,11 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+class OverloadException extends \YdbPlatform\Ydb\Exceptions\RetryableException
+{
+    public function isFastBackoff(): bool
+    {
+        return false;
+    }
+}
diff --git a/src/Exceptions/Ydb/PreconditionFailedException.php b/src/Exceptions/Ydb/PreconditionFailedException.php
@@ -0,0 +1,8 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+class PreconditionFailedException extends \YdbPlatform\Ydb\Exceptions\NonRetryableException
+{
+
+}
diff --git a/src/Exceptions/Ydb/SchemeError.php b/src/Exceptions/Ydb/SchemeError.php
@@ -0,0 +1,8 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+class SchemeError extends \YdbPlatform\Ydb\Exceptions\NonRetryableException
+{
+
+}
diff --git a/src/Exceptions/Ydb/SessionBusyException.php b/src/Exceptions/Ydb/SessionBusyException.php
@@ -0,0 +1,13 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+use YdbPlatform\Ydb\Exceptions\RetryableException;
+
+class SessionBusyException extends RetryableException
+{
+    public function isFastBackoff(): bool
+    {
+        return true;
+    }
+}
diff --git a/src/Exceptions/Ydb/SessionExpiredException.php b/src/Exceptions/Ydb/SessionExpiredException.php
@@ -0,0 +1,8 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+class SessionExpiredException extends \YdbPlatform\Ydb\Exceptions\NonRetryableException
+{
+
+}
diff --git a/src/Exceptions/Ydb/TimeoutException.php b/src/Exceptions/Ydb/TimeoutException.php
@@ -0,0 +1,8 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+class TimeoutException extends \YdbPlatform\Ydb\Exceptions\NonRetryableException
+{
+
+}
diff --git a/src/Exceptions/Ydb/UnauthorisedException.php b/src/Exceptions/Ydb/UnauthorisedException.php
@@ -0,0 +1,10 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+use YdbPlatform\Ydb\Exceptions\NonRetryableException;
+
+class UnauthorisedException extends NonRetryableException
+{
+
+}
diff --git a/src/Exceptions/Ydb/UnavailableException.php b/src/Exceptions/Ydb/UnavailableException.php
@@ -0,0 +1,12 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+class UnavailableException extends \YdbPlatform\Ydb\Exceptions\RetryableException
+{
+    public function isFastBackoff(): bool
+    {
+        return true;
+    }
+
+}
diff --git a/src/Exceptions/Ydb/UndeterminedException.php b/src/Exceptions/Ydb/UndeterminedException.php
@@ -0,0 +1,13 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+use YdbPlatform\Ydb\Exceptions\RetryableException;
+
+class UndeterminedException extends RetryableException
+{
+    public function isFastBackoff(): bool
+    {
+        return true;
+    }
+}
diff --git a/src/Exceptions/Ydb/UnsupportedException.php b/src/Exceptions/Ydb/UnsupportedException.php
@@ -0,0 +1,8 @@
+<?php
+
+namespace YdbPlatform\Ydb\Exceptions\Ydb;
+
+class UnsupportedException extends \YdbPlatform\Ydb\Exceptions\NonRetryableException
+{
+
+}
diff --git a/src/Retry/Backoff.php b/src/Retry/Backoff.php
@@ -0,0 +1,36 @@
+<?php
+
+namespace YdbPlatform\Ydb\Retry;
+
+class Backoff
+{
+    protected $backoffCeiling;
+    protected $backoffSlotMillis;
+
+    /**
+     * @param $backoffCeiling
+     * @param $backoffSlotMillis
+     */
+    public function __construct($backoffCeiling, $backoffSlotMillis)
+    {
+        $this->backoffCeiling = $backoffCeiling;
+        $this->backoffSlotMillis = $backoffSlotMillis;
+    }
+
+    /**
+     * @return mixed
+     */
+    public function getBackoffCeiling()
+    {
+        return $this->backoffCeiling;
+    }
+
+    /**
+     * @return mixed
+     */
+    public function getBackoffSlotMillis()
+    {
+        return $this->backoffSlotMillis;
+    }
+
+}
diff --git a/src/Retry/Retry.php b/src/Retry/Retry.php
@@ -0,0 +1,41 @@
+<?php
+
+namespace YdbPlatform\Ydb\Retry;
+
+use Closure;
+use YdbPlatform\Ydb\Exceptions\NonRetryableException;
+use YdbPlatform\Ydb\Exceptions\RetryableException;
+
+class Retry
+{
+
+    protected function retryDelay(int $retryCount, Backoff $backoff)
+    {
+        return $backoff->getBackoffSlotMillis()*(1<<min($retryCount, $backoff->getBackoffCeiling()));
+    }
+
+    /**
+     * @throws NonRetryableException
+     * @throws RetryableException
+     */
+    public function retry(Closure $closure, RetryParams $params, bool $idempotent){
+        $startTime = microtime(true);
+        $retryCount = 0;
+        $lastException = null;
+        while (microtime(true) < $startTime+$params->getTimeoutMs()/1000){
+            try {
+                return $closure();
+            } catch (RetryableException $e){
+                $retryCount++;
+                $this->retryDelay($retryCount,
+                    $e->isFastBackoff() ? $params->getFastBackOff() : $params->getSlowBackOff());
+                $lastException = $e;
+            } catch (NonRetryableException $e){
+                throw $e;
+            }
+        }
+        throw $lastException;
+    }
+
+
+}
diff --git a/src/Retry/RetryParams.php b/src/Retry/RetryParams.php
diff --git a/src/Table.php b/src/Table.php
diff --git a/src/Ydb.php b/src/Ydb.php

Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,5 @@`
	`1`	`+* added retry function`
	`2`	`+`
`1`	`3`	`## 1.5.5`
`2`	`4`
`3`	`5`	`* fixed iam auth for PHP < 8.0`
-Original file line number
+Diff line change
@@ @@ -0,0 +1,7 @@ @@
 +<?php
++
 +namespace YdbPlatform\Ydb\Exceptions;
++
 +class NonRetryableException extends \Exception
 +{
 +}