as2252258@163.com
130 lines
3.0 KiB
PHP
130 lines
3.0 KiB
PHP
<?php
|
|
|
|
|
|
namespace Snowflake\Crontab;
|
|
|
|
|
|
use Exception;
|
|
use ReflectionException;
|
|
use Snowflake\Event;
|
|
use Snowflake\Exception\ComponentException;
|
|
use Snowflake\Exception\ConfigException;
|
|
use Snowflake\Exception\NotFindClassException;
|
|
use Snowflake\Process\Process;
|
|
use Snowflake\Snowflake;
|
|
use Swoole\Coroutine;
|
|
|
|
|
|
/**
|
|
* Class Consumer
|
|
* @package Snowflake\Crontab
|
|
*/
|
|
class Consumer extends Process
|
|
{
|
|
|
|
public Coroutine\Channel $channel;
|
|
|
|
|
|
/**
|
|
* @param \Swoole\Process $process
|
|
* @throws Exception
|
|
*/
|
|
public function onHandler(\Swoole\Process $process): void
|
|
{
|
|
if (Snowflake::getPlatform()->isLinux()) {
|
|
name($this->pid, 'Crontab consumer');
|
|
}
|
|
|
|
$this->channel = new Coroutine\Channel(2000);
|
|
go(function () {
|
|
$this->popChannel();
|
|
});
|
|
|
|
$this->recovery();
|
|
$this->tick($process);
|
|
}
|
|
|
|
|
|
private function recovery()
|
|
{
|
|
$redis = redis()->hGetAll('crontab:wait:execute');
|
|
foreach ($redis as $redi) {
|
|
$this->channel->push($redi);
|
|
}
|
|
fire(Event::SYSTEM_RESOURCE_RELEASES);
|
|
}
|
|
|
|
|
|
/**
|
|
* @throws Exception
|
|
*/
|
|
public function popChannel()
|
|
{
|
|
/** @var Crontab $crontab */
|
|
$crontab = $this->channel->pop(-1);
|
|
go(function () use ($crontab) {
|
|
try {
|
|
$crontab->increment()->execute();
|
|
if ($crontab->getExecuteNumber() < $crontab->getMaxExecuteNumber()) {
|
|
Consumer::addTask($crontab);
|
|
} else if ($crontab->isLoop()) {
|
|
Consumer::addTask($crontab);
|
|
}
|
|
} catch (\Throwable $throwable) {
|
|
logger()->addError($throwable, 'throwable');
|
|
} finally {
|
|
fire(Event::SYSTEM_RESOURCE_RELEASES);
|
|
}
|
|
});
|
|
$this->popChannel();
|
|
}
|
|
|
|
|
|
/**
|
|
* @param \Swoole\Process $process
|
|
* @throws ReflectionException
|
|
* @throws ComponentException
|
|
* @throws ConfigException
|
|
* @throws NotFindClassException
|
|
* @throws Exception
|
|
*/
|
|
public function tick(\Swoole\Process $process)
|
|
{
|
|
$value = $process->read(40);
|
|
|
|
$redis = Snowflake::app()->getRedis();
|
|
|
|
/** @var Crontab $crontab */
|
|
$crontab = swoole_unserialize($redis->get($value));
|
|
$redis->hSet('crontab:wait:execute', $crontab->getName(), $redis->get($value));
|
|
$redis->del($value);
|
|
if (is_object($crontab)) {
|
|
$this->channel->push($crontab);
|
|
}
|
|
|
|
$redis->release();
|
|
|
|
$this->tick($process);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param Crontab $crontab
|
|
* @throws Exception
|
|
*/
|
|
private static function addTask(Crontab $crontab)
|
|
{
|
|
$redis = Snowflake::app()->getRedis();
|
|
|
|
$name = md5($crontab->getName());
|
|
|
|
$redis->set('crontab:' . $name, swoole_serialize($crontab));
|
|
|
|
$tickTime = time() + $crontab->getTickTime();
|
|
|
|
$redis->zAdd(Producer::CRONTAB_KEY, $tickTime, $crontab->getName());
|
|
}
|
|
|
|
|
|
}
|