初始化代码

2025-12-22 14:34:25 +08:00
parent c2c5ae2fdd
commit a77dbc743f
1510 changed files with 213008 additions and 0 deletions
--- a/vendor/topthink/think-swoole/src/command/RpcInterface.php
+++ b/vendor/topthink/think-swoole/src/command/RpcInterface.php
@@ -0,0 +1,79 @@
+<?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>');
+        });
+    }
+}