zhamao-framework/src/ZM/API/CQAPI.php

<?php


namespace ZM\API;

use Closure;
use ZM\Console\Console;
use ZM\Store\LightCacheInside;
use ZM\Store\Lock\SpinLock;
use ZM\Store\ZMAtomic;
use ZM\Utils\CoMessage;

trait CQAPI
{
    /** @var null|Closure */
    private static $filter = null;

    public static function registerFilter(callable $callable) {
        self::$filter = $callable;
    }

    /**
     * @param $connection
     * @param $reply
     * @param |null $function
     * @return bool|array
     */
    private function processAPI($connection, $reply, $function = null) {
        if (is_callable(self::$filter)) {
            $reply2 = call_user_func(self::$filter, $reply);
            if (is_bool($reply2)) return $reply2;
            else $reply = $reply2;
        }
        if ($connection->getOption("type") === CONN_WEBSOCKET)
            return $this->processWebsocketAPI($connection, $reply, $function);
        else
            return $this->processHttpAPI($connection, $reply, $function);
    }

    private function processWebsocketAPI($connection, $reply, $function = false) {
        $api_id = ZMAtomic::get("wait_msg_id")->add(1);
        $reply["echo"] = $api_id;
        if (server()->push($connection->getFd(), json_encode($reply))) {
            if ($function === true) {
                $obj = [
                    "data" => $reply,
                    "time" => microtime(true),
                    "self_id" => $connection->getOption("connect_id"),
                    "echo" => $api_id
                ];
                return CoMessage::yieldByWS($obj, ["echo"], 30);
            }
            return true;
        } else {
            Console::warning(zm_internal_errcode("E00036") . "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
     */
    private function processHttpAPI($connection, $reply, $function = null): bool {
        return false;
    }

    public function __call($name, $arguments) {
        return false;
    }
}