123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162 |
- <?php
- /*
- * This file is part of the Symfony package.
- *
- * (c) Fabien Potencier <fabien@symfony.com>
- *
- * For the full copyright and license information, please view the LICENSE
- * file that was distributed with this source code.
- */
- namespace Symfony\Component\HttpClient;
- use Psr\Log\LoggerInterface;
- use Psr\Log\NullLogger;
- use Symfony\Component\HttpClient\Response\AsyncContext;
- use Symfony\Component\HttpClient\Response\AsyncResponse;
- use Symfony\Component\HttpClient\Retry\GenericRetryStrategy;
- use Symfony\Component\HttpClient\Retry\RetryStrategyInterface;
- use Symfony\Contracts\HttpClient\ChunkInterface;
- use Symfony\Contracts\HttpClient\Exception\TransportExceptionInterface;
- use Symfony\Contracts\HttpClient\HttpClientInterface;
- use Symfony\Contracts\HttpClient\ResponseInterface;
- /**
- * Automatically retries failing HTTP requests.
- *
- * @author Jérémy Derussé <jeremy@derusse.com>
- */
- class RetryableHttpClient implements HttpClientInterface
- {
- use AsyncDecoratorTrait;
- private $strategy;
- private $maxRetries;
- private $logger;
- /**
- * @param int $maxRetries The maximum number of times to retry
- */
- public function __construct(HttpClientInterface $client, RetryStrategyInterface $strategy = null, int $maxRetries = 3, LoggerInterface $logger = null)
- {
- $this->client = $client;
- $this->strategy = $strategy ?? new GenericRetryStrategy();
- $this->maxRetries = $maxRetries;
- $this->logger = $logger ?: new NullLogger();
- }
- public function request(string $method, string $url, array $options = []): ResponseInterface
- {
- if ($this->maxRetries <= 0) {
- return new AsyncResponse($this->client, $method, $url, $options);
- }
- $retryCount = 0;
- $content = '';
- $firstChunk = null;
- return new AsyncResponse($this->client, $method, $url, $options, function (ChunkInterface $chunk, AsyncContext $context) use ($method, $url, $options, &$retryCount, &$content, &$firstChunk) {
- $exception = null;
- try {
- if ($chunk->isTimeout() || null !== $chunk->getInformationalStatus()) {
- yield $chunk;
- return;
- }
- } catch (TransportExceptionInterface $exception) {
- // catch TransportExceptionInterface to send it to the strategy
- }
- if (null !== $exception) {
- // always retry request that fail to resolve DNS
- if ('' !== $context->getInfo('primary_ip')) {
- $shouldRetry = $this->strategy->shouldRetry($context, null, $exception);
- if (null === $shouldRetry) {
- throw new \LogicException(sprintf('The "%s::shouldRetry()" method must not return null when called with an exception.', \get_class($this->decider)));
- }
- if (false === $shouldRetry) {
- $context->passthru();
- if (null !== $firstChunk) {
- yield $firstChunk;
- yield $context->createChunk($content);
- yield $chunk;
- } else {
- yield $chunk;
- }
- $content = '';
- return;
- }
- }
- } elseif ($chunk->isFirst()) {
- if (false === $shouldRetry = $this->strategy->shouldRetry($context, null, null)) {
- $context->passthru();
- yield $chunk;
- return;
- }
- // Body is needed to decide
- if (null === $shouldRetry) {
- $firstChunk = $chunk;
- $content = '';
- return;
- }
- } else {
- $content .= $chunk->getContent();
- if (!$chunk->isLast()) {
- return;
- }
- if (null === $shouldRetry = $this->strategy->shouldRetry($context, $content, null)) {
- throw new \LogicException(sprintf('The "%s::shouldRetry()" method must not return null when called with a body.', \get_class($this->strategy)));
- }
- if (false === $shouldRetry) {
- $context->passthru();
- yield $firstChunk;
- yield $context->createChunk($content);
- $content = '';
- return;
- }
- }
- $context->getResponse()->cancel();
- $delay = $this->getDelayFromHeader($context->getHeaders()) ?? $this->strategy->getDelay($context, !$exception && $chunk->isLast() ? $content : null, $exception);
- ++$retryCount;
- $this->logger->info('Try #{count} after {delay}ms'.($exception ? ': '.$exception->getMessage() : ', status code: '.$context->getStatusCode()), [
- 'count' => $retryCount,
- 'delay' => $delay,
- ]);
- $context->setInfo('retry_count', $retryCount);
- $context->replaceRequest($method, $url, $options);
- $context->pause($delay / 1000);
- if ($retryCount >= $this->maxRetries) {
- $context->passthru();
- }
- });
- }
- private function getDelayFromHeader(array $headers): ?int
- {
- if (null !== $after = $headers['retry-after'][0] ?? null) {
- if (is_numeric($after)) {
- return (int) $after * 1000;
- }
- if (false !== $time = strtotime($after)) {
- return max(0, $time - time()) * 1000;
- }
- }
- return null;
- }
- }
|