eee

core/QueueLoop.php

@@ -0,0 +1,47 @@
+<?php
+
+namespace Coroutine\Server;
+
+use Kiri\Di\Inject\Container;
+use Kiri\NoSql\Redis;
+use Swoole\Coroutine;
+use Swoole\Coroutine\Http\Client;
+
+class QueueLoop
+{
+
+	#[Container(Transport::class)]
+	public Transport $transport;
+
+	/**
+	 * @return void
+	 */
+	public function loop(): void
+	{
+		$this->cleanOnline();
+		$redis = \Kiri::getDi()->get(Redis::class);
+		if ($redis instanceof Redis) {
+			$data = $redis->blPop(\config('redis.key'), 10);
+			if (!empty($data)) {
+				[$key, $data] = $data;
+				$json = json_decode($data, true);
+
+				$this->transport->send($json['userId'], json_encode($json['data']));
+			}
+		} else {
+			Coroutine::sleep(0.05);
+		}
+		$this->loop();
+	}
+
+
+	/**
+	 * @return void
+	 */
+	public function cleanOnline(): void
+	{
+		$client = new Client(config('notice.host'), config('notice.port'));
+		$client->post('/cleanOnline', []);
+		$client->close();
+	}
+}

core/Struct.php

@@ -0,0 +1,18 @@
+<?php
+namespace Coroutine\Server;
+
+use Swoole\Http\Response;
+
+class Struct
+{
+	public mixed    $user;
+	public int      $fd;
+	public Response $ws;
+
+	public function __construct(mixed $user, int $fd, Response $ws)
+	{
+		$this->fd   = $fd;
+		$this->user = $user;
+		$this->ws   = $ws;
+	}
+}

core/Transport.php

@@ -0,0 +1,88 @@
+<?php
+
+namespace Coroutine\Server;
+
+class Transport
+{
+
+	/**
+	 * @var array<Struct>
+	 */
+	private array $clients = [];
+
+
+	/**
+	 * @param int    $fd
+	 * @param Struct $data
+	 * @return void
+	 */
+	public function add(int $fd, Struct $data): void
+	{
+		if (isset($this->clients[$fd])) {
+			$this->clients[$fd]->ws->close();
+		}
+		$this->clients[$fd] = $data;
+	}
+
+	/**
+	 * @param int   $fd
+	 * @param mixed $data
+	 * @return void
+	 */
+	public function send(int $fd, mixed $data): void
+	{
+		if (isset($this->clients[$fd])) {
+			$this->clients[$fd]->ws->push($data);
+		}
+	}
+
+
+	/**
+	 * @param int $fd
+	 * @return void
+	 */
+	public function close(int $fd): void
+	{
+		if (isset($this->clients[$fd])) {
+			$this->clients[$fd]->ws->close();
+		}
+	}
+
+
+	/**
+	 * @param int      $fd
+	 * @return void
+	 */
+	public function remove(int $fd): void
+	{
+		if ($this->clients[$fd]) {
+			$this->clients[$fd]->ws->close();
+		}
+		unset($this->clients[$fd]);
+	}
+
+
+	/**
+	 * @return array
+	 */
+	public function getLists(): array
+	{
+		$array = [];
+		foreach ($this->clients as $fd => $client) {
+			$array[] = [
+				'userId'   => $fd,
+				'nickname' => $client->user['nickname'],
+			];
+		}
+		return $array;
+	}
+
+
+	/**
+	 * @return int
+	 */
+	public function size(): int
+	{
+		return count($this->clients);
+	}
+}

core/Websocket.php

