@taicode/common-server 3.7.7 → 3.7.10

package/output/context/index.d.ts

@@ -0,0 +1,6 @@
+import type { FastifyBaseLogger, FastifyRequest } from 'fastify';
+export interface RequestContext {
+    log: FastifyBaseLogger;
+}
+export declare function createContext(request: FastifyRequest): RequestContext;
+//# sourceMappingURL=index.d.ts.map

package/output/context/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../source/context/index.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,iBAAiB,EAAE,cAAc,EAAE,MAAM,SAAS,CAAA;AAEhE,MAAM,WAAW,cAAc;IAC7B,GAAG,EAAE,iBAAiB,CAAA;CACvB;AAcD,wBAAgB,aAAa,CAAC,OAAO,EAAE,cAAc,GAAG,cAAc,CAErE"}

package/output/context/index.js

@@ -0,0 +1,14 @@
+const fallbackLogger = {
+    trace: () => { },
+    debug: () => { },
+    info: () => { },
+    warn: () => { },
+    error: () => { },
+    fatal: () => { },
+    child: () => fallbackLogger,
+    level: 'info',
+    silent: () => { },
+};
+export function createContext(request) {
+    return { log: request.log || fallbackLogger };
+}

package/output/context/index.test.d.ts

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=index.test.d.ts.map

package/output/context/index.test.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.test.d.ts","sourceRoot":"","sources":["../../source/context/index.test.ts"],"names":[],"mappings":""}

package/output/context/index.test.js

@@ -0,0 +1,14 @@
+import { describe, expect, it } from 'vitest';
+import { createContext } from './index.js';
+describe('createContext', () => {
+    it('uses the request logger when one is available', () => {
+        const log = { info: () => { } };
+        const context = createContext({ log });
+        expect(context.log).toBe(log);
+    });
+    it('provides a fallback logger for lightweight test requests', () => {
+        const context = createContext({ log: undefined });
+        expect(context.log.level).toBe('info');
+        expect(() => context.log.info('message')).not.toThrow();
+    });
+});

package/output/index.d.ts

@@ -0,0 +1,5 @@
+export * from './schema/index.js';
+export * from './redis-queue/index.js';
+export * from './rate-limiter/index.js';
+export * from './context/index.js';
+//# sourceMappingURL=index.d.ts.map

package/output/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../source/index.ts"],"names":[],"mappings":"AACA,cAAc,mBAAmB,CAAA;AACjC,cAAc,wBAAwB,CAAA;AACtC,cAAc,yBAAyB,CAAA;AACvC,cAAc,oBAAoB,CAAA"}

package/output/index.js

@@ -0,0 +1,5 @@
+// 根 index.ts，统一导出 mock 目录下所有能力
+export * from './schema/index.js';
+export * from './redis-queue/index.js';
+export * from './rate-limiter/index.js';
+export * from './context/index.js';

package/output/rate-limiter/index.d.ts

@@ -0,0 +1,4 @@
+export * from './types.js';
+export * from './rate-limiter.js';
+export * from './memory-store.js';
+//# sourceMappingURL=index.d.ts.map

package/output/rate-limiter/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../source/rate-limiter/index.ts"],"names":[],"mappings":"AAAA,cAAc,YAAY,CAAA;AAC1B,cAAc,mBAAmB,CAAA;AACjC,cAAc,mBAAmB,CAAA"}

package/output/rate-limiter/index.js

@@ -0,0 +1,3 @@
+export * from './types.js';
+export * from './rate-limiter.js';
+export * from './memory-store.js';

package/output/rate-limiter/memory-store.d.ts

@@ -0,0 +1,12 @@
+import type { RateLimitInfo, RateLimitStore } from './types.js';
+export declare class MemoryRateLimitStore implements RateLimitStore {
+    private store;
+    private cleanupInterval;
+    constructor(cleanupIntervalMs?: number);
+    get(key: string): Promise<RateLimitInfo | null>;
+    set(key: string, info: RateLimitInfo): Promise<void>;
+    delete(key: string): Promise<void>;
+    cleanup(): Promise<void>;
+    destroy(): void;
+}
+//# sourceMappingURL=memory-store.d.ts.map

