MessageListener.php

<?php declare(strict_types=1);

namespace Swoft\WebSocket\Server\Swoole;

use Swoft;
use Swoft\Bean\Annotation\Mapping\Bean;
use Swoft\Bean\BeanFactory;
use Swoft\Context\Context;
use Swoft\Server\Contract\MessageInterface;
use Swoft\Session\Session;
use Swoft\SwoftEvent;
use Swoft\WebSocket\Server\Context\WsMessageContext;
use Swoft\WebSocket\Server\Message\Request;
use Swoft\WebSocket\Server\Message\Response;
use Swoft\WebSocket\Server\WsErrorDispatcher;
use Swoft\WebSocket\Server\WsMessageDispatcher;
use Swoft\WebSocket\Server\WsServerEvent;
use Swoole\Websocket\Frame;
use Swoole\Websocket\Server;
use Throwable;
use function server;

/**
 * Class MessageListener
 *
 * @Bean()
 *
 * @since 2.0
 */
class MessageListener implements MessageInterface
{
    /**
     * @param Server $server
     * @param Frame  $frame
     *
     * @throws Throwable
     */
    public function onMessage(Server $server, Frame $frame): void
    {
        $fd  = $frame->fd;
        $sid = (string)$fd;

        server()->log("Message: conn#{$fd} received message: {$frame->data}", [], 'debug');

        $request  = Request::new($frame);
        $response = Response::new($fd);

        /** @var WsMessageContext $ctx */
        $ctx = WsMessageContext::new($request, $response);

        // Storage context
        Context::set($ctx);
        // Bind cid => sid(fd)
        Session::bindCo($sid);

        /** @var WsMessageDispatcher $dispatcher */
        $dispatcher = BeanFactory::getSingleton('wsMsgDispatcher');

        try {
            // Trigger message before event
            Swoft::trigger(WsServerEvent::MESSAGE_RECEIVE, $fd, $server, $frame);

            // Parse and dispatch message
            $dispatcher->dispatch($server, $request, $response);

            // Trigger message after event
            Swoft::trigger(WsServerEvent::MESSAGE_AFTER, $fd, $server, $frame);
        } catch (Throwable $e) {
            Swoft::trigger(WsServerEvent::MESSAGE_ERROR, $e, $frame);

            server()->log("Message: conn#{$fd} error: " . $e->getMessage(), [], 'error');

            /** @var WsErrorDispatcher $errDispatcher */
            $errDispatcher = BeanFactory::getSingleton(WsErrorDispatcher::class);
            $errDispatcher->messageError($e, $frame);
        } finally {
            // Defer event
            Swoft::trigger(SwoftEvent::COROUTINE_DEFER);

            // Destroy event
            Swoft::trigger(SwoftEvent::COROUTINE_COMPLETE);

            // Unbind cid => sid(fd)
            Session::unbindCo();
        }
    }
}