implement FailedMessageTrait

2026-03-17 10:36:38 +08:00
parent 20fe8ccf5e
commit fcd376cab4
4 changed files with 114 additions and 295 deletions
@@ -0,0 +1,107 @@
+<?php
+
+declare(strict_types=1);
+
+namespace App\Platform\Traits;
+
+use App\Model\FailedMessage;
+use App\Platform\ErrorProducer;
+use App\Utils\Log;
+use Hyperf\Amqp\Producer;
+use Hyperf\Context\ApplicationContext;
+use PhpAmqpLib\Message\AMQPMessage;
+use Throwable;
+
+/**
+ * 失败消息处理 Trait
+ *
+ * 提供 Consumer 共用的重试计数、错误队列发送、失败消息持久化能力
+ */
+trait FailedMessageTrait
+{
+    /**
+     * 获取消息的重试次数
+     */
+    protected function getRetryCount(AMQPMessage $message): int
+    {
+        if (!$message->has('application_headers')) {
+            return 0;
+        }
+
+        $headers = $message->get('application_headers');
+        if (!$headers) {
+            return 0;
+        }
+
+        $header_data = $headers->getNativeData();
+        $x_death = $header_data['x-death'] ?? [];
+
+        if (empty($x_death)) {
+            return 0;
+        }
+
+        return $x_death[0]['count'] ?? 0;
+    }
+
+    /**
+     * 发送消息到错误队列
+     * 当重试次数超过上限时调用
+     *
+     * @param AMQPMessage $message 原始消息
+     * @param Throwable $error 错误信息
+     */
+    protected function sendToErrorQueue(AMQPMessage $message, Throwable $error): void
+    {
+        try {
+            $retry_count = $this->getRetryCount($message);
+
+            $producer = ApplicationContext::getContainer()->get(Producer::class);
+            $error_producer = new ErrorProducer($message, $error, $retry_count);
+            $producer->produce($error_producer);
+
+            // 同步写入 failed_messages 表
+            $this->persistFailedMessage($error_producer->payload);
+
+            Log::get()->warning('Message sent to error queue after exceeding retry limit', [
+                'error_id' => $error_producer->payload['error_id'] ?? 'unknown',
+                'retry_count' => $retry_count,
+                'error_message' => $error->getMessage(),
+            ]);
+        } catch (Throwable $e) {
+            Log::get()->error('Failed to send message to error queue', [
+                'error' => $e->getMessage(),
+                'original_error' => $error->getMessage(),
+            ]);
+        }
+    }
+
+    /**
+     * 持久化失败消息到数据库
+     */
+    protected function persistFailedMessage(array $payload): void
+    {
+        try {
+            FailedMessage::query()->create([
+                'error_id' => $payload['error_id'],
+                'data_type' => $payload['metadata']['data_type'] ?? 'unknown',
+                'platform' => $payload['metadata']['platform'] ?? null,
+                'platform_id' => $payload['metadata']['platform_id'] ?? null,
+                'company_id' => $payload['metadata']['company_id'] ?? null,
+                'store_id' => $payload['metadata']['store_id'] ?? null,
+                'error_type' => $payload['error']['type'] ?? 'Unknown',
+                'error_message' => $payload['error']['message'] ?? '',
+                'error_code' => $payload['error']['code'] ?? 0,
+                'error_trace' => $payload['error']['trace'] ?? '',
+                'original_message' => $payload['original_message'] ?? [],
+                'retry_count' => $payload['metadata']['retry_count'] ?? 0,
+                'message_id' => $payload['metadata']['message_id'] ?? null,
+                'failed_at' => $payload['metadata']['failed_at'] ?? date('c'),
+            ]);
+        } catch (Throwable $e) {
+            Log::get()->error('Failed to persist failed message to database', [
+                'error' => $e->getMessage(),
+                'error_id' => $payload['error_id'] ?? 'unknown',
+            ]);
+        }
+    }
+}