mirror of
https://gitee.com/hyperf/hyperf.git
synced 2024-11-30 10:47:44 +08:00
140 lines
4.0 KiB
PHP
Executable File
140 lines
4.0 KiB
PHP
Executable File
#!/usr/bin/env php
|
|
<?php
|
|
/**
|
|
* This file is part of Hyperf.
|
|
*
|
|
* @link https://www.hyperf.io
|
|
* @document https://hyperf.wiki
|
|
* @contact group@hyperf.io
|
|
* @license https://github.com/hyperf/hyperf/blob/master/LICENSE
|
|
*/
|
|
use PhpParser\Node;
|
|
use PhpParser\NodeTraverser;
|
|
use PhpParser\NodeVisitorAbstract;
|
|
use PhpParser\Parser;
|
|
use PhpParser\ParserFactory;
|
|
use PhpParser\PrettyPrinter\Standard;
|
|
|
|
require_once 'vendor/autoload.php';
|
|
|
|
class RunInCoroutineVisitor extends NodeVisitorAbstract
|
|
{
|
|
/**
|
|
* @var Parser
|
|
*/
|
|
protected $parser;
|
|
|
|
public function __construct(Parser $parser)
|
|
{
|
|
$this->parser = $parser;
|
|
}
|
|
|
|
public function afterTraverse(array $nodes)
|
|
{
|
|
$main = array_pop($nodes);
|
|
/** @var Node\Stmt\Expression $main */
|
|
$main = $main->expr;
|
|
$main = $main->expr;
|
|
|
|
$require = array_pop($nodes);
|
|
|
|
$nodes[] = $this->getPrependSetter()[0];
|
|
$nodes[] = $require;
|
|
|
|
$nodes[] = new Node\Stmt\Expression(new Node\Expr\Assign(
|
|
new Node\Expr\Variable('code'),
|
|
new Node\Scalar\LNumber(0)
|
|
));
|
|
|
|
$nodes[] = $this->getCoroutineSetter()[0];
|
|
|
|
$nodes[] = new Node\Stmt\Expression(
|
|
new Node\Expr\FuncCall(
|
|
new Node\Name('Swoole\Coroutine\run'),
|
|
[
|
|
new Node\Arg(new Node\Expr\Closure([
|
|
'stmts' => [
|
|
new Node\Stmt\Expression(new Node\Expr\Assign(
|
|
new Node\Expr\Variable('code'),
|
|
$main
|
|
)),
|
|
$this->getTimerClearAll()[0],
|
|
$this->getCoordinatorResume()[0],
|
|
],
|
|
'uses' => [
|
|
new Node\Expr\ClosureUse(new Node\Expr\Variable('code'), true),
|
|
],
|
|
])),
|
|
]
|
|
)
|
|
);
|
|
$nodes[] = new Node\Stmt\Expression(new Node\Expr\Exit_(new Node\Expr\Variable('code')));
|
|
return $nodes;
|
|
}
|
|
|
|
protected function getPrependSetter(): array
|
|
{
|
|
return $this->parser->parse("<?php
|
|
if (!isset(getopt('', ['prepend:'])['prepend'])) {
|
|
(function () {
|
|
\$prepend = null;
|
|
foreach (\$_SERVER['argv'] as \$index => \$argv) {
|
|
// --prepend /path/to/file
|
|
if (\$argv === '--prepend') {
|
|
\$prepend = \$_SERVER['argv'][\$index + 1] ?? null;
|
|
break;
|
|
}
|
|
// --prepend=/path/to/file
|
|
if (strpos(\$argv, '--prepend=') === 0) {
|
|
\$prepend = substr(\$argv, 10);
|
|
break;
|
|
}
|
|
}
|
|
if (\$prepend !== null && file_exists(\$prepend)) {
|
|
require \$prepend;
|
|
}
|
|
})();
|
|
}
|
|
");
|
|
}
|
|
|
|
protected function getCoroutineSetter(): array
|
|
{
|
|
return $this->parser->parse("<?php
|
|
Swoole\\Coroutine::set([
|
|
'hook_flags' => SWOOLE_HOOK_ALL,
|
|
'exit_condition' => function() {
|
|
return Swoole\\Coroutine::stats()['coroutine_num'] === 0;
|
|
}
|
|
]);");
|
|
}
|
|
|
|
protected function getTimerClearAll(): array
|
|
{
|
|
return $this->parser->parse('<?php Swoole\\Timer::clearAll();');
|
|
}
|
|
|
|
protected function getCoordinatorResume(): array
|
|
{
|
|
return $this->parser->parse('<?php Hyperf\\Coordinator\\CoordinatorManager::until(Hyperf\\Coordinator\\Constants::WORKER_EXIT)->resume();');
|
|
}
|
|
}
|
|
|
|
$parser = (new ParserFactory())->create(ParserFactory::ONLY_PHP7);
|
|
$printer = new Standard();
|
|
$traverser = new NodeTraverser();
|
|
|
|
$code = file_get_contents(__DIR__ . '/../vendor/phpunit/phpunit/phpunit');
|
|
|
|
$stmts = $parser->parse($code);
|
|
$traverser->addVisitor(new RunInCoroutineVisitor($parser));
|
|
$stmts = $traverser->traverse($stmts);
|
|
|
|
$code = $printer->prettyPrint($stmts);
|
|
|
|
// TODO: Unknown reason
|
|
$code = ltrim($code, '?>' . PHP_EOL);
|
|
|
|
file_put_contents(__DIR__ . '/co-phpunit', $code);
|
|
file_put_contents(__DIR__ . '/../src/testing/co-phpunit', $code);
|