<?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());
    }


}