198 lines
		
	
	
		
			5.5 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
			
		
		
	
	
			198 lines
		
	
	
		
			5.5 KiB
		
	
	
	
		
			PHP
		
	
	
	
	
	
<?php
 | 
						|
namespace GuzzleHttp\Handler;
 | 
						|
 | 
						|
use GuzzleHttp\Promise as P;
 | 
						|
use GuzzleHttp\Promise\Promise;
 | 
						|
use GuzzleHttp\Psr7;
 | 
						|
use Psr\Http\Message\RequestInterface;
 | 
						|
 | 
						|
/**
 | 
						|
 * Returns an asynchronous response using curl_multi_* functions.
 | 
						|
 *
 | 
						|
 * When using the CurlMultiHandler, custom curl options can be specified as an
 | 
						|
 * associative array of curl option constants mapping to values in the
 | 
						|
 * **curl** key of the provided request options.
 | 
						|
 *
 | 
						|
 * @property resource $_mh Internal use only. Lazy loaded multi-handle.
 | 
						|
 */
 | 
						|
class CurlMultiHandler
 | 
						|
{
 | 
						|
    /** @var CurlFactoryInterface */
 | 
						|
    private $factory;
 | 
						|
    private $selectTimeout;
 | 
						|
    private $active;
 | 
						|
    private $handles = [];
 | 
						|
    private $delays = [];
 | 
						|
 | 
						|
    /**
 | 
						|
     * This handler accepts the following options:
 | 
						|
     *
 | 
						|
     * - handle_factory: An optional factory  used to create curl handles
 | 
						|
     * - select_timeout: Optional timeout (in seconds) to block before timing
 | 
						|
     *   out while selecting curl handles. Defaults to 1 second.
 | 
						|
     *
 | 
						|
     * @param array $options
 | 
						|
     */
 | 
						|
    public function __construct(array $options = [])
 | 
						|
    {
 | 
						|
        $this->factory = isset($options['handle_factory'])
 | 
						|
            ? $options['handle_factory'] : new CurlFactory(50);
 | 
						|
        $this->selectTimeout = isset($options['select_timeout'])
 | 
						|
            ? $options['select_timeout'] : 1;
 | 
						|
    }
 | 
						|
 | 
						|
    public function __get($name)
 | 
						|
    {
 | 
						|
        if ($name === '_mh') {
 | 
						|
            return $this->_mh = curl_multi_init();
 | 
						|
        }
 | 
						|
 | 
						|
        throw new \BadMethodCallException();
 | 
						|
    }
 | 
						|
 | 
						|
    public function __destruct()
 | 
						|
    {
 | 
						|
        if (isset($this->_mh)) {
 | 
						|
            curl_multi_close($this->_mh);
 | 
						|
            unset($this->_mh);
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    public function __invoke(RequestInterface $request, array $options)
 | 
						|
    {
 | 
						|
        $easy = $this->factory->create($request, $options);
 | 
						|
        $id = (int) $easy->handle;
 | 
						|
 | 
						|
        $promise = new Promise(
 | 
						|
            [$this, 'execute'],
 | 
						|
            function () use ($id) { return $this->cancel($id); }
 | 
						|
        );
 | 
						|
 | 
						|
        $this->addRequest(['easy' => $easy, 'deferred' => $promise]);
 | 
						|
 | 
						|
        return $promise;
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Ticks the curl event loop.
 | 
						|
     */
 | 
						|
    public function tick()
 | 
						|
    {
 | 
						|
        // Add any delayed handles if needed.
 | 
						|
        if ($this->delays) {
 | 
						|
            $currentTime = microtime(true);
 | 
						|
            foreach ($this->delays as $id => $delay) {
 | 
						|
                if ($currentTime >= $delay) {
 | 
						|
                    unset($this->delays[$id]);
 | 
						|
                    curl_multi_add_handle(
 | 
						|
                        $this->_mh,
 | 
						|
                        $this->handles[$id]['easy']->handle
 | 
						|
                    );
 | 
						|
                }
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        // Step through the task queue which may add additional requests.
 | 
						|
        P\queue()->run();
 | 
						|
 | 
						|
        if ($this->active &&
 | 
						|
            curl_multi_select($this->_mh, $this->selectTimeout) === -1
 | 
						|
        ) {
 | 
						|
            // Perform a usleep if a select returns -1.
 | 
						|
            // See: https://bugs.php.net/bug.php?id=61141
 | 
						|
            usleep(250);
 | 
						|
        }
 | 
						|
 | 
						|
        while (curl_multi_exec($this->_mh, $this->active) === CURLM_CALL_MULTI_PERFORM);
 | 
						|
 | 
						|
        $this->processMessages();
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Runs until all outstanding connections have completed.
 | 
						|
     */
 | 
						|
    public function execute()
 | 
						|
    {
 | 
						|
        $queue = P\queue();
 | 
						|
 | 
						|
        while ($this->handles || !$queue->isEmpty()) {
 | 
						|
            // If there are no transfers, then sleep for the next delay
 | 
						|
            if (!$this->active && $this->delays) {
 | 
						|
                usleep($this->timeToNext());
 | 
						|
            }
 | 
						|
            $this->tick();
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    private function addRequest(array $entry)
 | 
						|
    {
 | 
						|
        $easy = $entry['easy'];
 | 
						|
        $id = (int) $easy->handle;
 | 
						|
        $this->handles[$id] = $entry;
 | 
						|
        if (empty($easy->options['delay'])) {
 | 
						|
            curl_multi_add_handle($this->_mh, $easy->handle);
 | 
						|
        } else {
 | 
						|
            $this->delays[$id] = microtime(true) + ($easy->options['delay'] / 1000);
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    /**
 | 
						|
     * Cancels a handle from sending and removes references to it.
 | 
						|
     *
 | 
						|
     * @param int $id Handle ID to cancel and remove.
 | 
						|
     *
 | 
						|
     * @return bool True on success, false on failure.
 | 
						|
     */
 | 
						|
    private function cancel($id)
 | 
						|
    {
 | 
						|
        // Cannot cancel if it has been processed.
 | 
						|
        if (!isset($this->handles[$id])) {
 | 
						|
            return false;
 | 
						|
        }
 | 
						|
 | 
						|
        $handle = $this->handles[$id]['easy']->handle;
 | 
						|
        unset($this->delays[$id], $this->handles[$id]);
 | 
						|
        curl_multi_remove_handle($this->_mh, $handle);
 | 
						|
        curl_close($handle);
 | 
						|
 | 
						|
        return true;
 | 
						|
    }
 | 
						|
 | 
						|
    private function processMessages()
 | 
						|
    {
 | 
						|
        while ($done = curl_multi_info_read($this->_mh)) {
 | 
						|
            $id = (int) $done['handle'];
 | 
						|
            curl_multi_remove_handle($this->_mh, $done['handle']);
 | 
						|
 | 
						|
            if (!isset($this->handles[$id])) {
 | 
						|
                // Probably was cancelled.
 | 
						|
                continue;
 | 
						|
            }
 | 
						|
 | 
						|
            $entry = $this->handles[$id];
 | 
						|
            unset($this->handles[$id], $this->delays[$id]);
 | 
						|
            $entry['easy']->errno = $done['result'];
 | 
						|
            $entry['deferred']->resolve(
 | 
						|
                CurlFactory::finish(
 | 
						|
                    $this,
 | 
						|
                    $entry['easy'],
 | 
						|
                    $this->factory
 | 
						|
                )
 | 
						|
            );
 | 
						|
        }
 | 
						|
    }
 | 
						|
 | 
						|
    private function timeToNext()
 | 
						|
    {
 | 
						|
        $currentTime = microtime(true);
 | 
						|
        $nextTime = PHP_INT_MAX;
 | 
						|
        foreach ($this->delays as $time) {
 | 
						|
            if ($time < $nextTime) {
 | 
						|
                $nextTime = $time;
 | 
						|
            }
 | 
						|
        }
 | 
						|
 | 
						|
        return max(0, $nextTime - $currentTime) * 1000000;
 | 
						|
    }
 | 
						|
}
 |