|
|
|
<?php
|
|
|
|
|
|
|
|
/*
|
|
|
|
* This file is part of Composer.
|
|
|
|
*
|
|
|
|
* (c) Nils Adermann <naderman@naderman.de>
|
|
|
|
* Jordi Boggiano <j.boggiano@seld.be>
|
|
|
|
*
|
|
|
|
* For the full copyright and license information, please view the LICENSE
|
|
|
|
* file that was distributed with this source code.
|
|
|
|
*/
|
|
|
|
|
|
|
|
namespace Composer\Util;
|
|
|
|
|
|
|
|
use React\Promise\Promise;
|
|
|
|
use Symfony\Component\Console\Helper\ProgressBar;
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @author Jordi Boggiano <j.boggiano@seld.be>
|
|
|
|
*/
|
|
|
|
class Loop
|
|
|
|
{
|
|
|
|
/** @var HttpDownloader */
|
|
|
|
private $httpDownloader;
|
|
|
|
/** @var ProcessExecutor|null */
|
|
|
|
private $processExecutor;
|
|
|
|
/** @var Promise[][] */
|
|
|
|
private $currentPromises = array();
|
|
|
|
/** @var int */
|
|
|
|
private $waitIndex = 0;
|
|
|
|
|
|
|
|
public function __construct(HttpDownloader $httpDownloader, ProcessExecutor $processExecutor = null)
|
|
|
|
{
|
|
|
|
$this->httpDownloader = $httpDownloader;
|
|
|
|
$this->httpDownloader->enableAsync();
|
|
|
|
|
|
|
|
$this->processExecutor = $processExecutor;
|
|
|
|
if ($this->processExecutor) {
|
|
|
|
$this->processExecutor->enableAsync();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @return HttpDownloader
|
|
|
|
*/
|
|
|
|
public function getHttpDownloader()
|
|
|
|
{
|
|
|
|
return $this->httpDownloader;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* @return ProcessExecutor|null
|
|
|
|
*/
|
|
|
|
public function getProcessExecutor()
|
|
|
|
{
|
|
|
|
return $this->processExecutor;
|
|
|
|
}
|
|
|
|
|
|
|
|
public function wait(array $promises, ProgressBar $progress = null)
|
|
|
|
{
|
|
|
|
/** @var \Exception|null */
|
|
|
|
$uncaught = null;
|
|
|
|
|
|
|
|
\React\Promise\all($promises)->then(
|
|
|
|
function () {
|
|
|
|
},
|
|
|
|
function ($e) use (&$uncaught) {
|
|
|
|
$uncaught = $e;
|
|
|
|
}
|
|
|
|
);
|
|
|
|
|
|
|
|
// keep track of every group of promises that is waited on, so abortJobs can
|
|
|
|
// cancel them all, even if wait() was called within a wait()
|
|
|
|
$waitIndex = $this->waitIndex++;
|
|
|
|
$this->currentPromises[$waitIndex] = $promises;
|
|
|
|
|
|
|
|
if ($progress) {
|
|
|
|
$totalJobs = 0;
|
|
|
|
$totalJobs += $this->httpDownloader->countActiveJobs();
|
|
|
|
if ($this->processExecutor) {
|
|
|
|
$totalJobs += $this->processExecutor->countActiveJobs();
|
|
|
|
}
|
|
|
|
$progress->start($totalJobs);
|
|
|
|
}
|
|
|
|
|
|
|
|
$lastUpdate = 0;
|
|
|
|
while (true) {
|
|
|
|
$activeJobs = 0;
|
|
|
|
|
|
|
|
$activeJobs += $this->httpDownloader->countActiveJobs();
|
|
|
|
if ($this->processExecutor) {
|
|
|
|
$activeJobs += $this->processExecutor->countActiveJobs();
|
|
|
|
}
|
|
|
|
|
|
|
|
if ($progress && microtime(true) - $lastUpdate > 0.1) {
|
|
|
|
$lastUpdate = microtime(true);
|
|
|
|
$progress->setProgress($progress->getMaxSteps() - $activeJobs);
|
|
|
|
}
|
|
|
|
|
|
|
|
if (!$activeJobs) {
|
|
|
|
break;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
// as we skip progress updates if they are too quick, make sure we do one last one here at 100%
|
|
|
|
if ($progress) {
|
|
|
|
$progress->finish();
|
|
|
|
}
|
|
|
|
|
|
|
|
unset($this->currentPromises[$waitIndex]);
|
|
|
|
if ($uncaught) {
|
|
|
|
throw $uncaught;
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
public function abortJobs()
|
|
|
|
{
|
|
|
|
foreach ($this->currentPromises as $promiseGroup) {
|
|
|
|
foreach ($promiseGroup as $promise) {
|
|
|
|
$promise->cancel();
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|
|
|
|
}
|