package/output/rate-limiter/memory-store.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"memory-store.d.ts","sourceRoot":"","sources":["../../source/rate-limiter/memory-store.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,aAAa,EAAE,cAAc,EAAE,MAAM,YAAY,CAAA;AAO/D,qBAAa,oBAAqB,YAAW,cAAc;IACzD,OAAO,CAAC,KAAK,CAAsC;IACnD,OAAO,CAAC,eAAe,CAA8B;gBAEzC,iBAAiB,SAAQ;IAY/B,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC;IAY/C,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC;IAKpD,MAAM,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAIlC,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAS9B,OAAO,IAAI,IAAI;CAOhB"}

package/output/rate-limiter/memory-store.js

@@ -0,0 +1,46 @@
+export class MemoryRateLimitStore {
+    store = new Map();
+    cleanupInterval = null;
+    constructor(cleanupIntervalMs = 60000) {
+        // 定期清理过期数据
+        this.cleanupInterval = setInterval(() => {
+            this.cleanup();
+        }, cleanupIntervalMs);
+        // 防止内存泄漏
+        if (this.cleanupInterval.unref) {
+            this.cleanupInterval.unref();
+        }
+    }
+    async get(key) {
+        const entry = this.store.get(key);
+        if (!entry)
+            return null;
+        if (Date.now() > entry.expiresAt) {
+            this.store.delete(key);
+            return null;
+        }
+        return entry.info;
+    }
+    async set(key, info) {
+        const expiresAt = info.windowEnd.getTime();
+        this.store.set(key, { info, expiresAt });
+    }
+    async delete(key) {
+        this.store.delete(key);
+    }
+    async cleanup() {
+        const now = Date.now();
+        for (const [key, entry] of this.store.entries()) {
+            if (now > entry.expiresAt) {
+                this.store.delete(key);
+            }
+        }
+    }
+    destroy() {
+        if (this.cleanupInterval) {
+            clearInterval(this.cleanupInterval);
+            this.cleanupInterval = null;
+        }
+        this.store.clear();
+    }
+}

package/output/rate-limiter/rate-limiter.d.ts

@@ -0,0 +1,17 @@
+import type { RateLimitConfig, RateLimitInfo, RateLimiterOptions, RateLimitResult } from './types.js';
+export declare class RateLimiter {
+    private store;
+    constructor(options?: RateLimiterOptions);
+    checkLimit(key: string, config: RateLimitConfig): Promise<RateLimitResult>;
+    resetLimit(key: string): Promise<void>;
+    getLimitInfo(key: string): Promise<RateLimitInfo | null>;
+    /**
+     * 检查多个限流键，取最严格的限制
+     */
+    checkMultiLimit(keys: string[], config: RateLimitConfig): Promise<RateLimitResult>;
+}
+/**
+ * 创建限流键的辅助函数
+ */
+export declare function createRateLimitKey(prefix: string, ...parts: string[]): string;
+//# sourceMappingURL=rate-limiter.d.ts.map

package/output/rate-limiter/rate-limiter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"rate-limiter.d.ts","sourceRoot":"","sources":["../../source/rate-limiter/rate-limiter.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,eAAe,EAAE,aAAa,EAAE,kBAAkB,EAAE,eAAe,EAAkB,MAAM,YAAY,CAAA;AAGrH,qBAAa,WAAW;IACtB,OAAO,CAAC,KAAK,CAAgB;gBAEjB,OAAO,GAAE,kBAAuB;IAItC,UAAU,CACd,GAAG,EAAE,MAAM,EACX,MAAM,EAAE,eAAe,GACtB,OAAO,CAAC,eAAe,CAAC;IAoCrB,UAAU,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAItC,YAAY,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC;IAI9D;;OAEG;IACG,eAAe,CACnB,IAAI,EAAE,MAAM,EAAE,EACd,MAAM,EAAE,eAAe,GACtB,OAAO,CAAC,eAAe,CAAC;CAe5B;AAED;;GAEG;AACH,wBAAgB,kBAAkB,CAAC,MAAM,EAAE,MAAM,EAAE,GAAG,KAAK,EAAE,MAAM,EAAE,GAAG,MAAM,CAE7E"}

