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;
|
||
|
}
|
||
|
}
|