mirror of
https://github.com/zhamao-robot/zhamao-framework.git
synced 2026-07-15 04:35:34 +08:00
103 lines
3.9 KiB
PHP
103 lines
3.9 KiB
PHP
<?php
|
|
|
|
declare(strict_types=1);
|
|
|
|
namespace ZM\Command;
|
|
|
|
use Swoole\Atomic;
|
|
use Swoole\Http\Request;
|
|
use Swoole\Http\Response;
|
|
use Swoole\Http\Server;
|
|
use Swoole\Process;
|
|
use Symfony\Component\Console\Command\Command;
|
|
use Symfony\Component\Console\Input\InputArgument;
|
|
use Symfony\Component\Console\Input\InputInterface;
|
|
use Symfony\Component\Console\Input\InputOption;
|
|
use Symfony\Component\Console\Output\OutputInterface;
|
|
use ZM\Config\ZMConfig;
|
|
use ZM\Console\Console;
|
|
use ZM\Logger\TablePrinter;
|
|
use ZM\Store\ZMAtomic;
|
|
use ZM\Utils\DataProvider;
|
|
use ZM\Utils\HttpUtil;
|
|
|
|
class PureHttpCommand extends Command
|
|
{
|
|
// the name of the command (the part after "bin/console")
|
|
protected static $defaultName = 'simple-http-server';
|
|
|
|
protected function configure()
|
|
{
|
|
$this->setDescription('Run a simple http server | 启动一个简单的文件 HTTP 服务器');
|
|
$this->setHelp('直接运行可以启动');
|
|
$this->addArgument('dir', InputArgument::REQUIRED, 'Your directory');
|
|
$this->addOption('host', 'H', InputOption::VALUE_REQUIRED, '启动监听地址');
|
|
$this->addOption('port', 'P', InputOption::VALUE_REQUIRED, '启动监听地址的端口');
|
|
// ...
|
|
}
|
|
|
|
protected function execute(InputInterface $input, OutputInterface $output): int
|
|
{
|
|
$tty_width = explode(' ', trim(exec('stty size')))[1];
|
|
if (realpath($input->getArgument('dir') ?? '.') === false) {
|
|
$output->writeln('<error>Directory error(' . ($input->getArgument('dir') ?? '.') . '): no such file or directory.</error>');
|
|
return self::FAILURE;
|
|
}
|
|
ZMConfig::setDirectory(DataProvider::getSourceRootDir() . '/config');
|
|
$global = ZMConfig::get('global');
|
|
$host = $input->getOption('host') ?? $global['host'];
|
|
$port = $input->getOption('port') ?? $global['port'];
|
|
|
|
$index = ['index.html', 'index.htm'];
|
|
$out = [
|
|
'listen' => $host . ':' . $port,
|
|
'version' => ZM_VERSION,
|
|
'web_root' => realpath($input->getArgument('dir') ?? '.'),
|
|
'index' => implode(',', $index),
|
|
];
|
|
$printer = new TablePrinter($out);
|
|
$printer->printAll();
|
|
$server = new Server($host, $port);
|
|
$server->set(ZMConfig::get('global', 'swoole'));
|
|
Console::init(2, $server);
|
|
ZMAtomic::$atomics['request'] = [];
|
|
for ($i = 0; $i < 32; ++$i) {
|
|
ZMAtomic::$atomics['request'][$i] = new Atomic(0);
|
|
}
|
|
$server->on('request', function (Request $request, Response $response) use ($input, $index, $server) {
|
|
ZMAtomic::$atomics['request'][$server->worker_id]->add(1);
|
|
HttpUtil::handleStaticPage(
|
|
$request->server['request_uri'],
|
|
$response,
|
|
[
|
|
'document_root' => realpath($input->getArgument('dir') ?? '.'),
|
|
'document_index' => $index,
|
|
]
|
|
);
|
|
// echo "\r" . Coroutine::stats()["coroutine_peak_num"];
|
|
});
|
|
$server->on('start', function ($server) {
|
|
Process::signal(SIGINT, function () use ($server) {
|
|
echo "\r";
|
|
Console::warning('Server interrupted by keyboard.');
|
|
for ($i = 0; $i < 32; ++$i) {
|
|
$num = ZMAtomic::$atomics['request'][$i]->get();
|
|
if ($num != 0) {
|
|
echo "[{$i}]: " . $num . "\n";
|
|
}
|
|
}
|
|
$server->shutdown();
|
|
$server->stop();
|
|
});
|
|
Console::success('Server started. Use Ctrl+C to stop.');
|
|
});
|
|
$server->start();
|
|
// return this if there was no problem running the command
|
|
// (it's equivalent to returning int(0))
|
|
return 0;
|
|
// or return this if some error happened during the execution
|
|
// (it's equivalent to returning int(1))
|
|
// return 1;
|
|
}
|
|
}
|