package/output/rate-limiter/rate-limiter.js

@@ -0,0 +1,64 @@
+import { MemoryRateLimitStore } from './memory-store.js';
+export class RateLimiter {
+    store;
+    constructor(options = {}) {
+        this.store = options.store || new MemoryRateLimitStore();
+    }
+    async checkLimit(key, config) {
+        const now = new Date();
+        const windowStart = new Date(now.getTime());
+        const windowEnd = new Date(now.getTime() + config.window * 1000);
+        // 获取当前计数
+        let currentInfo = await this.store.get(key);
+        // 如果没有记录，或者窗口已过期，创建新记录
+        if (!currentInfo || currentInfo.windowEnd < now) {
+            currentInfo = {
+                count: 0,
+                windowStart,
+                windowEnd,
+            };
+        }
+        // 增加计数
+        currentInfo.count += 1;
+        // 保存到存储
+        await this.store.set(key, currentInfo);
+        // 计算是否允许
+        const allowed = currentInfo.count <= config.limit;
+        const remaining = Math.max(0, config.limit - currentInfo.count);
+        return {
+            allowed,
+            remaining,
+            resetAt: currentInfo.windowEnd,
+            limit: config.limit,
+            window: config.window,
+        };
+    }
+    async resetLimit(key) {
+        await this.store.delete(key);
+    }
+    async getLimitInfo(key) {
+        return await this.store.get(key);
+    }
+    /**
+     * 检查多个限流键，取最严格的限制
+     */
+    async checkMultiLimit(keys, config) {
+        let strictestResult = null;
+        for (const key of keys) {
+            const result = await this.checkLimit(key, config);
+            if (!strictestResult || !result.allowed) {
+                strictestResult = result;
+            }
+            else if (result.remaining < strictestResult.remaining) {
+                strictestResult = result;
+            }
+        }
+        return strictestResult;
+    }
+}
+/**
+ * 创建限流键的辅助函数
+ */
+export function createRateLimitKey(prefix, ...parts) {
+    return `rate-limit:${prefix}:${parts.join(':')}`;
+}

package/output/rate-limiter/types.d.ts

@@ -0,0 +1,27 @@
+export interface RateLimitInfo {
+    count: number;
+    windowStart: Date;
+    windowEnd: Date;
+}
+export interface RateLimitResult {
+    allowed: boolean;
+    remaining: number;
+    resetAt: Date;
+    limit: number;
+    window: number;
+}
+export interface RateLimitStore {
+    get(key: string): Promise<RateLimitInfo | null>;
+    set(key: string, info: RateLimitInfo): Promise<void>;
+    delete(key: string): Promise<void>;
+    cleanup?(): Promise<void>;
+}
+export interface RateLimitConfig {
+    limit: number;
+    window: number;
+    lockout?: number;
+}
+export interface RateLimiterOptions {
+    store?: RateLimitStore;
+}
+//# sourceMappingURL=types.d.ts.map

package/output/rate-limiter/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../source/rate-limiter/types.ts"],"names":[],"mappings":"AAAA,MAAM,WAAW,aAAa;IAC5B,KAAK,EAAE,MAAM,CAAA;IACb,WAAW,EAAE,IAAI,CAAA;IACjB,SAAS,EAAE,IAAI,CAAA;CAChB;AAED,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,OAAO,CAAA;IAChB,SAAS,EAAE,MAAM,CAAA;IACjB,OAAO,EAAE,IAAI,CAAA;IACb,KAAK,EAAE,MAAM,CAAA;IACb,MAAM,EAAE,MAAM,CAAA;CACf;AAED,MAAM,WAAW,cAAc;IAC7B,GAAG,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,aAAa,GAAG,IAAI,CAAC,CAAA;IAC/C,GAAG,CAAC,GAAG,EAAE,MAAM,EAAE,IAAI,EAAE,aAAa,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IACpD,MAAM,CAAC,GAAG,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC,CAAA;IAClC,OAAO,CAAC,IAAI,OAAO,CAAC,IAAI,CAAC,CAAA;CAC1B;AAED,MAAM,WAAW,eAAe;IAC9B,KAAK,EAAE,MAAM,CAAA;IACb,MAAM,EAAE,MAAM,CAAA;IACd,OAAO,CAAC,EAAE,MAAM,CAAA;CACjB;AAED,MAAM,WAAW,kBAAkB;IACjC,KAAK,CAAC,EAAE,cAAc,CAAA;CACvB"}

