80 lines
2.7 KiB
PHP
80 lines
2.7 KiB
PHP
<?php
|
|
|
|
namespace think\swoole\command;
|
|
|
|
use Nette\PhpGenerator\ClassType;
|
|
use Nette\PhpGenerator\Helpers;
|
|
use Nette\PhpGenerator\PhpFile;
|
|
use think\console\Command;
|
|
use think\helper\Arr;
|
|
use think\swoole\contract\rpc\ParserInterface;
|
|
use think\swoole\exception\RpcResponseException;
|
|
use think\swoole\rpc\client\Client;
|
|
use think\swoole\rpc\Error;
|
|
use think\swoole\rpc\JsonParser;
|
|
use think\swoole\rpc\server\Dispatcher;
|
|
|
|
class RpcInterface extends Command
|
|
{
|
|
public function configure()
|
|
{
|
|
$this->setName('rpc:interface')
|
|
->setDescription('Generate Rpc Service Interfaces');
|
|
}
|
|
|
|
public function handle()
|
|
{
|
|
go(function () {
|
|
$clients = $this->app->config->get('swoole.rpc.client', []);
|
|
|
|
$file = new PhpFile;
|
|
$file->addComment('This file is auto-generated.');
|
|
$file->setStrictTypes();
|
|
|
|
foreach ($clients as $name => $config) {
|
|
|
|
$client = new Client($config['host'], $config['port']);
|
|
|
|
$response = $client->sendAndRecv(Dispatcher::ACTION_INTERFACE);
|
|
|
|
$parserClass = Arr::get($config, 'parser', JsonParser::class);
|
|
/** @var ParserInterface $parser */
|
|
$parser = new $parserClass;
|
|
|
|
$result = $parser->decodeResponse($response);
|
|
|
|
if ($result instanceof Error) {
|
|
throw new RpcResponseException($result);
|
|
}
|
|
|
|
$namespace = $file->addNamespace("rpc\\contract\\${name}");
|
|
foreach ($result as $interface => $methods) {
|
|
$class = $namespace->addInterface($interface);
|
|
$class->addConstant("RPC", $name);
|
|
|
|
foreach ($methods as $methodName => ['parameters' => $parameters, 'returnType' => $returnType, 'comment' => $comment]) {
|
|
$method = $class->addMethod($methodName)
|
|
->setVisibility(ClassType::VISIBILITY_PUBLIC)
|
|
->setComment(Helpers::unformatDocComment($comment))
|
|
->setReturnType($returnType);
|
|
|
|
foreach ($parameters as $parameter) {
|
|
|
|
$param = $method->addParameter($parameter['name'])
|
|
->setTypeHint($parameter['type']);
|
|
|
|
if (array_key_exists("default", $parameter)) {
|
|
$param->setDefaultValue($parameter['default']);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
file_put_contents($this->app->getBasePath() . 'rpc.php', $file);
|
|
|
|
$this->output->writeln('<info>Succeed!</info>');
|
|
});
|
|
}
|
|
}
|