改名

Annotation/Rpc/RpcClient.php

@@ -0,0 +1,65 @@
+<?php
+
+
+namespace Annotation\Rpc;
+
+
+use Annotation\Attribute;
+use Exception;
+use ReflectionException;
+use Rpc\IProducer;
+use Snowflake\Exception\ComponentException;
+use Snowflake\Exception\NotFindClassException;
+use Snowflake\Snowflake;
+
+
+/**
+ * Class RpcClient
+ * @package Annotation\Rpc
+ */
+#[\Attribute(\Attribute::TARGET_CLASS)] class RpcClient extends Attribute
+{
+
+	private array $config;
+
+
+	/**
+	 * RpcClient constructor.
+	 * @param string $cmd
+	 * @param int $port
+	 * @param int $timeout
+	 * @param int $mode
+	 */
+	public function __construct(
+		public string $cmd,
+		public int $port,
+		public int $timeout = 1,
+		public int $mode = SWOOLE_SOCK_TCP6
+	)
+	{
+		$this->config = ['port' => $port, 'mode' => $mode, 'timeout' => $timeout];
+	}
+
+
+	/**
+	 * @param array $handler
+	 * @return mixed
+	 * @throws ReflectionException
+	 * @throws ComponentException
+	 * @throws NotFindClassException
+	 * @throws Exception
+	 */
+	public function execute(array $handler): mixed
+	{
+		$rpc = Snowflake::app()->getRpc();
+		$rpc->addProducer($this->cmd, $handler, $this->config);
+
+		if ($handler[0] instanceof IProducer) {
+			$handler[0]->initClient();
+		}
+
+		return parent::execute($handler);
+	}
+
+
+}