package/output/rate-limiter/types.js

@@ -0,0 +1 @@
+export {};

package/output/redis-queue/index.d.ts

@@ -0,0 +1,8 @@
+export { RedisQueueProvider } from './redis-queue-provider.js';
+export { RedisQueueConsumer } from './redis-queue-consumer.js';
+export { RedisQueueBatchConsumer } from './redis-queue-batch-consumer.js';
+export type { RedisQueueProviderConfig, RedisQueueConsumerConfig, BatchConsumerConfig } from './types.js';
+export type { TaskData } from './types.js';
+export type { RedisQueueRegistry } from './types.js';
+export * as test from './test-helpers.js';
+//# sourceMappingURL=index.d.ts.map

package/output/redis-queue/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../source/redis-queue/index.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAA;AAC9D,OAAO,EAAE,kBAAkB,EAAE,MAAM,2BAA2B,CAAA;AAC9D,OAAO,EAAE,uBAAuB,EAAE,MAAM,iCAAiC,CAAA;AAGzE,YAAY,EAAE,wBAAwB,EAAE,wBAAwB,EAAE,mBAAmB,EAAE,MAAM,YAAY,CAAA;AAGzG,YAAY,EAAE,QAAQ,EAAE,MAAM,YAAY,CAAA;AAG1C,YAAY,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAA;AAGpD,OAAO,KAAK,IAAI,MAAM,mBAAmB,CAAA"}

package/output/redis-queue/index.js

@@ -0,0 +1,6 @@
+// 核心类导出
+export { RedisQueueProvider } from './redis-queue-provider.js';
+export { RedisQueueConsumer } from './redis-queue-consumer.js';
+export { RedisQueueBatchConsumer } from './redis-queue-batch-consumer.js';
+// 测试辅助函数导出
+export * as test from './test-helpers.js';

package/output/redis-queue/redis-queue-batch-consumer.d.ts

