as2252258@163.com
127 lines
3.0 KiB
PHP
127 lines
3.0 KiB
PHP
<?php
|
|
|
|
|
|
namespace Snowflake\Process;
|
|
|
|
|
|
use Snowflake\Crontab;
|
|
use Snowflake\Abstracts\Crontab as ACrontab;
|
|
use Snowflake\Event;
|
|
use Snowflake\Snowflake;
|
|
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 = [];
|
|
|
|
|
|
public array $scores = [];
|
|
public array $timers = [];
|
|
|
|
|
|
/**
|
|
* @param \Swoole\Process $process
|
|
*/
|
|
public function onHandler(\Swoole\Process $process): void
|
|
{
|
|
$redis = Snowflake::app()->getRedis();
|
|
$redis->del(ACrontab::CRONTAB_KEY);
|
|
$redis->release();
|
|
|
|
Timer::tick(1000, function () {
|
|
$startTime = time();
|
|
|
|
$redis = Snowflake::app()->getRedis();
|
|
|
|
$range = $redis->zRangeByScore(ACrontab::CRONTAB_KEY, '0', (string)$startTime);
|
|
var_dump($range);
|
|
$redis->zRemRangeByScore(ACrontab::CRONTAB_KEY, '0', (string)$startTime);
|
|
foreach ($range as $value) {
|
|
$crontab = $redis->get('crontab:' . md5($value));
|
|
var_dump($value, $crontab);
|
|
if (empty($crontab) || !($crontab = unserialize($crontab))) {
|
|
continue;
|
|
}
|
|
Coroutine::create(function (Crontab $value, int $startTime) {
|
|
$this->dispatch($value);
|
|
}, $crontab, $startTime);
|
|
}
|
|
$redis->release();
|
|
});
|
|
}
|
|
|
|
|
|
/**
|
|
* @param Crontab $value
|
|
* @param int $startTime
|
|
* @throws \Exception
|
|
*/
|
|
private function dispatch(Crontab $value)
|
|
{
|
|
try {
|
|
$value->increment()->execute();
|
|
if ($value->getExecuteNumber() < $value->getMaxExecuteNumber()) {
|
|
$this->addTask($value);
|
|
} else if ($value->isLoop()) {
|
|
$this->addTask($value);
|
|
}
|
|
} catch (\Throwable $exception) {
|
|
$this->application->error($exception->getMessage());
|
|
} finally {
|
|
fire(Event::SYSTEM_RESOURCE_RELEASES);
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
* @param string $name
|
|
*/
|
|
public function clear(string $name)
|
|
{
|
|
if (!isset($this->names[$name])) {
|
|
return;
|
|
}
|
|
$timers = $this->timers[$name];
|
|
|
|
$search = array_search($name, $this->scores[$timers]);
|
|
if ($search !== false) {
|
|
unset($this->scores[$timers][$search]);
|
|
}
|
|
unset($this->timers[$name], $this->names[$name]);
|
|
}
|
|
|
|
|
|
/**
|
|
* @param Crontab $content
|
|
* @param $ticker
|
|
*/
|
|
private function addTask(Crontab $crontab)
|
|
{
|
|
$redis = Snowflake::app()->getRedis();
|
|
|
|
$name = md5($crontab->getName());
|
|
|
|
$redis->set('crontab:' . $name, serialize($crontab));
|
|
|
|
$tickTime = time() + $crontab->getTickTime();
|
|
|
|
$redis->zAdd(ACrontab::CRONTAB_KEY, $tickTime, $crontab->getName());
|
|
}
|
|
|
|
|
|
}
|