zhamao-framework/src/ZM/Event/EventManager.php

<?php

declare(strict_types=1);

namespace ZM\Event;

use Closure;
use Error;
use Exception;
use Swoole\Timer;
use ZM\Annotation\AnnotationBase;
use ZM\Annotation\AnnotationParser;
use ZM\Annotation\Swoole\OnTick;
use ZM\Config\ZMConfig;
use ZM\Console\Console;
use ZM\Exception\AnnotationException;
use ZM\Store\LightCache;
use ZM\Store\ZMAtomic;

class EventManager
{
    public static $events = [];

    public static $middleware_map = [];

    public static $event_map = [];

    public static $middlewares = [];

    public static $req_mapping = [];

    public static function addEvent($event_name, ?AnnotationBase $event_obj)
    {
        if ($event_obj->method instanceof Closure) {
            logger()->debug("Adding event {$event_name} at @Anonymous");
        } else {
            logger()->debug("Adding event {$event_name} at " . ($event_obj->class) . ':' . ($event_obj->method));
            self::$event_map[$event_obj->class][$event_obj->method][] = $event_obj;
        }
        self::$events[$event_name][] = $event_obj;
        (new AnnotationParser())->sortByLevel(self::$events, $event_name);
    }

    /**
     * @throws AnnotationException
     */
    public static function loadEventByParser(AnnotationParser $parser)
    {
        self::$events = array_merge(self::$events, $parser->generateAnnotationEvents());
        self::$middlewares = $parser->getMiddlewares();
        self::$middleware_map = $parser->getMiddlewareMap();
        self::$req_mapping = $parser->getReqMapping();
        $parser->verifyMiddlewares();
    }

    /**
     * 注册所有计时器给每个进程
     * @throws Exception
     */
    public static function registerTimerTick()
    {
        $dispatcher = new EventDispatcher(OnTick::class);
        foreach (self::$events[OnTick::class] ?? [] as $vss) {
            if (server()->worker_id !== $vss->worker_id && $vss->worker_id != -1) {
                return;
            }
            // echo server()->worker_id.PHP_EOL;
            $plain_class = $vss->class;
            logger()->debug('Added Middleware-based timer: ' . $plain_class . ' -> ' . $vss->method);
            Timer::tick($vss->tick_ms, function () use ($vss, $dispatcher) {
                set_coroutine_params([]);
                if (ZMAtomic::get('stop_signal')->get() != 0) {
                    Timer::clearAll();
                    return;
                }
                try {
                    $dispatcher->dispatchEvent($vss, null);
                } catch (Exception $e) {
                    Console::error(zm_internal_errcode('E00034') . 'Uncaught error from TimerTick: ' . $e->getMessage() . ' at ' . $e->getFile() . "({$e->getLine()})");
                } catch (Error $e) {
                    Console::error(zm_internal_errcode('E00034') . 'Uncaught fatal error from TimerTick: ' . $e->getMessage());
                    echo Console::setColor($e->getTraceAsString(), 'gray');
                    Console::error('Please check your code!');
                }
            });
        }
        $conf = ZMConfig::get('global', 'worker_cache') ?? ['worker' => 0];
        if (server()->worker_id == $conf['worker']) {
            zm_timer_tick(ZMConfig::get('global', 'light_cache')['auto_save_interval'] * 1000, static function () {
                LightCache::savePersistence();
            });
        }
    }
}