llm-content-creator 0.1.0

package/dist/infrastructure/security/QuotaService.js

@@ -0,0 +1,303 @@
+/**
+ * 配额管理服务
+ *
+ * 提供用户配额检查、预留、消费和重置功能
+ * 使用乐观锁确保并发安全
+ */
+import crypto from 'crypto';
+import { createLogger } from '../logging/logger.js';
+import { BaseRepository } from '../database/BaseRepository.js';
+import { metricsService } from '../monitoring/MetricsService.js';
+const logger = createLogger('Quota');
+/**
+ * 配额服务类
+ */
+export class QuotaService extends BaseRepository {
+    defaultQuota = 100; // 默认每日配额
+    constructor() {
+        super();
+    }
+    /**
+     * 获取用户配额信息
+     */
+    async getUserQuota(userId) {
+        try {
+            // 先检查是否需要重置今日配额
+            await this.resetIfNeed(userId);
+            const result = await this.query(`SELECT user_id, quota_daily, quota_used_today, quota_reserved, last_reset_at
+         FROM users
+         WHERE user_id = $1`, [userId]);
+            if (result.rows.length === 0) {
+                logger.warn('User not found', { userId });
+                return null;
+            }
+            const row = result.rows[0];
+            const quotaDaily = row.quota_daily || this.defaultQuota;
+            const quotaUsedToday = row.quota_used_today || 0;
+            const quotaReserved = row.quota_reserved || 0;
+            const quotaAvailable = quotaDaily - quotaUsedToday - quotaReserved;
+            return {
+                userId: row.user_id,
+                quotaDaily,
+                quotaUsedToday,
+                quotaReserved,
+                quotaAvailable,
+                lastResetAt: row.last_reset_at,
+            };
+        }
+        catch (error) {
+            logger.error('Failed to get user quota', error, { userId });
+            return null;
+        }
+    }
+    /**
+     * 检查是否有足够配额
+     */
+    async checkQuota(userId, amount = 1) {
+        const quota = await this.getUserQuota(userId);
+        if (!quota) {
+            return false;
+        }
+        return quota.quotaAvailable >= amount;
+    }
+    /**
+     * 预留配额（使用乐观锁）
+     */
+    async reserveQuota(userId, amount, ttl = 300) {
+        try {
+            // 先检查是否有足够配额
+            const hasEnough = await this.checkQuota(userId, amount);
+            if (!hasEnough) {
+                logger.warn('Insufficient quota for reservation', { userId, amount });
+                return { success: false };
+            }
+            // 使用乐观锁预留配额
+            const result = await this.query(`UPDATE users
+         SET quota_reserved = quota_reserved + $1,
+             version = version + 1
+         WHERE user_id = $2
+           AND quota_daily - quota_used_today - quota_reserved >= $1
+         RETURNING quota_reserved, quota_daily, quota_used_today`, [amount, userId]);
+            if (result.rows.length === 0) {
+                logger.warn('Failed to reserve quota (concurrent modification)', { userId, amount });
+                return { success: false };
+            }
+            const reservationId = crypto.randomUUID();
+            const expiresAt = new Date(Date.now() + ttl * 1000);
+            // 记录预留
+            await this.query(`INSERT INTO quota_reservations (id, user_id, amount, created_at, expires_at)
+         VALUES ($1, $2, $3, NOW(), $4)`, [reservationId, userId, amount, expiresAt]);
+            logger.info('Quota reserved', {
+                reservationId,
+                userId,
+                amount,
+                expiresAt,
+            });
+            metricsService.recordCacheSet('quota_reservation');
+            return { success: true, reservationId };
+        }
+        catch (error) {
+            logger.error('Failed to reserve quota', error, { userId, amount });
+            return { success: false };
+        }
+    }
+    /**
+     * 消费预留的配额
+     */
+    async consumeQuota(userId, reservationId) {
+        try {
+            // 获取预留信息
+            const reservationResult = await this.query(`SELECT id, user_id, amount, expires_at
+         FROM quota_reservations
+         WHERE id = $1 AND user_id = $2 AND consumed = false`, [reservationId, userId]);
+            if (reservationResult.rows.length === 0) {
+                logger.warn('Quota reservation not found or already consumed', { reservationId, userId });
+                return false;
+            }
+            const reservation = reservationResult.rows[0];
+            // 检查是否过期
+            if (new Date(reservation.expires_at) < new Date()) {
+                logger.warn('Quota reservation expired', { reservationId, expiresAt: reservation.expires_at });
+                // 释放预留
+                await this.releaseQuota(userId, reservation.amount);
+                // 标记为已消费（虽然实际没消费）
+                await this.query(`UPDATE quota_reservations SET consumed = true WHERE id = $1`, [reservationId]);
+                return false;
+            }
+            // 使用事务确保原子性
+            await this.query('BEGIN', []);
+            try {
+                // 减少预留配额
+                await this.query(`UPDATE users
+           SET quota_reserved = quota_reserved - $1,
+               quota_used_today = quota_used_today + $1,
+               version = version + 1
+           WHERE user_id = $2`, [reservation.amount, userId]);
+                // 标记预留为已消费
+                await this.query(`UPDATE quota_reservations
+           SET consumed = true, consumed_at = NOW()
+           WHERE id = $1`, [reservationId]);
+                await this.query('COMMIT', []);
+                logger.info('Quota consumed', {
+                    reservationId,
+                    userId,
+                    amount: reservation.amount,
+                });
+                metricsService.recordCacheDelete('quota_reservation');
+                return true;
+            }
+            catch (error) {
+                await this.query('ROLLBACK', []);
+                throw error;
+            }
+        }
+        catch (error) {
+            logger.error('Failed to consume quota', error, { userId, reservationId });
+            return false;
+        }
+    }
+    /**
+     * 直接消费配额（无预留）
+     */
+    async consumeDirectly(userId, amount) {
+        try {
+            const result = await this.query(`UPDATE users
+         SET quota_used_today = quota_used_today + $1,
+             version = version + 1
+         WHERE user_id = $2
+           AND quota_daily - quota_used_today >= $1
+         RETURNING quota_used_today`, [amount, userId]);
+            if (result.rows.length === 0) {
+                logger.warn('Failed to consume quota directly (insufficient quota)', { userId, amount });
+                return false;
+            }
+            logger.info('Quota consumed directly', { userId, amount });
+            return true;
+        }
+        catch (error) {
+            logger.error('Failed to consume quota directly', error, { userId, amount });
+            return false;
+        }
+    }
+    /**
+     * 释放预留的配额
+     */
+    async releaseQuota(userId, amount) {
+        try {
+            await this.query(`UPDATE users
+         SET quota_reserved = GREATEST(0, quota_reserved - $1),
+             version = version + 1
+         WHERE user_id = $2`, [amount, userId]);
+            logger.info('Quota released', { userId, amount });
+            return true;
+        }
+        catch (error) {
+            logger.error('Failed to release quota', error, { userId, amount });
+            return false;
+        }
+    }
+    /**
+     * 重置用户今日配额
+     */
+    async resetIfNeed(userId) {
+        try {
+            const result = await this.query(`SELECT user_id, last_reset_at, quota_used_today
+         FROM users
+         WHERE user_id = $1`, [userId]);
+            if (result.rows.length === 0) {
+                return;
+            }
+            const lastReset = new Date(result.rows[0].last_reset_at);
+            const now = new Date();
+            // 如果上次重置不是今天，则重置
+            if (lastReset.toDateString() !== now.toDateString()) {
+                await this.query(`UPDATE users
+           SET quota_used_today = 0,
+               last_reset_at = NOW()
+           WHERE user_id = $1`, [userId]);
+                logger.info('User quota reset', { userId });
+            }
+        }
+        catch (error) {
+            logger.error('Failed to reset quota', error, { userId });
+        }
+    }
+    /**
+     * 手动重置用户配额
+     */
+    async resetUserQuota(userId) {
+        try {
+            await this.query(`UPDATE users
+         SET quota_used_today = 0,
+           quota_reserved = 0,
+           last_reset_at = NOW()
+         WHERE user_id = $1`, [userId]);
+            logger.info('User quota manually reset', { userId });
+            return true;
+        }
+        catch (error) {
+            logger.error('Failed to manually reset user quota', error, { userId });
+            return false;
+        }
+    }
+    /**
+     * 设置用户每日配额
+     */
+    async setUserQuota(userId, quotaDaily) {
+        try {
+            await this.query(`UPDATE users
+         SET quota_daily = $1
+         WHERE user_id = $2`, [quotaDaily, userId]);
+            logger.info('User daily quota updated', { userId, quotaDaily });
+            return true;
+        }
+        catch (error) {
+            logger.error('Failed to set user quota', error, { userId, quotaDaily });
+            return false;
+        }
+    }
+    /**
+     * 清理过期的配额预留
+     */
+    async cleanupExpiredReservations() {
+        try {
+            // 获取过期的预留
+            const result = await this.query(`SELECT id, user_id, amount
+         FROM quota_reservations
+         WHERE expires_at < NOW() AND consumed = false`, []);
+            const expired = result.rows;
+            // 释放预留
+            for (const row of expired) {
+                await this.releaseQuota(row.user_id, row.amount);
+            }
+            // 标记为已消费（实际未消费）
+            await this.query(`UPDATE quota_reservations
+         SET consumed = true
+         WHERE expires_at < NOW() AND consumed = false`, []);
+            logger.info('Expired quota reservations cleaned up', { count: expired.length });
+            return expired.length;
+        }
+        catch (error) {
+            logger.error('Failed to cleanup expired reservations', error);
+            return 0;
+        }
+    }
+    /**
+     * 健康检查
+     */
+    async healthCheck() {
+        try {
+            await this.query('SELECT 1', []);
+            return true;
+        }
+        catch (error) {
+            logger.error('Quota service health check failed', error);
+            return false;
+        }
+    }
+}
+/**
+ * 配额服务单例
+ */
+export const quotaService = new QuotaService();
+//# sourceMappingURL=QuotaService.js.map

