Nopolabs\Yabot\Guzzle\ReactAwareCurlFactory

Complexity

Total Complexity

14

Size/Duplication

Total Lines	99
Duplicated Lines	0 %

Coupling/Cohesion

Components	1
Dependencies	7

Importance

Changes

0

9 Methods

Rating	Name	Duplication	Size	Complexity
A	__construct()	0	8	1
A	setHandler()	0	4	1
A	tick()	0	8	3
A	create()	0	8	1
A	release()	0	6	1
A	incrementCount()	0	8	2
A	decrementCount()	0	4	1
A	startTimer()	0	8	2
A	stopTimer()	0	9	2

<?php

namespace Nopolabs\Yabot\Guzzle;


use GuzzleHttp\Handler\CurlFactory;
use GuzzleHttp\Handler\CurlFactoryInterface;
use GuzzleHttp\Handler\CurlMultiHandler;
use GuzzleHttp\Handler\EasyHandle;
use Psr\Http\Message\RequestInterface;
use Psr\Log\LoggerInterface;
use Psr\Log\NullLogger;
use React\EventLoop\LoopInterface;
use React\EventLoop\Timer\TimerInterface;

class ReactAwareCurlFactory implements CurlFactoryInterface
{
    private $eventLoop;
    private $logger;
    private $factory;
    private $count;
    private $handler;

    /** @var TimerInterface */
    private $timer;

    public function __construct(LoopInterface $eventLoop, LoggerInterface $logger = null)
    {
        $this->eventLoop = $eventLoop;
        $this->logger = $logger ?? new NullLogger();

        $this->factory = new CurlFactory(50);
        $this->count = 0;
    }

    public function setHandler(CurlMultiHandler $handler)
    {
        $this->handler = $handler;
    }

    public function tick()
    {
        $this->handler->tick();

        if ($this->count === 0 && \GuzzleHttp\Promise\queue()->isEmpty()) {
            $this->stopTimer();
        }
    }

    /**
     * Creates a cURL handle resource.
     *
     * @param RequestInterface $request Request
     * @param array $options Transfer options
     *
     * @return EasyHandle
     * @throws \RuntimeException when an option cannot be applied
     */
    public function create(RequestInterface $request, array $options)
    {
        $easy = $this->factory->create($request, $options);

        $this->incrementCount();

        return $easy;
    }

    /**
     * Release an easy handle, allowing it to be reused or closed.
     *
     * This function must call unset on the easy handle's "handle" property.
     *
     * @param EasyHandle $easy
     */
    public function release(EasyHandle $easy)
    {
        $this->factory->release($easy);

        $this->decrementCount();
    }

    private function incrementCount()
    {
        if ($this->count === 0) {
            $this->startTimer();
        }

        $this->count++;
    }

    private function decrementCount()
    {
        $this->count--;
    }

    private function startTimer()
    {
        if ($this->timer === null) {
            $this->timer = $this->eventLoop->addPeriodicTimer(0, [$this, 'tick']);

            $this->logger->debug('ReactAwareCurlFactory started periodic queue processing');
        }
    }

    private function stopTimer()
    {
        if ($this->timer !== null) {
            $this->timer->cancel();
            $this->timer = null;

            $this->logger->debug('ReactAwareCurlFactory stopped periodic queue processing');
        }
    }
}