kiri-core/HttpServer/Events/OnClose.php

<?php
declare(strict_types=1);

namespace HttpServer\Events;


use Annotation\Route\Socket;
use HttpServer\Abstracts\Callback;
use HttpServer\Route\Annotation\Http;
use HttpServer\Route\Annotation\Tcp;
use HttpServer\Route\Annotation\Websocket;
use HttpServer\Route\Annotation\Websocket as AWebsocket;
use HttpServer\Route\Node;
use Snowflake\Event;
use Snowflake\Exception\ComponentException;
use Snowflake\Snowflake;
use Swoole\Coroutine;
use Swoole\Server;
use Exception;
use Swoole\Http\Server as HServer;
use Swoole\WebSocket\Server as WServer;

/**
 * Class OnClose
 * @package HttpServer\Events
 *
 */
class OnClose extends Callback
{


	/**
	 * @param Server $server
	 * @param int $fd
	 * @throws Exception
	 */
	public function onHandler(Server $server, int $fd)
	{
		Coroutine::defer(function () {
			fire(Event::EVENT_AFTER_REQUEST);
		});
		$this->execute($server, $fd);
	}


	/**
	 * @param Server $server
	 * @param int $fd
	 * @throws ComponentException
	 * @throws Exception
	 */
	private function execute(Server $server, int $fd): void
	{
		try {
			$router = Snowflake::app()->getRouter();
			if (!($server instanceof WServer) || !$server->isEstablished($fd)) {
				return;
			}
			$node = $router->search(Socket::CLOSE . '::event', 'sw::socket');
			if ($node instanceof Node) {
				$node->dispatch();
			}
		} catch (\Throwable $exception) {
			$this->addError($exception);
		} finally {
			$logger = Snowflake::app()->getLogger();
			$logger->insert();
		}
	}


	/**
	 * @param $server
	 * @param $fd
	 * @return array|null
	 * @throws Exception
	 */
	public function resolve($server, $fd): ?array
	{
		if ($server instanceof WServer) {
			if (!$server->isEstablished($fd)) {
				return [null, null];
			}
			$router = Snowflake::app()->getRouter();
			$node = $router->search(Socket::HANDSHAKE . '::' . null, 'sw::socket');
			if ($node === null) {
				return [null, null];
			}
			return $node->dispatch();
		} else if ($server instanceof HServer) {
			$manager = Snowflake::app()->annotation->http;
			$name = $manager->getName(Http::CLOSE);
		} else {
			$manager = Snowflake::app()->annotation->tcp;
			$name = $manager->getName(Tcp::CLOSE);
		}
		return [$manager, $name];
	}


}