package/dist/infrastructure/security/QuotaService.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"QuotaService.js","sourceRoot":"","sources":["../../../src/infrastructure/security/QuotaService.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAEH,OAAO,MAAM,MAAM,QAAQ,CAAC;AAC5B,OAAO,EAAE,YAAY,EAAE,MAAM,sBAAsB,CAAC;AACpD,OAAO,EAAE,cAAc,EAAE,MAAM,+BAA+B,CAAC;AAC/D,OAAO,EAAE,cAAc,EAAE,MAAM,iCAAiC,CAAC;AAEjE,MAAM,MAAM,GAAG,YAAY,CAAC,OAAO,CAAC,CAAC;AAyBrC;;GAEG;AACH,MAAM,OAAO,YAAa,SAAQ,cAAc;IACtC,YAAY,GAAW,GAAG,CAAC,CAAC,SAAS;IAE7C;QACE,KAAK,EAAE,CAAC;IACV,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,MAAc;QAC/B,IAAI,CAAC;YACH,gBAAgB;YAChB,MAAM,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YAE/B,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAC7B;;4BAEoB,EACpB,CAAC,MAAM,CAAC,CACT,CAAC;YAEF,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC7B,MAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;gBAC1C,OAAO,IAAI,CAAC;YACd,CAAC;YAED,MAAM,GAAG,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC,CAAQ,CAAC;YAClC,MAAM,UAAU,GAAG,GAAG,CAAC,WAAW,IAAI,IAAI,CAAC,YAAY,CAAC;YACxD,MAAM,cAAc,GAAG,GAAG,CAAC,gBAAgB,IAAI,CAAC,CAAC;YACjD,MAAM,aAAa,GAAG,GAAG,CAAC,cAAc,IAAI,CAAC,CAAC;YAC9C,MAAM,cAAc,GAAG,UAAU,GAAG,cAAc,GAAG,aAAa,CAAC;YAEnE,OAAO;gBACL,MAAM,EAAE,GAAG,CAAC,OAAO;gBACnB,UAAU;gBACV,cAAc;gBACd,aAAa;gBACb,cAAc;gBACd,WAAW,EAAE,GAAG,CAAC,aAAa;aAC/B,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAc,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YACrE,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU,CAAC,MAAc,EAAE,SAAiB,CAAC;QACjD,MAAM,KAAK,GAAG,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;QAE9C,IAAI,CAAC,KAAK,EAAE,CAAC;YACX,OAAO,KAAK,CAAC;QACf,CAAC;QAED,OAAO,KAAK,CAAC,cAAc,IAAI,MAAM,CAAC;IACxC,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,MAAc,EAAE,MAAc,GAAG;QAClE,IAAI,CAAC;YACH,aAAa;YACb,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,UAAU,CAAC,MAAM,EAAE,MAAM,CAAC,CAAC;YACxD,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,CAAC,IAAI,CAAC,oCAAoC,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;gBACtE,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YAC5B,CAAC;YAED,YAAY;YACZ,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAC7B;;;;;iEAKyD,EACzD,CAAC,MAAM,EAAE,MAAM,CAAC,CACjB,CAAC;YAEF,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC7B,MAAM,CAAC,IAAI,CAAC,mDAAmD,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;gBACrF,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;YAC5B,CAAC;YAED,MAAM,aAAa,GAAG,MAAM,CAAC,UAAU,EAAE,CAAC;YAC1C,MAAM,SAAS,GAAG,IAAI,IAAI,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,GAAG,GAAG,IAAI,CAAC,CAAC;YAEpD,OAAO;YACP,MAAM,IAAI,CAAC,KAAK,CACd;wCACgC,EAChC,CAAC,aAAa,EAAE,MAAM,EAAE,MAAM,EAAE,SAAS,CAAC,CAC3C,CAAC;YAEF,MAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE;gBAC5B,aAAa;gBACb,MAAM;gBACN,MAAM;gBACN,SAAS;aACV,CAAC,CAAC;YAEH,cAAc,CAAC,cAAc,CAAC,mBAAmB,CAAC,CAAC;YAEnD,OAAO,EAAE,OAAO,EAAE,IAAI,EAAE,aAAa,EAAE,CAAC;QAC1C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAc,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5E,OAAO,EAAE,OAAO,EAAE,KAAK,EAAE,CAAC;QAC5B,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,aAAqB;QACtD,IAAI,CAAC;YACH,SAAS;YACT,MAAM,iBAAiB,GAAG,MAAM,IAAI,CAAC,KAAK,CACxC;;6DAEqD,EACrD,CAAC,aAAa,EAAE,MAAM,CAAC,CACxB,CAAC;YAEF,IAAI,iBAAiB,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBACxC,MAAM,CAAC,IAAI,CAAC,iDAAiD,EAAE,EAAE,aAAa,EAAE,MAAM,EAAE,CAAC,CAAC;gBAC1F,OAAO,KAAK,CAAC;YACf,CAAC;YAED,MAAM,WAAW,GAAG,iBAAiB,CAAC,IAAI,CAAC,CAAC,CAAQ,CAAC;YAErD,SAAS;YACT,IAAI,IAAI,IAAI,CAAC,WAAW,CAAC,UAAU,CAAC,GAAG,IAAI,IAAI,EAAE,EAAE,CAAC;gBAClD,MAAM,CAAC,IAAI,CAAC,2BAA2B,EAAE,EAAE,aAAa,EAAE,SAAS,EAAE,WAAW,CAAC,UAAU,EAAE,CAAC,CAAC;gBAC/F,OAAO;gBACP,MAAM,IAAI,CAAC,YAAY,CAAC,MAAM,EAAE,WAAW,CAAC,MAAM,CAAC,CAAC;gBACpD,kBAAkB;gBAClB,MAAM,IAAI,CAAC,KAAK,CACd,6DAA6D,EAC7D,CAAC,aAAa,CAAC,CAChB,CAAC;gBACF,OAAO,KAAK,CAAC;YACf,CAAC;YAED,YAAY;YACZ,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,EAAE,EAAE,CAAC,CAAC;YAE9B,IAAI,CAAC;gBACH,SAAS;gBACT,MAAM,IAAI,CAAC,KAAK,CACd;;;;8BAIoB,EACpB,CAAC,WAAW,CAAC,MAAM,EAAE,MAAM,CAAC,CAC7B,CAAC;gBAEF,WAAW;gBACX,MAAM,IAAI,CAAC,KAAK,CACd;;yBAEe,EACf,CAAC,aAAa,CAAC,CAChB,CAAC;gBAEF,MAAM,IAAI,CAAC,KAAK,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;gBAE/B,MAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE;oBAC5B,aAAa;oBACb,MAAM;oBACN,MAAM,EAAE,WAAW,CAAC,MAAM;iBAC3B,CAAC,CAAC;gBAEH,cAAc,CAAC,iBAAiB,CAAC,mBAAmB,CAAC,CAAC;gBAEtD,OAAO,IAAI,CAAC;YACd,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;gBACjC,MAAM,KAAK,CAAC;YACd,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAc,EAAE,EAAE,MAAM,EAAE,aAAa,EAAE,CAAC,CAAC;YACnF,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CAAC,MAAc,EAAE,MAAc;QAClD,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAC7B;;;;;oCAK4B,EAC5B,CAAC,MAAM,EAAE,MAAM,CAAC,CACjB,CAAC;YAEF,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC7B,MAAM,CAAC,IAAI,CAAC,uDAAuD,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;gBACzF,OAAO,KAAK,CAAC;YACf,CAAC;YAED,MAAM,CAAC,IAAI,CAAC,yBAAyB,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;YAC3D,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,kCAAkC,EAAE,KAAc,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;YACrF,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,MAAc;QAC/C,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,KAAK,CACd;;;4BAGoB,EACpB,CAAC,MAAM,EAAE,MAAM,CAAC,CACjB,CAAC;YAEF,MAAM,CAAC,IAAI,CAAC,gBAAgB,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;YAClD,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,yBAAyB,EAAE,KAAc,EAAE,EAAE,MAAM,EAAE,MAAM,EAAE,CAAC,CAAC;YAC5E,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,WAAW,CAAC,MAAc;QACtC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAC7B;;4BAEoB,EACpB,CAAC,MAAM,CAAC,CACT,CAAC;YAEF,IAAI,MAAM,CAAC,IAAI,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;gBAC7B,OAAO;YACT,CAAC;YAED,MAAM,SAAS,GAAG,IAAI,IAAI,CAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAS,CAAC,aAAa,CAAC,CAAC;YAClE,MAAM,GAAG,GAAG,IAAI,IAAI,EAAE,CAAC;YAEvB,iBAAiB;YACjB,IAAI,SAAS,CAAC,YAAY,EAAE,KAAK,GAAG,CAAC,YAAY,EAAE,EAAE,CAAC;gBACpD,MAAM,IAAI,CAAC,KAAK,CACd;;;8BAGoB,EACpB,CAAC,MAAM,CAAC,CACT,CAAC;gBAEF,MAAM,CAAC,IAAI,CAAC,kBAAkB,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAC9C,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,uBAAuB,EAAE,KAAc,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;QACpE,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,cAAc,CAAC,MAAc;QACjC,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,KAAK,CACd;;;;4BAIoB,EACpB,CAAC,MAAM,CAAC,CACT,CAAC;YAEF,MAAM,CAAC,IAAI,CAAC,2BAA2B,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YACrD,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,qCAAqC,EAAE,KAAc,EAAE,EAAE,MAAM,EAAE,CAAC,CAAC;YAChF,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,YAAY,CAAC,MAAc,EAAE,UAAkB;QACnD,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,KAAK,CACd;;4BAEoB,EACpB,CAAC,UAAU,EAAE,MAAM,CAAC,CACrB,CAAC;YAEF,MAAM,CAAC,IAAI,CAAC,0BAA0B,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YAChE,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,0BAA0B,EAAE,KAAc,EAAE,EAAE,MAAM,EAAE,UAAU,EAAE,CAAC,CAAC;YACjF,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,0BAA0B;QAC9B,IAAI,CAAC;YACH,UAAU;YACV,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,KAAK,CAC7B;;uDAE+C,EAC/C,EAAE,CACH,CAAC;YAEF,MAAM,OAAO,GAAG,MAAM,CAAC,IAAa,CAAC;YAErC,OAAO;YACP,KAAK,MAAM,GAAG,IAAI,OAAO,EAAE,CAAC;gBAC1B,MAAM,IAAI,CAAC,YAAY,CAAC,GAAG,CAAC,OAAO,EAAE,GAAG,CAAC,MAAM,CAAC,CAAC;YACnD,CAAC;YAED,gBAAgB;YAChB,MAAM,IAAI,CAAC,KAAK,CACd;;uDAE+C,EAC/C,EAAE,CACH,CAAC;YAEF,MAAM,CAAC,IAAI,CAAC,uCAAuC,EAAE,EAAE,KAAK,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;YAChF,OAAO,OAAO,CAAC,MAAM,CAAC;QACxB,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,wCAAwC,EAAE,KAAc,CAAC,CAAC;YACvE,OAAO,CAAC,CAAC;QACX,CAAC;IACH,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW;QACf,IAAI,CAAC;YACH,MAAM,IAAI,CAAC,KAAK,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;YACjC,OAAO,IAAI,CAAC;QACd,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,CAAC,KAAK,CAAC,mCAAmC,EAAE,KAAc,CAAC,CAAC;YAClE,OAAO,KAAK,CAAC;QACf,CAAC;IACH,CAAC;CACF;AAED;;GAEG;AACH,MAAM,CAAC,MAAM,YAAY,GAAG,IAAI,YAAY,EAAE,CAAC"}

