Files
kiri-core/System/Process/CrontabProcess.php
T

133 lines
2.8 KiB
PHP
Raw Normal View History

2021-03-19 17:47:41 +08:00
<?php
namespace Snowflake\Process;
2021-03-19 18:52:30 +08:00
use Snowflake\Crontab;
2021-03-19 17:47:41 +08:00
use Swoole\Coroutine;
2021-03-20 03:25:30 +08:00
use Swoole\Coroutine\WaitGroup;
2021-03-19 17:47:41 +08:00
use Swoole\Coroutine\Channel;
use Swoole\Timer;
/**
* Class CrontabProcess
* @package Snowflake\Process
*/
class CrontabProcess extends Process
{
2021-03-20 03:15:58 +08:00
private Channel $channel;
2021-03-20 03:25:30 +08:00
private WaitGroup $waitGroup;
2021-03-20 03:15:58 +08:00
2021-03-20 02:33:50 +08:00
/** @var Crontab[] $names */
public array $names = [];
/**
* @param \Swoole\Process $process
*/
public function onHandler(\Swoole\Process $process): void
{
2021-03-20 03:15:58 +08:00
$this->channel = new Channel(1000);
2021-03-20 03:25:30 +08:00
Coroutine\go(function () {
$this->waitGroup();
2021-03-20 03:15:58 +08:00
});
2021-03-20 03:35:44 +08:00
$this->readByWorker($process);
2021-03-20 03:15:58 +08:00
}
2021-03-20 03:35:44 +08:00
/**
* @param $process
*/
private function readByWorker($process)
{
$this->channel->push($process->read());
Coroutine::sleep(0.01);
$this->readByWorker($process);
}
/**
* @throws \Exception
*/
2021-03-20 03:25:30 +08:00
private function waitGroup()
2021-03-20 03:15:58 +08:00
{
try {
$content = $this->channel->pop(-1);
$_content = json_decode($content, true);
if (is_null($_content)) {
$this->jobDelivery($content);
} else {
$this->otherAction($_content);
2021-03-20 02:33:50 +08:00
}
2021-03-20 03:15:58 +08:00
} catch (\Throwable $exception) {
$this->application->error($exception->getMessage());
2021-03-20 03:31:04 +08:00
} finally {
$this->waitGroup();
2021-03-20 02:33:50 +08:00
}
}
/**
* @param $content
*/
private function otherAction($content)
{
call_user_func(match ($content['action']) {
'clear' => function ($content) {
2021-03-20 02:56:21 +08:00
$this->clear($content['name']);
2021-03-20 02:33:50 +08:00
},
'clearAll' => function () {
2021-03-20 03:51:07 +08:00
$this->names = [];
Timer::clearAll();
2021-03-20 02:33:50 +08:00
},
default => function () {
$this->application->error('unknown action');
}
2021-03-20 02:44:30 +08:00
}, $content);
2021-03-20 02:33:50 +08:00
}
2021-03-20 02:56:21 +08:00
/**
* @param string $name
*/
public function clear(string $name)
{
if (!isset($this->names[$name])) {
return;
}
2021-03-20 03:51:07 +08:00
Timer::exists($this->names[$name]) && Timer::clear($this->names[$name]);
2021-03-20 03:43:30 +08:00
unset($this->names[$name]);
2021-03-20 02:56:21 +08:00
}
2021-03-20 02:33:50 +08:00
/**
* @param $content
*/
private function jobDelivery($content)
{
2021-03-20 02:51:37 +08:00
/** @var Crontab $content */
2021-03-20 02:33:50 +08:00
$content = unserialize($content);
2021-03-20 03:54:21 +08:00
$name = $content->getName();
$callback = function () use ($content) {
var_dump('executes');
2021-03-20 03:52:28 +08:00
$content->execute($this);
2021-03-20 03:54:21 +08:00
};
$runTicker = [$content->getTickTime() * 1000, $callback];
2021-03-20 03:39:11 +08:00
if ($content->isLoop()) {
2021-03-20 03:54:21 +08:00
$this->names[$name] = Timer::tick(...$runTicker);
2021-03-20 03:39:11 +08:00
} else {
2021-03-20 03:54:21 +08:00
$this->names[$name] = Timer::after(...$runTicker);
2021-03-20 03:39:11 +08:00
}
2021-03-20 02:33:50 +08:00
}
2021-03-19 17:47:41 +08:00
}