zhamao-framework/src/ZM/Event/SwooleEvent/OnMessage.php

<?php

declare(strict_types=1);

namespace ZM\Event\SwooleEvent;

use Swoole\Coroutine;
use Swoole\WebSocket\Frame;
use Throwable;
use ZM\Annotation\Swoole\OnMessageEvent;
use ZM\Annotation\Swoole\OnSwooleEvent;
use ZM\Annotation\Swoole\SwooleHandler;
use ZM\ConnectionManager\ConnectionObject;
use ZM\ConnectionManager\ManagerGM;
use ZM\Console\Console;
use ZM\Console\TermColor;
use ZM\Container\Container;
use ZM\Context\Context;
use ZM\Context\ContextInterface;
use ZM\Event\EventDispatcher;
use ZM\Event\SwooleEvent;

/**
 * Class OnMessage
 * @SwooleHandler("message")
 */
class OnMessage implements SwooleEvent
{
    public function onCall($server, Frame $frame)
    {
        Console::debug('Calling Swoole "message" from fd=' . $frame->fd . ': ' . TermColor::ITALIC . $frame->data . TermColor::RESET);
        unset(Context::$context[Coroutine::getCid()]);
        $conn = ManagerGM::get($frame->fd);
        set_coroutine_params(['server' => $server, 'frame' => $frame, 'connection' => $conn]);

        $this->registerRequestContainerBindings($frame, $conn);

        $dispatcher1 = new EventDispatcher(OnMessageEvent::class);
        $dispatcher1->setRuleFunction(function ($v) use ($conn) {
            return $v->connect_type === $conn->getName() && ($v->getRule() === '' || eval('return ' . $v->getRule() . ';'));
        });

        $dispatcher = new EventDispatcher(OnSwooleEvent::class);
        $dispatcher->setRuleFunction(function ($v) {
            if ($v->getRule() == '') {
                return strtolower($v->type) == 'message';
            }
            /* @noinspection PhpUnreachableStatementInspection
             * @noinspection RedundantSuppression
             */
            if (strtolower($v->type) == 'message' && eval('return ' . $v->getRule() . ';')) {
                return true;
            }
            return false;
        });
        try {
            // $starttime = microtime(true);
            $dispatcher1->dispatchEvents($conn);
            $dispatcher->dispatchEvents($conn);
            // Console::success("Used ".round((microtime(true) - $starttime) * 1000, 3)." ms!");
        } catch (Throwable $e) {
            $error_msg = $e->getMessage() . ' at ' . $e->getFile() . '(' . $e->getLine() . ')';
            Console::error(zm_internal_errcode('E00017') . 'Uncaught ' . get_class($e) . ' when calling "message": ' . $error_msg);
            Console::trace();
        } finally {
            container()->flush();
        }
    }

    /**
     * 注册请求容器绑定
     */
    private function registerRequestContainerBindings(Frame $frame, ?ConnectionObject $conn): void
    {
        $container = Container::getInstance();
        $container->setLogPrefix("[Container#{$frame->fd}]");
        $container->instance(Frame::class, $frame);
        $container->instance(ConnectionObject::class, $conn);
        $container->bind(ContextInterface::class, function () {
            return ctx();
        });
        $container->alias(ContextInterface::class, Context::class);
    }
}