package/dist/infrastructure/security/RateLimiter.d.ts

@@ -0,0 +1,104 @@
+/**
+ * 速率限制服务
+ *
+ * 提供滑动窗口、令牌桶等速率限制算法
+ * 支持基于 IP、用户、API Key 的限流
+ */
+/**
+ * 速率限制结果
+ */
+export interface RateLimitResult {
+    allowed: boolean;
+    limit: number;
+    remaining: number;
+    resetTime: Date;
+    retryAfter?: number;
+}
+/**
+ * 速率限制配置
+ */
+export interface RateLimitConfig {
+    limit: number;
+    window: number;
+    burst?: number;
+}
+/**
+ * 速率限制器类型
+ */
+export type RateLimiterType = 'sliding-window' | 'token-bucket' | 'fixed-window';
+/**
+ * 速率限制服务类
+ */
+export declare class RateLimiter {
+    private redisClientWrapper;
+    private redis;
+    /**
+     * 获取 Redis 客户端
+     */
+    private getRedis;
+    /**
+     * 滑动窗口速率限制
+     */
+    slidingWindow(identifier: string, config: RateLimitConfig): Promise<RateLimitResult>;
+    /**
+     * 令牌桶速率限制
+     */
+    tokenBucket(identifier: string, config: Required<RateLimitConfig>): Promise<RateLimitResult>;
+    /**
+     * 固定窗口速率限制
+     */
+    fixedWindow(identifier: string, config: RateLimitConfig): Promise<RateLimitResult>;
+    /**
+     * 通用速率限制检查
+     */
+    checkLimit(identifier: string, config: RateLimitConfig, type?: RateLimiterType): Promise<RateLimitResult>;
+    /**
+     * 重置速率限制
+     */
+    resetLimit(identifier: string, type?: RateLimiterType): Promise<void>;
+    /**
+     * 获取当前速率限制状态
+     */
+    getStatus(identifier: string, type?: RateLimiterType): Promise<{
+        count: number;
+        limit: number;
+    } | null>;
+    /**
+     * 健康检查
+     */
+    healthCheck(): Promise<boolean>;
+    /**
+     * 关闭连接
+     */
+    close(): Promise<void>;
+}
+/**
+ * 速率限制服务单例
+ */
+export declare const rateLimiter: RateLimiter;
+/**
+ * 预定义的速率限制配置
+ */
+export declare const RateLimitPresets: {
+    api: {
+        limit: number;
+        window: number;
+    };
+    strict: {
+        limit: number;
+        window: number;
+    };
+    loose: {
+        limit: number;
+        window: number;
+    };
+    taskCreation: {
+        limit: number;
+        window: number;
+    };
+    llmCall: {
+        limit: number;
+        window: number;
+    };
+};
+//# sourceMappingURL=RateLimiter.d.ts.map

