106 lines
2.5 KiB
PHP
106 lines
2.5 KiB
PHP
<?php
|
|
|
|
namespace Kiri\Server\Processes;
|
|
|
|
use Exception;
|
|
use Kiri;
|
|
use Swoole\Process;
|
|
use const SIGKILL;
|
|
use const SIGTERM;
|
|
|
|
trait TraitProcess
|
|
{
|
|
|
|
/**
|
|
* @var array
|
|
*/
|
|
private array $_process = [];
|
|
|
|
|
|
/**
|
|
* @param string|array|AbstractProcess $class
|
|
* @return void
|
|
* @throws
|
|
*/
|
|
public function addProcess(string|array|AbstractProcess $class): void
|
|
{
|
|
if (!is_array($class)) $class = [$class];
|
|
foreach ($class as $name) {
|
|
if (is_string($name)) $name = Kiri::getDi()->get($name);
|
|
if (isset($this->_process[$name->getName()])) {
|
|
throw new Exception('AbstractProcess(' . $name->getName() . ') is exists.');
|
|
}
|
|
$this->_process[$name->getName()] = $this->genProcess($name);
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* @param AbstractProcess $name
|
|
* @return Process
|
|
*/
|
|
private function genProcess(AbstractProcess $name): Process
|
|
{
|
|
return new Process(function (Process $process) use ($name) {
|
|
$process->name('[' . \config('site.id', 'system-service') . '].' . $name->getName() . '[' . $process->pid . ']');
|
|
$name->onShutdown($process)->process($process);
|
|
},
|
|
$name->getRedirectStdinAndStdout(),
|
|
$name->getPipeType(),
|
|
$name->isEnableCoroutine());
|
|
}
|
|
|
|
|
|
/**
|
|
* @param string $name
|
|
* @return AbstractProcess|null
|
|
*/
|
|
public function getProcess(string $name): ?Process
|
|
{
|
|
return $this->_process[$name] ?? null;
|
|
}
|
|
|
|
|
|
/**
|
|
* @return array
|
|
*/
|
|
public function getProcesses(): array
|
|
{
|
|
return $this->_process;
|
|
}
|
|
|
|
/**
|
|
* @return void
|
|
*/
|
|
private function terminateProcesses(): void
|
|
{
|
|
$pids = [];
|
|
foreach ($this->_process as $process) {
|
|
$pid = (int)($process->pid ?? 0);
|
|
if ($pid > 1 && $pid !== getmypid()) {
|
|
$pids[] = $pid;
|
|
}
|
|
}
|
|
|
|
$pids = array_values(array_unique($pids));
|
|
foreach ($pids as $pid) {
|
|
if (@Process::kill($pid, 0)) {
|
|
@Process::kill($pid, SIGTERM);
|
|
}
|
|
}
|
|
|
|
for ($i = 0; $i < 10; $i++) {
|
|
$alive = array_values(array_filter($pids, static fn(int $pid): bool => @Process::kill($pid, 0)));
|
|
if ($alive === []) {
|
|
return;
|
|
}
|
|
usleep(100000);
|
|
}
|
|
|
|
foreach ($pids as $pid) {
|
|
if (@Process::kill($pid, 0)) {
|
|
@Process::kill($pid, SIGKILL);
|
|
}
|
|
}
|
|
}
|
|
} |