@@ -0,0 +1,97 @@
+import { RedisQueueCommon } from './redis-queue-common.js';
+import type { BatchConsumerConfig, QueueStats, RedisQueueRegistry } from './types.js';
+/**
+ * Redis 批量队列消费者
+ *
+ * 批量消费任务,每次处理多条
+ *
+ * @template R Registry 类型,默认为 RedisQueueRegistry
+ * @template K 队列键名类型
+ *
+ * @example
+ * ```ts
+ * // 方式 1: 使用默认 RedisQueueRegistry
+ * const consumer = new RedisQueueBatchConsumer('email-queue', {
+ *   redisUrl: 'redis://localhost:6379',
+ *   batchSize: 50,
+ *   handler: async (dataList) => {
+ *     await sendEmailsBatch(dataList.map(d => d.to))
+ *     return catchIt(() => {})
+ *   }
+ * })
+ *
+ * // 方式 2: 使用自定义 Registry
+ * interface MyRegistry {
+ *   'order-queue': { orderId: string; amount: number }
+ * }
+ *
+ * const consumer = new RedisQueueBatchConsumer<MyRegistry, 'order-queue'>(
+ *   'order-queue',
+ *   {
+ *     redisUrl: 'redis://localhost:6379',
+ *     batchSize: 50,
+ *     handler: async (dataList) => {
+ *       const total = dataList.reduce((sum, d) => sum + d.amount, 0)
+ *       logger.info('Total amount:', total)
+ *       return catchIt(() => {})
+ *     }
+ *   }
+ * )
+ *
+ * await consumer.connect() // 自动开始消费
+ *
+ * // 获取统计信息
+ * const stats = await consumer.statistics()
+ *
+ * // 停止消费
+ * consumer.disconnect()
+ * ```
+ */
+export declare class RedisQueueBatchConsumer<R extends Record<string, any> = RedisQueueRegistry, K extends keyof R & string = keyof R & string> extends RedisQueueCommon {
+    private consumerRunning;
+    private consumerInterval;
+    private recoveryInterval;
+    private processingBatches;
+    private processingPromises;
+    private readonly config;
+    constructor(queueKey: K, config: BatchConsumerConfig<R[K]>);
+    protected getLogPrefix(): string;
+    /**
+     * 连接 Redis 并自动启动消费者
+     */
+    connect(): Promise<void>;
+    /**
+     * 断开 Redis 连接并停止消费者
+     * 会等待所有正在处理的批次完成后再断开连接
+     */
+    disconnect(): Promise<void>;
+    /**
+     * 批量处理任务
+     */
+    private processBatch;
+    /**
+     * 启动恢复机制
+     */
+    private startRecovery;
+    /**
+     * 停止恢复机制
+     */
+    private stopRecovery;
+    /**
+     * 启动消费者
+     */
+    private startConsumer;
+    /**
+     * 停止消费者
+     */
+    private stopConsumer;
+    /**
+     * 获取队列统计信息（O(1) 时间复杂度）
+     */
+    statistics(): Promise<QueueStats>;
+    /**
+     * 健康检查
+     */
+    health(): Promise<boolean>;
+}
+//# sourceMappingURL=redis-queue-batch-consumer.d.ts.map

package/output/redis-queue/redis-queue-batch-consumer.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"redis-queue-batch-consumer.d.ts","sourceRoot":"","sources":["../../source/redis-queue/redis-queue-batch-consumer.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,gBAAgB,EAAE,MAAM,yBAAyB,CAAA;AAC1D,OAAO,KAAK,EAAE,mBAAmB,EAA0B,UAAU,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAA;AAE7G;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;GA8CG;AACH,qBAAa,uBAAuB,CAClC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,kBAAkB,EAClD,CAAC,SAAS,MAAM,CAAC,GAAG,MAAM,GAAG,MAAM,CAAC,GAAG,MAAM,CAC7C,SAAQ,gBAAgB;IACxB,OAAO,CAAC,eAAe,CAAQ;IAC/B,OAAO,CAAC,gBAAgB,CAA8B;IACtD,OAAO,CAAC,gBAAgB,CAA8B;IACtD,OAAO,CAAC,iBAAiB,CAAI;IAC7B,OAAO,CAAC,kBAAkB,CAA2B;IAErD,OAAO,CAAC,QAAQ,CAAC,MAAM,CAOtB;gBAEW,QAAQ,EAAE,CAAC,EAAE,MAAM,EAAE,mBAAmB,CAAC,CAAC,CAAC,CAAC,CAAC,CAAC;IAuB1D,SAAS,CAAC,YAAY,IAAI,MAAM;IAIhC;;OAEG;IACG,OAAO,IAAI,OAAO,CAAC,IAAI,CAAC;IAO9B;;;OAGG;IACG,UAAU,IAAI,OAAO,CAAC,IAAI,CAAC;IAgBjC;;OAEG;YACW,YAAY;IA2G1B;;OAEG;IACH,OAAO,CAAC,aAAa;IAoBrB;;OAEG;IACH,OAAO,CAAC,YAAY;IAQpB;;OAEG;IACH,OAAO,CAAC,aAAa;IAyErB;;OAEG;IACH,OAAO,CAAC,YAAY;IASpB;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,UAAU,CAAC;IAevC;;OAEG;IACG,MAAM,IAAI,OAAO,CAAC,OAAO,CAAC;CAMjC"}