package/dist/infrastructure/security/RateLimiter.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"RateLimiter.d.ts","sourceRoot":"","sources":["../../../src/infrastructure/security/RateLimiter.ts"],"names":[],"mappings":"AAAA;;;;;GAKG;AAQH;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,OAAO,EAAE,OAAO,CAAC;IACjB,KAAK,EAAE,MAAM,CAAC;IACd,SAAS,EAAE,MAAM,CAAC;IAClB,SAAS,EAAE,IAAI,CAAC;IAChB,UAAU,CAAC,EAAE,MAAM,CAAC;CACrB;AAED;;GAEG;AACH,MAAM,WAAW,eAAe;IAC9B,KAAK,EAAE,MAAM,CAAC;IACd,MAAM,EAAE,MAAM,CAAC;IACf,KAAK,CAAC,EAAE,MAAM,CAAC;CAChB;AAED;;GAEG;AACH,MAAM,MAAM,eAAe,GAAG,gBAAgB,GAAG,cAAc,GAAG,cAAc,CAAC;AAEjF;;GAEG;AACH,qBAAa,WAAW;IACtB,OAAO,CAAC,kBAAkB,CAAe;IACzC,OAAO,CAAC,KAAK,CAAoB;IAEjC;;OAEG;YACW,QAAQ;IAOtB;;OAEG;IACG,aAAa,CACjB,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,eAAe,GACtB,OAAO,CAAC,eAAe,CAAC;IAgF3B;;OAEG;IACG,WAAW,CACf,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,QAAQ,CAAC,eAAe,CAAC,GAChC,OAAO,CAAC,eAAe,CAAC;IAyE3B;;OAEG;IACG,WAAW,CACf,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,eAAe,GACtB,OAAO,CAAC,eAAe,CAAC;IAiD3B;;OAEG;IACG,UAAU,CACd,UAAU,EAAE,MAAM,EAClB,MAAM,EAAE,eAAe,EACvB,IAAI,GAAE,eAAkC,GACvC,OAAO,CAAC,eAAe,CAAC;IAa3B;;OAEG;IACG,UAAU,CAAC,UAAU,EAAE,MAAM,EAAE,IAAI,GAAE,eAAkC,GAAG,OAAO,CAAC,IAAI,CAAC;IAgB7F;;OAEG;IACG,SAAS,CACb,UAAU,EAAE,MAAM,EAClB,IAAI,GAAE,eAAkC,GACvC,OAAO,CAAC;QAAE,KAAK,EAAE,MAAM,CAAC;QAAC,KAAK,EAAE,MAAM,CAAA;KAAE,GAAG,IAAI,CAAC;IA8BnD;;OAEG;IACG,WAAW,IAAI,OAAO,CAAC,OAAO,CAAC;IAWrC;;OAEG;IACG,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC;CAS7B;AAED;;GAEG;AACH,eAAO,MAAM,WAAW,aAAoB,CAAC;AAE7C;;GAEG;AACH,eAAO,MAAM,gBAAgB;;;;;;;;;;;;;;;;;;;;;CA8B5B,CAAC"}