@taicode/common-server 1.0.10 → 1.0.11

package/output/redis-queue/redis-queue.js

@@ -2,7 +2,7 @@ import { randomUUID } from 'node:crypto';
 import { createClient } from 'redis';
 import { catchIt } from '@taicode/common-base';
 /**
- * 通用任务队列类
+ * 通用任务队列类（泛型）
  *
  * 提供基于 Redis 的任务队列功能，支持：
  * - 任务入队和持久化
@@ -10,24 +10,30 @@ import { catchIt } from '@taicode/common-base';
  * - 任务状态追踪
  * - 分布式消费
  *
+ * @template T 任务数据类型
+ *
  * @example
  * ```ts
- * const queue = new RedisQueue({
- *   redisUrl: 'redis://localhost:6379',
- *   queueKey: 'my-queue',
- *   concurrency: 5, // 并发处理 5 个任务
- * })
+ * interface EmailTask {
+ *   to: string
+ *   subject: string
+ * }
  *
- * // 注册任务处理器
- * queue.handle('send-email', async (data) => {
- *   await sendEmail(data.to, data.subject)
+ * const queue = new RedisQueue<EmailTask>({
+ *   redisUrl: 'redis://localhost:6379',
+ *   queueKey: 'email-queue',
+ *   concurrency: 5,
+ *   handler: async (data) => {
+ *     await sendEmail(data.to, data.subject)
+ *     return catchIt(() => {})
+ *   }
  * })
  *
  * // 连接 Redis（自动启动消费者）
  * await queue.connect()
  *
  * // 入队任务（自动开始处理）
- * await queue.enqueue('send-email', { to: 'user@example.com', subject: 'Hello' })
+ * await queue.enqueue({ to: 'user@example.com', subject: 'Hello' })
  *
  * // 获取队列统计信息（O(1) 时间复杂度）
  * const stats = await queue.statistics()
@@ -41,7 +47,7 @@ export class RedisQueue {
     recoveryInterval = null;
     processingTasks = 0; // 当前正在处理的任务数
     config;
-    handlers = new Map();
+    handler;
     // 不同状态队列的键名
     failedQueue;
     pendingQueue;
@@ -58,6 +64,10 @@ export class RedisQueue {
         if (config.queueKey.length < 6) {
             throw new Error('[TaskQueue] queueKey must be at least 6 characters long');
         }
+        if (!config.handler) {
+            throw new Error('[TaskQueue] handler is required');
+        }
+        this.handler = config.handler;
         this.config = {
             queueKey: config.queueKey,
             redisUrl: config.redisUrl,
@@ -105,13 +115,7 @@ export class RedisQueue {
             });
         }
     }
-    /**
-     * 注册任务处理器
-     */
-    handle(type, handler) {
-        this.handlers.set(type, handler);
-    }
-    async enqueue(type, data) {
+    async enqueue(data) {
         if (!this.redis) {
             console.warn('[TaskQueue] Redis not available, skipping task enqueue');
             return Array.isArray(data) ? [] : '';
@@ -141,11 +145,10 @@ export class RedisQueue {
         for (const item of dataList) {
             // 检查 data 中是否包含自定义 id
             const customId = item.id;
-            const taskId = customId ? `${type}:${customId}` : `${type}:${randomUUID()}`;
+            const taskId = customId || randomUUID();
             const taskKey = `${this.config.queueKey}:task:${taskId}`;
             const task = {
                 id: taskId,
-                type,
                 data: item,
                 retryCount: 0,
                 maxRetries: this.config.maxRetries,
@@ -162,7 +165,7 @@ export class RedisQueue {
                 console.log(`[TaskQueue] Task already exists, skipping: ${taskId}`);
             }
             else {
-                console.log(`[TaskQueue] Enqueued task: ${task.type} (${task.id})`);
+                console.log(`[TaskQueue] Enqueued task: ${task.id}`);
             }
             taskIds.push(taskId);
         }
@@ -314,12 +317,6 @@ export class RedisQueue {
                 await this.applyStatus(taskId, task.status, 'failed');
                 return;
             }
-            const handler = this.handlers.get(task.type);
-            if (!handler) {
-                console.error(`[TaskQueue] No handler registered for task type: ${task.type}`);
-                await this.applyStatus(taskId, 'pending', 'failed');
-                return;
-            }
             try {
                 // 任务已在 processing 队列中(由 Lua 脚本完成),只需更新状态和开始时间
                 task.status = 'processing';
@@ -327,7 +324,7 @@ export class RedisQueue {
                 const taskKey = `${this.config.queueKey}:task:${taskId}`;
                 await this.redis.setEx(taskKey, this.config.cleanupDelay, JSON.stringify(task));
                 // 执行任务处理器
-                await handler(task.data);
+                await this.handler(task.data);
                 // 更新状态为完成
                 await this.applyStatus(taskId, 'processing', 'completed');
                 console.log(`[TaskQueue] Task completed: ${taskId}`);