kiri-core/System/Crontab/Zookeeper.php

<?php


namespace Snowflake\Crontab;


use Exception;
use HttpServer\Server;
use Snowflake\Abstracts\Config;
use Snowflake\Cache\Redis;
use Snowflake\Process\Process;
use Snowflake\Snowflake;
use Swoole\Coroutine;
use Throwable;

/**
 * Class Zookeeper
 * @package Snowflake\Process
 */
class Zookeeper extends Process
{


    private int $workerNum = 0;


    private mixed $server;


    /**
     * @param \Swoole\Process $process
     */
    public function getProcessName(): string
    {
        $name = Config::get('id', 'system') . '[' . $this->pid . ']';
        if (!empty($prefix)) {
            $name .= '.Crontab zookeeper';
        }
        return $name;
    }


    /**
     * @param \Swoole\Process $process
     * @throws Exception
     */
    public function before(\Swoole\Process $process): void
    {
        /** @var Producer $crontab */
        $crontab = Snowflake::app()->get('crontab');
        $crontab->clearAll();

        $this->server = $server = Snowflake::app()->getSwoole();
        $this->workerNum = $server->setting['worker_num'] + $server->setting['task_worker_num'];
    }


    /**
     * @param \Swoole\Process $process
     * @throws Exception
     */
    public function onHandler(\Swoole\Process $process): void
    {
        $ticker = Config::get('crontab.ticker', 100) / 1000;
        $redis = Snowflake::app()->getRedis();
        while (true) {
            $range = $this->loadCarobTask($redis);
            foreach ($range as $value) {
                $this->dispatch($redis, $value);
            }
            Coroutine::sleep($ticker);
        }
    }


	/**
	 * @param Redis|\Redis $redis
	 * @param $value
	 * @throws Exception
	 */
    private function dispatch(Redis|\Redis $redis, $value)
    {
        try {
            $params['action'] = 'crontab';
            if (empty($handler = $redis->get('crontab:' . $value))) {
                return;
            }
            $params['handler'] = unserialize($handler);

            $this->server->sendMessage($params, $this->getWorker());
        } catch (Throwable $exception) {
            logger()->addError($exception);
        }
    }


    /**
     * @return int
     * @throws \Exception
     */
    private function getWorker(): int
    {
        return random_int(0, $this->workerNum - 1);
    }


    /**
     * @return array
     * @throws Exception
     */
    private function loadCarobTask(Redis|\Redis $redis): array
    {
        $range = $redis->zRangeByScore(Producer::CRONTAB_KEY, '0', (string)time());

        $redis->zRem(Producer::CRONTAB_KEY, ...$range);

        return $range;
    }

}