mirror of
https://github.com/zhamao-robot/zhamao-framework.git
synced 2026-07-04 15:25:38 +08:00
1ffb30a471
change global.php config load time and logic set context get server function available more time delete unused comment and @CQAPISend @CQCommand add start_with and end_with set exceptions extended by ZMException rename @SwooleSetup to @ZMSetup fix quotes for global.php fix LightCache empty presistence_path error remove RemoteShell
88 lines
2.7 KiB
PHP
88 lines
2.7 KiB
PHP
<?php
|
|
|
|
|
|
namespace ZM\API;
|
|
|
|
use Co;
|
|
use ZM\ConnectionManager\ConnectionObject;
|
|
use ZM\Console\Console;
|
|
use ZM\Store\LightCacheInside;
|
|
use ZM\Store\Lock\SpinLock;
|
|
use ZM\Store\ZMAtomic;
|
|
|
|
trait CQAPI
|
|
{
|
|
/**
|
|
* @param ConnectionObject $connection
|
|
* @param $reply
|
|
* @param |null $function
|
|
* @return bool|array
|
|
*/
|
|
private function processAPI($connection, $reply, $function = null) {
|
|
if ($connection->getOption("type") === CONN_WEBSOCKET)
|
|
return $this->processWebsocketAPI($connection, $reply, $function);
|
|
else
|
|
return $this->processHttpAPI($connection, $reply, $function);
|
|
|
|
|
|
}
|
|
|
|
public function processWebsocketAPI($connection, $reply, $function = false) {
|
|
$api_id = ZMAtomic::get("wait_msg_id")->add(1);
|
|
$reply["echo"] = $api_id;
|
|
SpinLock::lock("wait_api");
|
|
$r = LightCacheInside::get("wait_api", "wait_api");
|
|
$r[$api_id] = [
|
|
"data" => $reply,
|
|
"time" => microtime(true),
|
|
"self_id" => $connection->getOption("connect_id")
|
|
];
|
|
if ($function === true) $r[$api_id]["coroutine"] = Co::getuid();
|
|
LightCacheInside::set("wait_api", "wait_api", $r);
|
|
SpinLock::unlock("wait_api");
|
|
if (server()->push($connection->getFd(), json_encode($reply))) {
|
|
if ($function === true) {
|
|
Co::suspend();
|
|
SpinLock::lock("wait_api");
|
|
$r = LightCacheInside::get("wait_api", "wait_api");
|
|
$data = $r[$api_id];
|
|
unset($r[$api_id]);
|
|
LightCacheInside::set("wait_api", "wait_api", $r);
|
|
SpinLock::unlock("wait_api");
|
|
return isset($data['result']) ? $data['result'] : null;
|
|
}
|
|
return true;
|
|
} else {
|
|
Console::warning("CQAPI send failed, websocket push error.");
|
|
$response = [
|
|
"status" => "failed",
|
|
"retcode" => -1000,
|
|
"data" => null,
|
|
"self_id" => $connection->getOption("connect_id")
|
|
];
|
|
SpinLock::lock("wait_api");
|
|
$r = LightCacheInside::get("wait_api", "wait_api");
|
|
unset($r[$reply["echo"]]);
|
|
LightCacheInside::set("wait_api", "wait_api", $r);
|
|
SpinLock::unlock("wait_api");
|
|
if ($function === true) return $response;
|
|
return false;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* @param $connection
|
|
* @param $reply
|
|
* @param null $function
|
|
* @return bool
|
|
* @noinspection PhpUnusedParameterInspection
|
|
*/
|
|
public function processHttpAPI($connection, $reply, $function = null) {
|
|
return false;
|
|
}
|
|
|
|
public function __call($name, $arguments) {
|
|
return false;
|
|
}
|
|
}
|