zhamao-robot/zhamao-framework
1
0
Fork 0
mirror of https://github.com/zhamao-robot/zhamao-framework.git synced 2026-07-24 00:55:35 +08:00
Code Issues Packages Projects Releases Wiki Activity

add ws_socket function to get ws server socket

Browse Source
This commit is contained in:
crazywhalecc
2023-03-05 14:36:58 +08:00
committed by Jerry
parent 1a7c4607ee
commit a3c7ffa892
1 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

18
src/Globals/global_functions.php
View File

@@ -6,6 +6,7 @@ use Choir\Http\HttpFactory;
use OneBot\Driver\Coroutine\Adaptive; use OneBot\Driver\Coroutine\Adaptive;
use OneBot\Driver\Coroutine\CoroutineInterface; use OneBot\Driver\Coroutine\CoroutineInterface;
use OneBot\Driver\Process\ExecutionResult; use OneBot\Driver\Process\ExecutionResult;
use OneBot\Driver\Socket\WSServerSocketBase;
use OneBot\V12\Object\MessageSegment; use OneBot\V12\Object\MessageSegment;
use Psr\Http\Message\ResponseInterface; use Psr\Http\Message\ResponseInterface;
use Psr\Log\LoggerInterface; use Psr\Log\LoggerInterface;
@@ -13,8 +14,10 @@ use Psr\SimpleCache\CacheInterface;
use ZM\Config\Environment; use ZM\Config\Environment;
use ZM\Config\ZMConfig; use ZM\Config\ZMConfig;
use ZM\Container\ContainerHolder; use ZM\Container\ContainerHolder;
use ZM\Framework;
use ZM\Logger\ConsoleLogger; use ZM\Logger\ConsoleLogger;
use ZM\Middleware\MiddlewareHandler; use ZM\Middleware\MiddlewareHandler;
use ZM\Plugin\OneBot\BotMap;
use ZM\Schedule\Timer; use ZM\Schedule\Timer;
use ZM\Store\Database\DBException; use ZM\Store\Database\DBException;
use ZM\Store\Database\DBQueryBuilder; use ZM\Store\Database\DBQueryBuilder;
@@ -327,3 +330,18 @@ function zm_http_response(int $status_code = 200, ?string $reason = null, array
{ {
return HttpFactory::createResponse($status_code, $reason, $headers, $body, $protocol); return HttpFactory::createResponse($status_code, $reason, $headers, $body, $protocol);
} }
/**
* 【助手函数】获取驱动的 WebSocket 服务器对应 Socket 操作对象
*
* @param int $flag 对应的 Server 端口标记
* @throws Exception
*/
function ws_socket(int $flag = 1): WSServerSocketBase
{
$a = Framework::getInstance()->getDriver()->getWSServerSocketByFlag($flag);
if ($a === null) {
throw new Exception('找不到目标的 server socket，请检查 flag 值');
}
return $a;
}