ApiClients\Middleware\Pool\PoolMiddleware

Complexity

Total Complexity

5

Size/Duplication

Total Lines	48
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	0

Importance

Changes

0

3 Methods

Rating	Name	Duplication	Size	Complexity
A	pre()	0	13	2
A	post()	0	8	2
A	priority()	0	4	1

<?php declare(strict_types=1);

namespace ApiClients\Middleware\Pool;

use ApiClients\Foundation\Middleware\MiddlewareInterface;
use ApiClients\Foundation\Middleware\Priority;
use Psr\Http\Message\RequestInterface;
use Psr\Http\Message\ResponseInterface;
use React\Promise\CancellablePromiseInterface;
use ResourcePool\Allocation;
use ResourcePool\Pool;
use function React\Promise\resolve;

class PoolMiddleware implements MiddlewareInterface
{
    /**
     * @var Allocation
     */
    private $allocation;

    /**
     * @param RequestInterface $request
     * @param array $options
     * @return CancellablePromiseInterface
     */
    public function pre(RequestInterface $request, array $options = []): CancellablePromiseInterface
    {
        if (!isset($options[self::class][Options::POOL])) {
            return resolve($request);
        }

        /** @var Pool $pool */
        $pool = $options[self::class][Options::POOL];
        return $pool->allocateOne()->then(function (Allocation $allocation) use ($request) {
            $this->allocation = $allocation;
            return resolve($request);
        });
    }

    /**
     * @param ResponseInterface $response
     * @param array $options
     * @return CancellablePromiseInterface
     */
    public function post(ResponseInterface $response, array $options = []): CancellablePromiseInterface
    {
        if ($this->allocation instanceof Allocation) {
            $this->allocation->releaseOne();
        }

        return resolve($response);
    }

    /**
     * @return int
     */
    public function priority(): int
    {
        return Priority::FIRST;
    }
}