@@ -0,0 +1,155 @@
+<?php
+
+namespace Coroutine\Server;
+
+use Closure;
+use Kiri\Di\Context;
+use Kiri\Di\Inject\Container;
+use Kiri\Router\Constrict\ConstrictRequest;
+use Kiri\Router\Constrict\ConstrictResponse;
+use Kiri\Server\Contract\OnDisconnectInterface;
+use Kiri\Server\Contract\OnMessageInterface;
+use Psr\Http\Message\ResponseInterface;
+use Swoole\Coroutine;
+use Swoole\Coroutine\Http\Server;
+use Swoole\Http\Request;
+use Swoole\Http\Response;
+use Swoole\WebSocket\CloseFrame;
+use Swoole\WebSocket\Frame;
+use Throwable;
+
+abstract class Websocket implements OnMessageInterface, OnDisconnectInterface
+{
+
+	#[Container(Transport::class)]
+	public Transport $transport;
+
+
+	private Server $server;
+
+	/**
+	 * @param string $host
+	 * @param int    $port
+	 * @return void
+	 */
+	public function start(string $host, int $port): void
+	{
+		$this->clearAllOnlineUsers();
+		$this->queueLoop();
+		$this->server = new Server($host, $port, false);
+		$this->server->handle('/websocket', function (Request $request, Response $ws) {
+			$this->WebsocketHandler($request, $ws);
+		});
+		$this->server->handle('/online/lists', function (Request $request, Response $ws) {
+			$data = json_encode($this->transport->getLists(), JSON_UNESCAPED_UNICODE);
+			$ws->end($data);
+		});
+		echo 'websocket server start at ' . $host . ':' . $port . PHP_EOL;
+		$this->server->start();
+	}
+
+
+	/**
+	 * @return void
+	 */
+	abstract public function clearAllOnlineUsers(): void;
+
+
+	/**
+	 * @return void
+	 */
+	private function queueLoop(): void
+	{
+		Coroutine::create(function () {
+			$QueueLoop = \Kiri::getDi()->get(QueueLoop::class);
+			$QueueLoop->loop();
+		});
+	}
+
+	/**
+	 * @param Request  $request
+	 * @param Response $ws
+	 * @return void
+	 */
+	public function WebsocketHandler(Request $request, Response $ws): void
+	{
+		try {
+			$psr7Request = ConstrictRequest::builder($request);
+			if (!$psr7Request->hasQuery('auth')) {
+				throw new \Exception('Auth fail.', 401);
+			}
+
+			if (!$this->onAuthority($psr7Request, $ws)) {
+				throw new \Exception('Auth fail.', 401);
+			}
+
+			if ($psr7Request->getAuthority() == null || !$ws->upgrade()) {
+				throw new \Exception('Auth fail.', 401);
+			}
+			$userId = $psr7Request->getAuthority()->getUniqueId();
+
+			$ws->push($userId);
+			$user = [
+				'nickname' => $psr7Request->getAuthority()->getNickname(),
+				'avatar'   => $psr7Request->getAuthority()->getAvatar(),
+			];
+
+			if (!$this->onConnected($psr7Request)) {
+				throw new \Exception('Auth fail.', 401);
+			}
+			$this->transport->add($userId, new Struct($user, $request->fd, $ws));
+
+			while (true) {
+				$frame = $ws->recv();
+				if ($frame === '' || $frame === false || $frame->data == 'close' || get_class($frame) === CloseFrame::class) {
+					$ws->close();
+					break;
+				}
+
+				$this->onMessage($this->server, $frame);
+			}
+			$this->onDisconnect($this->server, $userId);
+		} catch (Throwable $throwable) {
+			echo sprintf("Message: %s \n
+			   File: %s \n
+			   Line: %d \n", $throwable->getMessage(), $throwable->getFile(), $throwable->getLine());
+			$ws->setStatusCode(500);
+			$ws->end($throwable->getMessage());
+			$ws->close();
+		} finally {
+			if (isset($userId)) {
+				$this->transport->remove($userId);
+			}
+		}
+	}
+
+
+	/**
+	 * @param ConstrictRequest $request
+	 * @return bool
+	 */
+	abstract public function onConnected(ConstrictRequest $request): bool;
+
+
+	/**
+	 * @param ConstrictRequest $request
+	 * @param Response         $ws
+	 * @return bool
+	 */
+	abstract public function onAuthority(ConstrictRequest $request, Response $ws): bool;
+
+
+	/**
+	 * @param \Swoole\Server|Server $server
+	 * @param Frame                 $frame
+	 * @return void
+	 */
+	abstract public function onMessage(\Swoole\Server|Server $server, Frame $frame): void;
+
+	/**
+	 * @param \Swoole\WebSocket\Server|Server $server
+	 * @param int                             $fd
+	 * @return void
+	 */
+	abstract public function onDisconnect(\Swoole\WebSocket\Server|Server $server, int $fd): void;
+}