83 lines
2.1 KiB
PHP
83 lines
2.1 KiB
PHP
|
<?php
|
||
|
|
|
||
|
|
declare(strict_types=1);
|
||
|
|
/**
|
||
|
|
* This file is part of Hyperf.
|
||
|
|
*
|
||
|
|
* @link https://www.hyperf.io
|
||
|
|
* @document https://hyperf.wiki
|
||
|
|
* @contact group@hyperf.io
|
||
|
|
* @license https://github.com/hyperf/hyperf/blob/master/LICENSE
|
||
|
|
*/
|
||
|
|
|
||
|
|
namespace Hyperf\Engine\WebSocket;
|
||
|
|
|
||
|
|
use Hyperf\Engine\Contract\WebSocket\FrameInterface;
|
||
|
|
use Hyperf\Engine\Contract\WebSocket\ResponseInterface;
|
||
|
|
use Hyperf\Engine\Exception\InvalidArgumentException;
|
||
|
|
use Swoole\Http\Request;
|
||
|
|
use Swoole\Http\Response as SwooleResponse;
|
||
|
|
use Swoole\WebSocket\Frame as SwooleFrame;
|
||
|
|
use Swoole\WebSocket\Server;
|
||
|
|
|
||
|
|
use function Hyperf\Engine\swoole_get_flags_from_frame;
|
||
|
|
|
||
|
|
class Response implements ResponseInterface
|
||
|
|
{
|
||
|
|
protected int $fd = 0;
|
||
|
|
|
||
|
|
public function __construct(protected mixed $connection)
|
||
|
|
{
|
||
|
|
}
|
||
|
|
|
||
|
|
public function push(FrameInterface $frame): bool
|
||
|
|
{
|
||
|
|
$data = (string) $frame->getPayloadData();
|
||
|
|
$flags = swoole_get_flags_from_frame($frame);
|
||
|
|
|
||
|
|
if ($this->connection instanceof SwooleResponse) {
|
||
|
|
$this->connection->push($data, $frame->getOpcode(), $flags);
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
if ($this->connection instanceof Server) {
|
||
|
|
$this->connection->push($this->fd, $data, $frame->getOpcode(), $flags);
|
||
|
|
return true;
|
||
|
|
}
|
||
|
|
|
||
|
|
throw new InvalidArgumentException('The websocket connection is invalid.');
|
||
|
|
}
|
||
|
|
|
||
|
|
public function init(mixed $frame): static
|
||
|
|
{
|
||
|
|
switch (true) {
|
||
|
|
case is_int($frame):
|
||
|
|
$this->fd = $frame;
|
||
|
|
break;
|
||
|
|
case $frame instanceof Request || $frame instanceof SwooleFrame:
|
||
|
|
$this->fd = $frame->fd;
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
|
||
|
|
return $this;
|
||
|
|
}
|
||
|
|
|
||
|
|
public function getFd(): int
|
||
|
|
{
|
||
|
|
return $this->fd;
|
||
|
|
}
|
||
|
|
|
||
|
|
public function close(): bool
|
||
|
|
{
|
||
|
|
if ($this->connection instanceof SwooleResponse) {
|
||
|
|
return $this->connection->close();
|
||
|
|
}
|
||
|
|
|
||
|
|
if ($this->connection instanceof Server) {
|
||
|
|
return $this->connection->disconnect($this->fd);
|
||
|
|
}
|
||
|
|
|
||
|
|
return false;
|
||
|
|
}
|
||
|
|
}
|