as2252258@163.com
134 lines
2.9 KiB
PHP
134 lines
2.9 KiB
PHP
<?php
|
|
|
|
|
|
namespace Snowflake\Process;
|
|
|
|
|
|
use Snowflake\Crontab;
|
|
use Swoole\Coroutine;
|
|
use Swoole\Coroutine\WaitGroup;
|
|
use Swoole\Coroutine\Channel;
|
|
use Swoole\Timer;
|
|
|
|
/**
|
|
* Class CrontabProcess
|
|
* @package Snowflake\Process
|
|
*/
|
|
class CrontabProcess extends Process
|
|
{
|
|
|
|
|
|
private Channel $channel;
|
|
private WaitGroup $waitGroup;
|
|
|
|
|
|
/** @var Crontab[] $names */
|
|
public array $names = [];
|
|
|
|
|
|
/**
|
|
* @param \Swoole\Process $process
|
|
*/
|
|
public function onHandler(\Swoole\Process $process): void
|
|
{
|
|
$this->channel = new Channel(1000);
|
|
Coroutine\go(function () {
|
|
$this->waitGroup();
|
|
});
|
|
$this->readByWorker($process);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $process
|
|
*/
|
|
private function readByWorker($process)
|
|
{
|
|
$this->channel->push($process->read());
|
|
|
|
Coroutine::sleep(0.01);
|
|
|
|
$this->readByWorker($process);
|
|
}
|
|
|
|
|
|
/**
|
|
* @throws \Exception
|
|
*/
|
|
private function waitGroup()
|
|
{
|
|
try {
|
|
$content = $this->channel->pop(-1);
|
|
|
|
$_content = json_decode($content, true);
|
|
if (is_null($_content)) {
|
|
$this->jobDelivery($content);
|
|
} else {
|
|
$this->otherAction($_content);
|
|
}
|
|
} catch (\Throwable $exception) {
|
|
$this->application->error($exception->getMessage());
|
|
} finally {
|
|
$this->waitGroup();
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $content
|
|
*/
|
|
private function otherAction($content)
|
|
{
|
|
call_user_func(match ($content['action']) {
|
|
'clear' => function ($content) {
|
|
$this->clear($content['name']);
|
|
},
|
|
'clearAll' => function () {
|
|
foreach ($this->names as $name => $crontab) {
|
|
$crontab->clearTimer();
|
|
|
|
unset($this->names[$name], $crontab);
|
|
}
|
|
},
|
|
default => function () {
|
|
$this->application->error('unknown action');
|
|
}
|
|
}, $content);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param string $name
|
|
*/
|
|
public function clear(string $name)
|
|
{
|
|
if (!isset($this->names[$name])) {
|
|
return;
|
|
}
|
|
$this->names[$name]->clearTimer();
|
|
}
|
|
|
|
|
|
/**
|
|
* @param $content
|
|
*/
|
|
private function jobDelivery($content)
|
|
{
|
|
/** @var Crontab $content */
|
|
$content = unserialize($content);
|
|
$runTicker = function () use ($content) {
|
|
$this->application->warning('execute crontab ' . date('Y-m-d H:i:s'));
|
|
$content->execute($this);
|
|
};
|
|
$timer = $content->getTickTime() * 10;
|
|
if ($content->isLoop()) {
|
|
$content->setTimerId(Timer::tick($timer, $runTicker));
|
|
} else {
|
|
$content->setTimerId(Timer::after($timer, $runTicker));
|
|
}
|
|
$this->names[$content->getName()] = $content;
|
|
}
|
|
|
|
|
|
}
|