eee

2026-02-26 14:39:04 +08:00
parent c0a61aebdd
commit d90d2eb844
6 changed files with 463 additions and 0 deletions
@@ -0,0 +1,131 @@
+<?php
+
+
+use Coroutine\Server\Transport;
+use Coroutine\Server\Websocket;
+use Kiri\Router\Constrict\ConstrictRequest;
+use Swoole\Coroutine\Http\Client;
+use Swoole\Coroutine\Http\Server;
+use Swoole\Http\Request;
+use Swoole\Http\Response;
+use Swoole\WebSocket\Frame;
+use function Co\run;
+
+
+class Test extends Websocket
+{
+
+
+	/**
+	 * @param ConstrictRequest $request
+	 * @return bool
+	 */
+	public function onConnected(ConstrictRequest $request): bool
+	{
+		// TODO: Implement onConnected() method.
+		$this->stateChange($request->getAuthority()->getUniqueId(), 1);
+
+		return true;
+	}
+
+
+	/**
+	 * @param ConstrictRequest $request
+	 * @param Response         $ws
+	 * @return bool
+	 */
+	public function onAuthority(ConstrictRequest $request, Response $ws): bool
+	{
+		// TODO: Implement onAuthority() method.
+		$auth   = $request->get['auth'] ?? '';
+		$client = new Client(config('user.host'), config('user.port'), true);
+		$client->post('/handshake', [
+			'token' => $auth,
+			'fd'    => $request->fd,
+			'ip'    => current(swoole_get_local_ip()),
+		]);
+		$client->close();
+		if ($client->getStatusCode() != 200) {
+			return false;
+		}
+
+		$data = json_decode($client->body, true);
+
+		$request->withAuthority($data);
+		return true;
+	}
+
+	/**
+	 * @param \Swoole\Server|Server $server
+	 * @param Frame                 $frame
+	 * @return void
+	 */
+	public function onMessage(\Swoole\Server|Server $server, Frame $frame): void
+	{
+		// TODO: Implement onMessage() method.
+	}
+
+
+	/**
+	 * @param Server|\Swoole\WebSocket\Server $server
+	 * @param int                             $fd
+	 * @return void
+	 */
+	public function onDisconnect(Server|\Swoole\WebSocket\Server $server, int $fd): void
+	{
+		// TODO: Implement onDisconnect() method.
+		$this->stateChange($fd, 0);
+	}
+
+
+
+	/**
+	 * @return void
+	 */
+	public function clearAllOnlineUsers(): void
+	{
+		$request = [
+			'query'  => [
+				'bool' => [
+					'filter' => [['term' => ['is_online' => 1]]],
+				],
+			],
+			'script' => [
+				'source' => "ctx._source['is_online'] = 0",
+				'lang'   => 'painless',
+			],
+		];
+		$client  = new Client(config('es_host'), +config('es_port'));
+		$client->setData(json_encode($request, JSON_NUMERIC_CHECK));
+		$client->setHeaders(['Content-Type' => 'application/json']);
+		$client->setMethod('POST');
+		$client->execute('/' . config('es_key') . '/_update_by_query');
+		$client->close();
+	}
+
+
+	/**
+	 * @param int $userId
+	 * @param int $status
+	 * @return void
+	 */
+	private function stateChange(int $userId, int $status): void
+	{
+		$data   = [
+			'doc' => [
+				'is_online' => $status,
+			],
+		];
+		$client = new Client(config('es_host'), +config('es_port'));
+		$client->setData(json_encode($data));
+		$client->setHeaders(['Content-Type' => 'application/json']);
+		$client->setMethod('POST');
+		$client->execute('/' . config('es_key') . '/_doc/' . $userId . '/_update');
+		$client->close();
+	}
+}
+
+run(function () {
+	$websocket = Kiri::getDi()->get(Test::class);
+	$websocket->start('0.0.0.0', 9504);
+});