PhpProfiler\Lib\Loop\AsyncLoopMiddleware\RetryOnExceptionMiddlewareAsync

Complexity

Total Complexity

6

Size/Duplication

Total Lines	34
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	1

2 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	5	1
A	invoke()	0	13	5

<?php

/**
 * This file is part of the sj-i/php-profiler package.
 *
 * (c) sji <sji@sj-i.dev>
 *
 * For the full copyright and license information, please view the LICENSE
 * file that was distributed with this source code.
 */

declare(strict_types=1);

namespace PhpProfiler\Lib\Loop\AsyncLoopMiddleware;

use Exception;
use PhpProfiler\Lib\Loop\AsyncLoopMiddlewareInterface;

final class RetryOnExceptionMiddlewareAsync implements AsyncLoopMiddlewareInterface
{
    private AsyncLoopMiddlewareInterface $chain;
    /** @var array<int, class-string<Exception>> */
    private array $exception_names;
    private int $max_retry;
    private int $current_retry_count = 0;

    /**
     * RetryOnExceptionLoop constructor.
     * @param int $max_retry
     * @param array<int, class-string<Exception>> $exception_names
     * @param AsyncLoopMiddlewareInterface $chain
     */
    public function __construct(int $max_retry, array $exception_names, AsyncLoopMiddlewareInterface $chain)
    {
        $this->max_retry = $max_retry;
        $this->exception_names = $exception_names;
        $this->chain = $chain;
    }

    public function invoke(): \Generator
    {
        while ($this->current_retry_count <= $this->max_retry or $this->max_retry === -1) {
            try {
                yield from $this->chain->invoke();
            } catch (Exception $e) {
                if (in_array(get_class($e), $this->exception_names, true)) {
                    $this->current_retry_count++;
                    continue;
                }
                throw $e;
            }
            $this->current_retry_count = 0;
        }
    }
}