diff --git a/src/Middleware/RetryMiddleware.php b/src/Middleware/RetryMiddleware.php new file mode 100644 index 000000000..86b7091ec --- /dev/null +++ b/src/Middleware/RetryMiddleware.php @@ -0,0 +1,97 @@ +retryAttempts = $config['retryAttempts']; + } + + /** + * Retry the response if it has retryable error codes. + * + * use Google\Auth\Middleware\AuthTokenMiddleware; + * use Google\Auth\Middleware\RetryMiddleware; + * use Google\Auth\OAuth2; + * use GuzzleHttp\Client; + * use GuzzleHttp\HandlerStack; + * + * $config = [...]; + * $oauth2 = new OAuth2($config) + * $authMiddleware = new AuthTokenMiddleware($oauth2); + * $retryMiddleware = new RetryMiddleware(['retryAttempts' => 3]); + * $stack = HandlerStack::create(); + * $stack->push($authMiddleware); + * $stack->push($retryMiddleware); + * + * $client = new Client([ + * 'handler' => $stack, + * 'base_uri' => 'https://www.googleapis.com/taskqueue/v1beta2/projects/', + * 'auth' => 'google_auth' // authorize all requests + * ]); + * + * $res = $client->get('myproject/taskqueues/myqueue'); + * + * @param callable $handler + * @return \Closure + */ + public function __invoke(callable $handler) + { + return function (RequestInterface $request, array $options) use ($handler) { + $response = $handler($request, $options); + for ($attempts = 0; $attempts < $this->retryAttempts; $attempts++) { + if ($this->isRetryable($response)) { + $response = $handler($request, $options); + } + } + return $response; + }; + } + + private function isRetryable(ResponseInterface $response) + { + // no-op for now, this is a Proof of Concept! + return false; + } +}