154 lines
2.5 KiB
PHP
154 lines
2.5 KiB
PHP
<?php
|
|
|
|
|
|
namespace HttpServer;
|
|
|
|
|
|
use Exception;
|
|
use Snowflake\Snowflake;
|
|
use Swoole\WebSocket\Server;
|
|
|
|
/**
|
|
* Class Action
|
|
* @package HttpServer
|
|
*/
|
|
trait Action
|
|
{
|
|
|
|
/**
|
|
* @param $argv
|
|
* @return Server
|
|
* @throws
|
|
*/
|
|
public function getSwooleServer($argv)
|
|
{
|
|
/** @var \HttpServer\Server $socket */
|
|
$socket = Snowflake::app()->get('server');
|
|
// if (isset($argv[2])) {
|
|
// $this->modify($argv, $socket);
|
|
// }
|
|
|
|
if (!isset($argv[1])) $argv[1] = 'start';
|
|
|
|
return $this->checkAction($argv, $socket);
|
|
}
|
|
|
|
/**
|
|
* @param $argv
|
|
* @param $socket
|
|
* @return Server
|
|
* @throws Exception
|
|
*/
|
|
private function checkAction($argv, $socket)
|
|
{
|
|
if (!in_array($argv[1], ['stop', 'start', 'restart'])) {
|
|
exit($this->error('action not exists.'));
|
|
}
|
|
return $this->{$argv[1]}($socket);
|
|
}
|
|
|
|
/**
|
|
* @param \HttpServer\Server $socket
|
|
* @return mixed
|
|
* @throws Exception
|
|
*/
|
|
public function restart($socket)
|
|
{
|
|
$this->_shutdown($socket);
|
|
|
|
return $this->start();
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
* @param \HttpServer\Server $socket
|
|
* @throws Exception
|
|
*/
|
|
public function stop($socket)
|
|
{
|
|
$this->_shutdown($socket);
|
|
}
|
|
|
|
/**
|
|
* @param $server
|
|
* @return void
|
|
* @throws Exception
|
|
*/
|
|
private function _shutdown($server)
|
|
{
|
|
$socket = storage('socket.sock');
|
|
if (!file_exists($socket)) {
|
|
$this->close($server);
|
|
} else {
|
|
$pathId = file_get_contents($socket);
|
|
@unlink($socket);
|
|
|
|
if (empty($pathId)) {
|
|
$this->close($server);
|
|
} else {
|
|
exec("kill -TERM $pathId");
|
|
$this->close($server);
|
|
}
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @param \HttpServer\Server $server
|
|
* @return void
|
|
* @throws Exception
|
|
*/
|
|
private function close($server)
|
|
{
|
|
echo 'waite.';
|
|
while ($server->isRunner()) {
|
|
echo '.';
|
|
$pods = glob(storage('workerIds') . '/*');
|
|
if (count($pods) < 1) {
|
|
break;
|
|
}
|
|
foreach ($pods as $pid) {
|
|
if (!file_exists($pid)) {
|
|
continue;
|
|
}
|
|
$content = file_get_contents($pid);
|
|
exec("ps -ax | awk '{ print $1 }' | grep -e '^{$content}$'", $output);
|
|
if (count($output) > 0) {
|
|
$this->closeByPid($content);
|
|
} else {
|
|
file_exists($pid) && @unlink($pid);
|
|
}
|
|
}
|
|
usleep(100);
|
|
}
|
|
echo PHP_EOL;
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $port
|
|
* @return bool|array
|
|
*/
|
|
private function isUse($port)
|
|
{
|
|
if (empty($port)) {
|
|
return false;
|
|
}
|
|
exec('netstat -tunlp tcp | grep ' . $port, $output);
|
|
if (empty($output)) {
|
|
return false;
|
|
}
|
|
return $output;
|
|
}
|
|
|
|
/**
|
|
* @param $pid
|
|
*/
|
|
private function closeByPid($pid)
|
|
{
|
|
shell_exec('kill -TERM ' . $pid);
|
|
}
|
|
|
|
|
|
}
|