协程版仓库后端项目

app/Amqp/Consumer/BaseConsumer.php

@@ -0,0 +1,89 @@
+<?php
+/**
+ * Author: ykxiao
+ * Date: 2025/6/3
+ * Time: 下午6:37
+ * Description:
+ *
+ * (c) ykxiao <yk_9001@hotmail.com>
+ *
+ * This source file is subject to the MIT license that is bundled
+ * with this source code in the file LICENSE.
+ */
+
+declare(strict_types=1);
+
+namespace App\Amqp\Consumer;
+
+use Hyperf\Amqp\Message\ConsumerMessage;
+use App\Context\QueueContext;
+use Hyperf\Amqp\Result;
+use Hyperf\DbConnection\Db;
+use PhpAmqpLib\Exception\AMQPChannelClosedException;
+use PhpAmqpLib\Message\AMQPMessage;
+use App\Log\Log;
+use Exception;
+use PhpAmqpLib\Exception\AMQPConnectionClosedException;
+
+/**
+ * Author: ykxiao
+ * Date: 2025/6/3
+ * Time: 下午7:18
+ * Description: 消费者抽象基类.
+ *
+ * (c) ykxiao <yk_9001@hotmail.com>
+ *
+ * This source file is subject to the MIT license that is bundled
+ * with this source code in the file LICENSE.
+ */
+abstract class BaseConsumer extends ConsumerMessage
+{
+    public function consumeMessage($data, AMQPMessage $message): Result
+    {
+        $consumerClass = get_class($this);
+
+        // 设置用户信息上下文信息
+        if (!empty($data['user'])) {
+            QueueContext::setUser($data['user']);
+        }
+        if (!empty($company = $data['company'])) {
+            QueueContext::setCompanyInfo($company);
+        }
+
+        Db::beginTransaction();
+
+        try {
+            $handle = $this->handle($data);
+
+            Db::commit();
+
+            return $handle;
+        } catch (AMQPChannelClosedException|AMQPConnectionClosedException $e) {
+            Log::get('queue', 'queue')->error("AMQP通道关闭异常 ($consumerClass): " . $e->getMessage(), [
+                'data' => $data,
+                'exception' => $e,
+            ]);
+
+            Db::rollBack();
+
+            // 可选：重连逻辑 or 丢弃
+            return Result::ACK; // 或 NACK
+        } catch (Exception $e) {
+            Log::get('queue', 'queue')->error("AMQP消费者异常 ($consumerClass): " . $e->getMessage(), [
+                'data' => $data,
+                'exception' => $e,
+            ]);
+
+            Db::rollBack();
+
+            return Result::ACK;
+        }
+    }
+
+    /**
+     * 子类实现的核心处理逻辑
+     * @param mixed $data
+     * @return Result
+     */
+    abstract protected function handle(mixed $data): Result;
+}