From d2db326ecfe448544fddf754b33a1400ffbb5410 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Mr=C2=B7x?= <as2252258@163.com>
Date: Tue, 23 Mar 2021 15:13:33 +0800
Subject: [PATCH] =?UTF-8?q?=E6=94=B9=E5=90=8D?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 Annotation/Rpc/Consumer.php   | 47 ++++++++++++++++++++++++++++
 Annotation/Rpc/RpcService.php | 58 +++++++++++++++++++++++++++++++++++
 2 files changed, 105 insertions(+)
 create mode 100644 Annotation/Rpc/Consumer.php
 create mode 100644 Annotation/Rpc/RpcService.php

diff --git a/Annotation/Rpc/Consumer.php b/Annotation/Rpc/Consumer.php
new file mode 100644
index 00000000..301beb97
--- /dev/null
+++ b/Annotation/Rpc/Consumer.php
@@ -0,0 +1,47 @@
+<?php
+
+
+namespace Annotation\Rpc;
+
+
+use Annotation\Attribute;
+use ReflectionException;
+use Snowflake\Exception\ComponentException;
+use Snowflake\Exception\NotFindClassException;
+use Snowflake\Snowflake;
+
+
+/**
+ * Class Consumer
+ * @package Annotation\Rpc
+ */
+#[\Attribute(\Attribute::TARGET_METHOD)] class Consumer extends Attribute
+{
+
+
+	/**
+	 * Consumer constructor.
+	 * @param string $cmd
+	 */
+	public function __construct(public string $cmd)
+	{
+
+	}
+
+
+	/**
+	 * @param array $handler
+	 * @return mixed
+	 * @throws ReflectionException
+	 * @throws ComponentException
+	 * @throws NotFindClassException
+	 */
+	public function execute(array $handler): mixed
+	{
+		$rpc = Snowflake::app()->getRpc();
+		$rpc->addConsumer($this->cmd, $handler);
+		return parent::execute($handler); // TODO: Change the autogenerated stub
+	}
+
+
+}
diff --git a/Annotation/Rpc/RpcService.php b/Annotation/Rpc/RpcService.php
new file mode 100644
index 00000000..f8bca780
--- /dev/null
+++ b/Annotation/Rpc/RpcService.php
@@ -0,0 +1,58 @@
+<?php
+
+
+namespace Annotation\Rpc;
+
+
+use Annotation\Attribute;
+use ReflectionException;
+use Snowflake\Exception\ComponentException;
+use Snowflake\Exception\NotFindClassException;
+use Snowflake\Snowflake;
+
+
+/**
+ * Class RpcClient
+ * @package Annotation\Rpc
+ */
+#[\Attribute(\Attribute::TARGET_CLASS)] class RpcService extends Attribute
+{
+
+
+	/**
+	 * RpcClient constructor.
+	 * @param string $cmd
+	 * @param string $host
+	 * @param int $port
+	 * @param int $mode
+	 */
+	public function __construct(
+		public string $cmd,
+		public string $host,
+		public int $port,
+		public int $mode = SWOOLE_SOCK_TCP6
+	)
+	{
+	}
+
+
+	/**
+	 * @param array $handler
+	 * @return mixed
+	 * @throws ReflectionException
+	 * @throws ComponentException
+	 * @throws NotFindClassException
+	 */
+	public function execute(array $handler): mixed
+	{
+		$rpc = Snowflake::app()->getRpc();
+		$rpc->addProducer($this->cmd, $handler, [
+			'host' => $this->host,
+			'port' => $this->port,
+			'mode' => $this->mode
+		]);
+		return parent::execute($handler); // TODO: Change the autogenerated stub
+	}
+
+
+}