LeoCarmo\CircuitBreaker\GuzzleMiddleware

Complexity

Total Complexity

17

Size/Duplication

Total Lines	87
Duplicated Lines	0 %

Importance

Changes	1
Bugs	0	Features	1

8 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	3	1
A	setCustomSuccessCodes()	0	3	1
A	setCustomIgnoreCodes()	0	3	1
A	__invoke()	0	21	2
A	isStatusCodeSuccess()	0	3	2
A	isStatusCodeRangeValid()	0	3	2
A	executeCircuitBreakerOnResponse()	0	23	6
A	isStatusCodeRedirect()	0	3	2

<?php declare(strict_types=1);

namespace LeoCarmo\CircuitBreaker;

use Psr\Http\Message\RequestInterface;
use Psr\Http\Message\ResponseInterface;

class GuzzleMiddleware
{
    protected CircuitBreaker $circuitBreaker;

    protected array $customSuccessCodes = [];

    protected array $customIgnoreCodes = [];

    public function __construct(CircuitBreaker $circuitBreaker)
    {
        $this->circuitBreaker = $circuitBreaker;
    }

    public function setCustomSuccessCodes(array $codes): void
    {
        $this->customSuccessCodes = $codes;
    }

    public function setCustomIgnoreCodes(array $codes): void
    {
        $this->customIgnoreCodes = $codes;
    }

    public function __invoke(callable $handler): \Closure
    {
        return function (RequestInterface $request, array $options) use ($handler) {

            if (! $this->circuitBreaker->isAvailable()) {
                throw new CircuitBreakerException(
                    sprintf('"%s" is not available', $this->circuitBreaker->getService())
                );
            }

            $promise = $handler($request, $options);

            return $promise->then(
                function (ResponseInterface $response) {
                    $this->executeCircuitBreakerOnResponse($response);

                    return $response;
                },
                function (\Throwable $exception) {
                    $this->circuitBreaker->failure();
                    throw $exception;
                }
            );
        };
    }

    protected function executeCircuitBreakerOnResponse(ResponseInterface $response): void
    {
        $statusCode = $response->getStatusCode();

        if (in_array($statusCode, $this->customIgnoreCodes)) {
            return;
        }

        if (! $this->isStatusCodeRangeValid($statusCode)) {
            $this->circuitBreaker->failure();
            return;
        }

        if ($this->isStatusCodeRedirect($statusCode)) {
            return;
        }

        if ($this->isStatusCodeSuccess($statusCode) || in_array($statusCode, $this->customSuccessCodes)) {
            $this->circuitBreaker->success();
            return;
        }

        $this->circuitBreaker->failure();
    }

    protected function isStatusCodeRangeValid(int $statusCode): bool
    {
        return ($statusCode >= 100 && $statusCode < 600);
    }

    protected function isStatusCodeRedirect(int $statusCode): bool
    {
        return ($statusCode >= 300 && $statusCode < 400);
    }

    protected function isStatusCodeSuccess(int $statusCode): bool
    {
        return ($statusCode >= 200 && $statusCode < 300);
    }
}