@roomkit/gateway 1.0.0

package/dist/src/metrics/metrics.service.d.ts

@@ -0,0 +1,81 @@
+import { OnModuleInit } from '@nestjs/common';
+import { Counter, Gauge, Histogram } from 'prom-client';
+import { ConfigService } from '../config/config.service';
+/**
+ * Gateway Prometheus 指标服务
+ * 提供连接、消息、限流等监控指标
+ */
+export declare class MetricsService implements OnModuleInit {
+    private readonly configService;
+    private readonly registry;
+    readonly connectionsTotal: Counter;
+    readonly connectionsCurrent: Gauge;
+    readonly connectionsAuthenticated: Gauge;
+    readonly connectionDuration: Histogram;
+    readonly messagesReceived: Counter;
+    readonly messagesSent: Counter;
+    readonly messageProcessingTime: Histogram;
+    readonly messagesByType: Counter;
+    readonly rateLimitRejections: Counter;
+    readonly connectionRateLimitHits: Counter;
+    readonly messageRateLimitHits: Counter;
+    readonly uptime: Gauge;
+    readonly memoryUsage: Gauge;
+    readonly wsErrors: Counter;
+    readonly wsReconnections: Counter;
+    constructor(configService: ConfigService);
+    onModuleInit(): void;
+    /**
+     * 启动定期指标收集
+     */
+    private startMetricsCollection;
+    /**
+     * 获取所有指标（Prometheus 格式）
+     */
+    getMetrics(): Promise<string>;
+    /**
+     * 获取指标 Content-Type
+     */
+    getContentType(): string;
+    /**
+     * 记录新连接
+     */
+    recordNewConnection(): void;
+    /**
+     * 记录连接断开
+     */
+    recordDisconnection(durationSeconds: number): void;
+    /**
+     * 记录连接被拒绝
+     */
+    recordConnectionRejected(reason: string): void;
+    /**
+     * 更新当前连接数
+     */
+    updateCurrentConnections(total: number, authenticated: number): void;
+    /**
+     * 记录收到的消息
+     */
+    recordMessageReceived(msgId: number): void;
+    /**
+     * 记录发送的消息
+     */
+    recordMessageSent(msgId: number): void;
+    /**
+     * 记录消息处理时间
+     */
+    recordMessageProcessingTime(msgId: number, durationSeconds: number): void;
+    /**
+     * 记录消息限流
+     */
+    recordMessageRateLimited(connectionId: string): void;
+    /**
+     * 记录全局限流
+     */
+    recordGlobalRateLimited(): void;
+    /**
+     * 记录 WebSocket 错误
+     */
+    recordWsError(errorType: string): void;
+}
+//# sourceMappingURL=metrics.service.d.ts.map

package/dist/src/metrics/metrics.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"metrics.service.d.ts","sourceRoot":"","sources":["../../../src/metrics/metrics.service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAc,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC1D,OAAO,EAEL,OAAO,EACP,KAAK,EACL,SAAS,EAEV,MAAM,aAAa,CAAC;AACrB,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAEzD;;;GAGG;AACH,qBACa,cAAe,YAAW,YAAY;IA4BrC,OAAO,CAAC,QAAQ,CAAC,aAAa;IA3B1C,OAAO,CAAC,QAAQ,CAAC,QAAQ,CAAW;IAGpC,QAAQ,CAAC,gBAAgB,EAAE,OAAO,CAAC;IACnC,QAAQ,CAAC,kBAAkB,EAAE,KAAK,CAAC;IACnC,QAAQ,CAAC,wBAAwB,EAAE,KAAK,CAAC;IACzC,QAAQ,CAAC,kBAAkB,EAAE,SAAS,CAAC;IAGvC,QAAQ,CAAC,gBAAgB,EAAE,OAAO,CAAC;IACnC,QAAQ,CAAC,YAAY,EAAE,OAAO,CAAC;IAC/B,QAAQ,CAAC,qBAAqB,EAAE,SAAS,CAAC;IAC1C,QAAQ,CAAC,cAAc,EAAE,OAAO,CAAC;IAGjC,QAAQ,CAAC,mBAAmB,EAAE,OAAO,CAAC;IACtC,QAAQ,CAAC,uBAAuB,EAAE,OAAO,CAAC;IAC1C,QAAQ,CAAC,oBAAoB,EAAE,OAAO,CAAC;IAGvC,QAAQ,CAAC,MAAM,EAAE,KAAK,CAAC;IACvB,QAAQ,CAAC,WAAW,EAAE,KAAK,CAAC;IAG5B,QAAQ,CAAC,QAAQ,EAAE,OAAO,CAAC;IAC3B,QAAQ,CAAC,eAAe,EAAE,OAAO,CAAC;gBAEL,aAAa,EAAE,aAAa;IAoHzD,YAAY;IAWZ;;OAEG;IACH,OAAO,CAAC,sBAAsB;IAgB9B;;OAEG;IACG,UAAU,IAAI,OAAO,CAAC,MAAM,CAAC;IAInC;;OAEG;IACH,cAAc,IAAI,MAAM;IAIxB;;OAEG;IACH,mBAAmB;IAInB;;OAEG;IACH,mBAAmB,CAAC,eAAe,EAAE,MAAM;IAK3C;;OAEG;IACH,wBAAwB,CAAC,MAAM,EAAE,MAAM;IAQvC;;OAEG;IACH,wBAAwB,CAAC,KAAK,EAAE,MAAM,EAAE,aAAa,EAAE,MAAM;IAK7D;;OAEG;IACH,qBAAqB,CAAC,KAAK,EAAE,MAAM;IAKnC;;OAEG;IACH,iBAAiB,CAAC,KAAK,EAAE,MAAM;IAK/B;;OAEG;IACH,2BAA2B,CAAC,KAAK,EAAE,MAAM,EAAE,eAAe,EAAE,MAAM;IAIlE;;OAEG;IACH,wBAAwB,CAAC,YAAY,EAAE,MAAM;IAK7C;;OAEG;IACH,uBAAuB;IAIvB;;OAEG;IACH,aAAa,CAAC,SAAS,EAAE,MAAM;CAGhC"}

package/dist/src/metrics/metrics.service.js

@@ -0,0 +1,255 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.MetricsService = void 0;
+const common_1 = require("@nestjs/common");
+const prom_client_1 = require("prom-client");
+const config_service_1 = require("../config/config.service");
+/**
+ * Gateway Prometheus 指标服务
+ * 提供连接、消息、限流等监控指标
+ */
+let MetricsService = class MetricsService {
+    configService;
+    registry;
+    // 连接指标
+    connectionsTotal;
+    connectionsCurrent;
+    connectionsAuthenticated;
+    connectionDuration;
+    // 消息指标
+    messagesReceived;
+    messagesSent;
+    messageProcessingTime;
+    messagesByType;
+    // 限流指标
+    rateLimitRejections;
+    connectionRateLimitHits;
+    messageRateLimitHits;
+    // 系统指标
+    uptime;
+    memoryUsage;
+    // WebSocket 指标
+    wsErrors;
+    wsReconnections;
+    constructor(configService) {
+        this.configService = configService;
+        this.registry = new prom_client_1.Registry();
+        // 设置默认标签
+        this.registry.setDefaultLabels({
+            app: 'gateway',
+            instance: this.configService.gatewayId,
+        });
+        // 连接指标
+        this.connectionsTotal = new prom_client_1.Counter({
+            name: 'gateway_connections_total',
+            help: 'Total number of WebSocket connections',
+            labelNames: ['status'], // connected, disconnected, rejected
+            registers: [this.registry],
+        });
+        this.connectionsCurrent = new prom_client_1.Gauge({
+            name: 'gateway_connections_current',
+            help: 'Current number of active WebSocket connections',
+            registers: [this.registry],
+        });
+        this.connectionsAuthenticated = new prom_client_1.Gauge({
+            name: 'gateway_connections_authenticated',
+            help: 'Number of authenticated connections',
+            registers: [this.registry],
+        });
+        this.connectionDuration = new prom_client_1.Histogram({
+            name: 'gateway_connection_duration_seconds',
+            help: 'Duration of WebSocket connections in seconds',
+            buckets: [1, 5, 15, 30, 60, 120, 300, 600, 1800, 3600],
+            registers: [this.registry],
+        });
+        // 消息指标
+        this.messagesReceived = new prom_client_1.Counter({
+            name: 'gateway_messages_received_total',
+            help: 'Total number of messages received from clients',
+            labelNames: ['msg_type'],
+            registers: [this.registry],
+        });
+        this.messagesSent = new prom_client_1.Counter({
+            name: 'gateway_messages_sent_total',
+            help: 'Total number of messages sent to clients',
+            labelNames: ['msg_type'],
+            registers: [this.registry],
+        });
+        this.messageProcessingTime = new prom_client_1.Histogram({
+            name: 'gateway_message_processing_seconds',
+            help: 'Time spent processing messages',
+            buckets: [0.001, 0.005, 0.01, 0.025, 0.05, 0.1, 0.25, 0.5, 1],
+            labelNames: ['msg_type'],
+            registers: [this.registry],
+        });
+        this.messagesByType = new prom_client_1.Counter({
+            name: 'gateway_messages_by_type_total',
+            help: 'Messages count by message type',
+            labelNames: ['direction', 'msg_id'],
+            registers: [this.registry],
+        });
+        // 限流指标
+        this.rateLimitRejections = new prom_client_1.Counter({
+            name: 'gateway_rate_limit_rejections_total',
+            help: 'Total number of rate limit rejections',
+            labelNames: ['type'], // connection, message, global
+            registers: [this.registry],
+        });
+        this.connectionRateLimitHits = new prom_client_1.Counter({
+            name: 'gateway_connection_rate_limit_hits_total',
+            help: 'Number of connection rate limit hits',
+            registers: [this.registry],
+        });
+        this.messageRateLimitHits = new prom_client_1.Counter({
+            name: 'gateway_message_rate_limit_hits_total',
+            help: 'Number of message rate limit hits',
+            labelNames: ['connection_id'],
+            registers: [this.registry],
+        });
+        // 系统指标
+        this.uptime = new prom_client_1.Gauge({
+            name: 'gateway_uptime_seconds',
+            help: 'Gateway uptime in seconds',
+            registers: [this.registry],
+        });
+        this.memoryUsage = new prom_client_1.Gauge({
+            name: 'gateway_memory_usage_bytes',
+            help: 'Memory usage in bytes',
+            labelNames: ['type'], // heap_used, heap_total, rss, external
+            registers: [this.registry],
+        });
+        // WebSocket 指标
+        this.wsErrors = new prom_client_1.Counter({
+            name: 'gateway_websocket_errors_total',
+            help: 'Total number of WebSocket errors',
+            labelNames: ['error_type'],
+            registers: [this.registry],
+        });
+        this.wsReconnections = new prom_client_1.Counter({
+            name: 'gateway_websocket_reconnections_total',
+            help: 'Total number of WebSocket reconnection attempts',
+            registers: [this.registry],
+        });
+    }
+    onModuleInit() {
+        // 收集默认的 Node.js 指标
+        (0, prom_client_1.collectDefaultMetrics)({
+            register: this.registry,
+            prefix: 'gateway_nodejs_',
+        });
+        // 定期更新系统指标
+        this.startMetricsCollection();
+    }
+    /**
+     * 启动定期指标收集
+     */
+    startMetricsCollection() {
+        const startTime = Date.now();
+        setInterval(() => {
+            // 更新运行时间
+            this.uptime.set((Date.now() - startTime) / 1000);
+            // 更新内存使用
+            const memory = process.memoryUsage();
+            this.memoryUsage.labels('heap_used').set(memory.heapUsed);
+            this.memoryUsage.labels('heap_total').set(memory.heapTotal);
+            this.memoryUsage.labels('rss').set(memory.rss);
+            this.memoryUsage.labels('external').set(memory.external);
+        }, 5000);
+    }
+    /**
+     * 获取所有指标（Prometheus 格式）
+     */
+    async getMetrics() {
+        return this.registry.metrics();
+    }
+    /**
+     * 获取指标 Content-Type
+     */
+    getContentType() {
+        return this.registry.contentType;
+    }
+    /**
+     * 记录新连接
+     */
+    recordNewConnection() {
+        this.connectionsTotal.labels('connected').inc();
+    }
+    /**
+     * 记录连接断开
+     */
+    recordDisconnection(durationSeconds) {
+        this.connectionsTotal.labels('disconnected').inc();
+        this.connectionDuration.observe(durationSeconds);
+    }
+    /**
+     * 记录连接被拒绝
+     */
+    recordConnectionRejected(reason) {
+        this.connectionsTotal.labels('rejected').inc();
+        if (reason === 'rate_limit') {
+            this.rateLimitRejections.labels('connection').inc();
+            this.connectionRateLimitHits.inc();
+        }
+    }
+    /**
+     * 更新当前连接数
+     */
+    updateCurrentConnections(total, authenticated) {
+        this.connectionsCurrent.set(total);
+        this.connectionsAuthenticated.set(authenticated);
+    }
+    /**
+     * 记录收到的消息
+     */
+    recordMessageReceived(msgId) {
+        this.messagesReceived.labels(String(msgId)).inc();
+        this.messagesByType.labels('inbound', String(msgId)).inc();
+    }
+    /**
+     * 记录发送的消息
+     */
+    recordMessageSent(msgId) {
+        this.messagesSent.labels(String(msgId)).inc();
+        this.messagesByType.labels('outbound', String(msgId)).inc();
+    }
+    /**
+     * 记录消息处理时间
+     */
+    recordMessageProcessingTime(msgId, durationSeconds) {
+        this.messageProcessingTime.labels(String(msgId)).observe(durationSeconds);
+    }
+    /**
+     * 记录消息限流
+     */
+    recordMessageRateLimited(connectionId) {
+        this.rateLimitRejections.labels('message').inc();
+        this.messageRateLimitHits.labels(connectionId).inc();
+    }
+    /**
+     * 记录全局限流
+     */
+    recordGlobalRateLimited() {
+        this.rateLimitRejections.labels('global').inc();
+    }
+    /**
+     * 记录 WebSocket 错误
+     */
+    recordWsError(errorType) {
+        this.wsErrors.labels(errorType).inc();
+    }
+};
+exports.MetricsService = MetricsService;
+exports.MetricsService = MetricsService = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [config_service_1.ConfigService])
+], MetricsService);
+//# sourceMappingURL=metrics.service.js.map

package/dist/src/metrics/metrics.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"metrics.service.js","sourceRoot":"","sources":["../../../src/metrics/metrics.service.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,2CAA0D;AAC1D,6CAMqB;AACrB,6DAAyD;AAEzD;;;GAGG;AAEI,IAAM,cAAc,GAApB,MAAM,cAAc;IA4BI;IA3BZ,QAAQ,CAAW;IAEpC,OAAO;IACE,gBAAgB,CAAU;IAC1B,kBAAkB,CAAQ;IAC1B,wBAAwB,CAAQ;IAChC,kBAAkB,CAAY;IAEvC,OAAO;IACE,gBAAgB,CAAU;IAC1B,YAAY,CAAU;IACtB,qBAAqB,CAAY;IACjC,cAAc,CAAU;IAEjC,OAAO;IACE,mBAAmB,CAAU;IAC7B,uBAAuB,CAAU;IACjC,oBAAoB,CAAU;IAEvC,OAAO;IACE,MAAM,CAAQ;IACd,WAAW,CAAQ;IAE5B,eAAe;IACN,QAAQ,CAAU;IAClB,eAAe,CAAU;IAElC,YAA6B,aAA4B;QAA5B,kBAAa,GAAb,aAAa,CAAe;QACvD,IAAI,CAAC,QAAQ,GAAG,IAAI,sBAAQ,EAAE,CAAC;QAE/B,SAAS;QACT,IAAI,CAAC,QAAQ,CAAC,gBAAgB,CAAC;YAC7B,GAAG,EAAE,SAAS;YACd,QAAQ,EAAE,IAAI,CAAC,aAAa,CAAC,SAAS;SACvC,CAAC,CAAC;QAEH,OAAO;QACP,IAAI,CAAC,gBAAgB,GAAG,IAAI,qBAAO,CAAC;YAClC,IAAI,EAAE,2BAA2B;YACjC,IAAI,EAAE,uCAAuC;YAC7C,UAAU,EAAE,CAAC,QAAQ,CAAC,EAAE,oCAAoC;YAC5D,SAAS,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,kBAAkB,GAAG,IAAI,mBAAK,CAAC;YAClC,IAAI,EAAE,6BAA6B;YACnC,IAAI,EAAE,gDAAgD;YACtD,SAAS,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,wBAAwB,GAAG,IAAI,mBAAK,CAAC;YACxC,IAAI,EAAE,mCAAmC;YACzC,IAAI,EAAE,qCAAqC;YAC3C,SAAS,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,kBAAkB,GAAG,IAAI,uBAAS,CAAC;YACtC,IAAI,EAAE,qCAAqC;YAC3C,IAAI,EAAE,8CAA8C;YACpD,OAAO,EAAE,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,IAAI,EAAE,IAAI,CAAC;YACtD,SAAS,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;SAC3B,CAAC,CAAC;QAEH,OAAO;QACP,IAAI,CAAC,gBAAgB,GAAG,IAAI,qBAAO,CAAC;YAClC,IAAI,EAAE,iCAAiC;YACvC,IAAI,EAAE,gDAAgD;YACtD,UAAU,EAAE,CAAC,UAAU,CAAC;YACxB,SAAS,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,YAAY,GAAG,IAAI,qBAAO,CAAC;YAC9B,IAAI,EAAE,6BAA6B;YACnC,IAAI,EAAE,0CAA0C;YAChD,UAAU,EAAE,CAAC,UAAU,CAAC;YACxB,SAAS,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,qBAAqB,GAAG,IAAI,uBAAS,CAAC;YACzC,IAAI,EAAE,oCAAoC;YAC1C,IAAI,EAAE,gCAAgC;YACtC,OAAO,EAAE,CAAC,KAAK,EAAE,KAAK,EAAE,IAAI,EAAE,KAAK,EAAE,IAAI,EAAE,GAAG,EAAE,IAAI,EAAE,GAAG,EAAE,CAAC,CAAC;YAC7D,UAAU,EAAE,CAAC,UAAU,CAAC;YACxB,SAAS,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,cAAc,GAAG,IAAI,qBAAO,CAAC;YAChC,IAAI,EAAE,gCAAgC;YACtC,IAAI,EAAE,gCAAgC;YACtC,UAAU,EAAE,CAAC,WAAW,EAAE,QAAQ,CAAC;YACnC,SAAS,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;SAC3B,CAAC,CAAC;QAEH,OAAO;QACP,IAAI,CAAC,mBAAmB,GAAG,IAAI,qBAAO,CAAC;YACrC,IAAI,EAAE,qCAAqC;YAC3C,IAAI,EAAE,uCAAuC;YAC7C,UAAU,EAAE,CAAC,MAAM,CAAC,EAAE,8BAA8B;YACpD,SAAS,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,uBAAuB,GAAG,IAAI,qBAAO,CAAC;YACzC,IAAI,EAAE,0CAA0C;YAChD,IAAI,EAAE,sCAAsC;YAC5C,SAAS,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,oBAAoB,GAAG,IAAI,qBAAO,CAAC;YACtC,IAAI,EAAE,uCAAuC;YAC7C,IAAI,EAAE,mCAAmC;YACzC,UAAU,EAAE,CAAC,eAAe,CAAC;YAC7B,SAAS,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;SAC3B,CAAC,CAAC;QAEH,OAAO;QACP,IAAI,CAAC,MAAM,GAAG,IAAI,mBAAK,CAAC;YACtB,IAAI,EAAE,wBAAwB;YAC9B,IAAI,EAAE,2BAA2B;YACjC,SAAS,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,GAAG,IAAI,mBAAK,CAAC;YAC3B,IAAI,EAAE,4BAA4B;YAClC,IAAI,EAAE,uBAAuB;YAC7B,UAAU,EAAE,CAAC,MAAM,CAAC,EAAE,uCAAuC;YAC7D,SAAS,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;SAC3B,CAAC,CAAC;QAEH,eAAe;QACf,IAAI,CAAC,QAAQ,GAAG,IAAI,qBAAO,CAAC;YAC1B,IAAI,EAAE,gCAAgC;YACtC,IAAI,EAAE,kCAAkC;YACxC,UAAU,EAAE,CAAC,YAAY,CAAC;YAC1B,SAAS,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;SAC3B,CAAC,CAAC;QAEH,IAAI,CAAC,eAAe,GAAG,IAAI,qBAAO,CAAC;YACjC,IAAI,EAAE,uCAAuC;YAC7C,IAAI,EAAE,iDAAiD;YACvD,SAAS,EAAE,CAAC,IAAI,CAAC,QAAQ,CAAC;SAC3B,CAAC,CAAC;IACL,CAAC;IAED,YAAY;QACV,mBAAmB;QACnB,IAAA,mCAAqB,EAAC;YACpB,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,MAAM,EAAE,iBAAiB;SAC1B,CAAC,CAAC;QAEH,WAAW;QACX,IAAI,CAAC,sBAAsB,EAAE,CAAC;IAChC,CAAC;IAED;;OAEG;IACK,sBAAsB;QAC5B,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,WAAW,CAAC,GAAG,EAAE;YACf,SAAS;YACT,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC,GAAG,IAAI,CAAC,CAAC;YAEjD,SAAS;YACT,MAAM,MAAM,GAAG,OAAO,CAAC,WAAW,EAAE,CAAC;YACrC,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAC1D,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC;YAC5D,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YAC/C,IAAI,CAAC,WAAW,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;QAC3D,CAAC,EAAE,IAAI,CAAC,CAAC;IACX,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,OAAO,IAAI,CAAC,QAAQ,CAAC,OAAO,EAAE,CAAC;IACjC,CAAC;IAED;;OAEG;IACH,cAAc;QACZ,OAAO,IAAI,CAAC,QAAQ,CAAC,WAAW,CAAC;IACnC,CAAC;IAED;;OAEG;IACH,mBAAmB;QACjB,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,WAAW,CAAC,CAAC,GAAG,EAAE,CAAC;IAClD,CAAC;IAED;;OAEG;IACH,mBAAmB,CAAC,eAAuB;QACzC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,cAAc,CAAC,CAAC,GAAG,EAAE,CAAC;QACnD,IAAI,CAAC,kBAAkB,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;IACnD,CAAC;IAED;;OAEG;IACH,wBAAwB,CAAC,MAAc;QACrC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,UAAU,CAAC,CAAC,GAAG,EAAE,CAAC;QAC/C,IAAI,MAAM,KAAK,YAAY,EAAE,CAAC;YAC5B,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,EAAE,CAAC;YACpD,IAAI,CAAC,uBAAuB,CAAC,GAAG,EAAE,CAAC;QACrC,CAAC;IACH,CAAC;IAED;;OAEG;IACH,wBAAwB,CAAC,KAAa,EAAE,aAAqB;QAC3D,IAAI,CAAC,kBAAkB,CAAC,GAAG,CAAC,KAAK,CAAC,CAAC;QACnC,IAAI,CAAC,wBAAwB,CAAC,GAAG,CAAC,aAAa,CAAC,CAAC;IACnD,CAAC;IAED;;OAEG;IACH,qBAAqB,CAAC,KAAa;QACjC,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;QAClD,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,SAAS,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;IAC7D,CAAC;IAED;;OAEG;IACH,iBAAiB,CAAC,KAAa;QAC7B,IAAI,CAAC,YAAY,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;QAC9C,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,UAAU,EAAE,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,GAAG,EAAE,CAAC;IAC9D,CAAC;IAED;;OAEG;IACH,2BAA2B,CAAC,KAAa,EAAE,eAAuB;QAChE,IAAI,CAAC,qBAAqB,CAAC,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,OAAO,CAAC,eAAe,CAAC,CAAC;IAC5E,CAAC;IAED;;OAEG;IACH,wBAAwB,CAAC,YAAoB;QAC3C,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,EAAE,CAAC;QACjD,IAAI,CAAC,oBAAoB,CAAC,MAAM,CAAC,YAAY,CAAC,CAAC,GAAG,EAAE,CAAC;IACvD,CAAC;IAED;;OAEG;IACH,uBAAuB;QACrB,IAAI,CAAC,mBAAmB,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC,GAAG,EAAE,CAAC;IAClD,CAAC;IAED;;OAEG;IACH,aAAa,CAAC,SAAiB;QAC7B,IAAI,CAAC,QAAQ,CAAC,MAAM,CAAC,SAAS,CAAC,CAAC,GAAG,EAAE,CAAC;IACxC,CAAC;CACF,CAAA;AA1QY,wCAAc;yBAAd,cAAc;IAD1B,IAAA,mBAAU,GAAE;qCA6BiC,8BAAa;GA5B9C,cAAc,CA0Q1B"}

package/dist/src/ratelimit/rate-limit.module.d.ts

@@ -0,0 +1,3 @@
+export declare class RateLimitModule {
+}
+//# sourceMappingURL=rate-limit.module.d.ts.map

package/dist/src/ratelimit/rate-limit.module.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"rate-limit.module.d.ts","sourceRoot":"","sources":["../../../src/ratelimit/rate-limit.module.ts"],"names":[],"mappings":"AAIA,qBAKa,eAAe;CAAG"}

package/dist/src/ratelimit/rate-limit.module.js

@@ -0,0 +1,23 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RateLimitModule = void 0;
+const common_1 = require("@nestjs/common");
+const rate_limiter_service_1 = require("./rate-limiter.service");
+const config_module_1 = require("../config/config.module");
+let RateLimitModule = class RateLimitModule {
+};
+exports.RateLimitModule = RateLimitModule;
+exports.RateLimitModule = RateLimitModule = __decorate([
+    (0, common_1.Module)({
+        imports: [config_module_1.ConfigModule],
+        providers: [rate_limiter_service_1.RateLimiterService],
+        exports: [rate_limiter_service_1.RateLimiterService],
+    })
+], RateLimitModule);
+//# sourceMappingURL=rate-limit.module.js.map

package/dist/src/ratelimit/rate-limit.module.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"rate-limit.module.js","sourceRoot":"","sources":["../../../src/ratelimit/rate-limit.module.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAAwC;AACxC,iEAA4D;AAC5D,2DAAuD;AAOhD,IAAM,eAAe,GAArB,MAAM,eAAe;CAAG,CAAA;AAAlB,0CAAe;0BAAf,eAAe;IAL3B,IAAA,eAAM,EAAC;QACN,OAAO,EAAE,CAAC,4BAAY,CAAC;QACvB,SAAS,EAAE,CAAC,yCAAkB,CAAC;QAC/B,OAAO,EAAE,CAAC,yCAAkB,CAAC;KAC9B,CAAC;GACW,eAAe,CAAG"}

package/dist/src/ratelimit/rate-limiter.service.d.ts

@@ -0,0 +1,68 @@
+import { ConfigService } from '../config/config.service';
+/**
+ * 限流服务
+ * 提供连接限流和消息限流功能
+ */
+export declare class RateLimiterService {
+    private readonly configService;
+    private readonly logger;
+    private readonly connectionRateLimit;
+    private readonly messageRateLimitPerConnection;
+    private readonly messageBurstLimit;
+    private readonly globalMessageRateLimit;
+    private readonly windowSize;
+    private readonly windowSlots;
+    private readonly slotDuration;
+    private connectionWindow;
+    private globalMessageWindow;
+    private connectionMessageWindows;
+    private connectionBurstCounts;
+    constructor(configService: ConfigService);
+    /**
+     * 创建新的滑动窗口
+     */
+    private createWindow;
+    /**
+     * 获取当前窗口分片索引
+     */
+    private getCurrentSlot;
+    /**
+     * 滑动窗口并返回总计数
+     */
+    private slideAndCount;
+    /**
+     * 增加窗口计数
+     */
+    private incrementWindow;
+    /**
+     * 检查连接速率限制
+     * @returns true 如果允许连接，false 如果被限流
+     */
+    checkConnectionRate(): boolean;
+    /**
+     * 检查消息速率限制
+     * @param connectionId 连接ID
+     * @returns { allowed: boolean, retryAfter?: number }
+     */
+    checkMessageRate(connectionId: string): {
+        allowed: boolean;
+        retryAfter?: number;
+    };
+    /**
+     * 移除连接的限流计数器
+     */
+    removeConnection(connectionId: string): void;
+    /**
+     * 清理过期的窗口数据
+     */
+    private cleanupExpiredWindows;
+    /**
+     * 获取当前统计信息
+     */
+    getStats(): {
+        connectionRate: number;
+        globalMessageRate: number;
+        trackedConnections: number;
+    };
+}
+//# sourceMappingURL=rate-limiter.service.d.ts.map

package/dist/src/ratelimit/rate-limiter.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"rate-limiter.service.d.ts","sourceRoot":"","sources":["../../../src/ratelimit/rate-limiter.service.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAUzD;;;GAGG;AACH,qBACa,kBAAkB;IA4BjB,OAAO,CAAC,QAAQ,CAAC,aAAa;IA3B1C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAuC;IAG9D,OAAO,CAAC,QAAQ,CAAC,mBAAmB,CAAS;IAE7C,OAAO,CAAC,QAAQ,CAAC,6BAA6B,CAAS;IAEvD,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAS;IAE3C,OAAO,CAAC,QAAQ,CAAC,sBAAsB,CAAS;IAGhD,OAAO,CAAC,QAAQ,CAAC,UAAU,CAAQ;IAEnC,OAAO,CAAC,QAAQ,CAAC,WAAW,CAAM;IAElC,OAAO,CAAC,QAAQ,CAAC,YAAY,CAAS;IAGtC,OAAO,CAAC,gBAAgB,CAAgB;IAExC,OAAO,CAAC,mBAAmB,CAAgB;IAE3C,OAAO,CAAC,wBAAwB,CAAoC;IAEpE,OAAO,CAAC,qBAAqB,CAA2D;gBAE3D,aAAa,EAAE,aAAa;IAuBzD;;OAEG;IACH,OAAO,CAAC,YAAY;IAOpB;;OAEG;IACH,OAAO,CAAC,cAAc;IAKtB;;OAEG;IACH,OAAO,CAAC,aAAa;IAsBrB;;OAEG;IACH,OAAO,CAAC,eAAe;IAMvB;;;OAGG;IACH,mBAAmB,IAAI,OAAO;IAY9B;;;;OAIG;IACH,gBAAgB,CAAC,YAAY,EAAE,MAAM,GAAG;QAAE,OAAO,EAAE,OAAO,CAAC;QAAC,UAAU,CAAC,EAAE,MAAM,CAAA;KAAE;IA6CjF;;OAEG;IACH,gBAAgB,CAAC,YAAY,EAAE,MAAM,GAAG,IAAI;IAK5C;;OAEG;IACH,OAAO,CAAC,qBAAqB;IAiB7B;;OAEG;IACH,QAAQ,IAAI;QACV,cAAc,EAAE,MAAM,CAAC;QACvB,iBAAiB,EAAE,MAAM,CAAC;QAC1B,kBAAkB,EAAE,MAAM,CAAC;KAC5B;CAOF"}

package/dist/src/ratelimit/rate-limiter.service.js

@@ -0,0 +1,201 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
+    return c > 3 && r && Object.defineProperty(target, key, r), r;
+};
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+var RateLimiterService_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RateLimiterService = void 0;
+const common_1 = require("@nestjs/common");
+const config_service_1 = require("../config/config.service");
+/**
+ * 限流服务
+ * 提供连接限流和消息限流功能
+ */
+let RateLimiterService = RateLimiterService_1 = class RateLimiterService {
+    configService;
+    logger = new common_1.Logger(RateLimiterService_1.name);
+    // 连接速率限制（每秒新连接数）
+    connectionRateLimit;
+    // 消息速率限制（每连接每秒）
+    messageRateLimitPerConnection;
+    // 消息突发上限
+    messageBurstLimit;
+    // 全局消息速率限制（每秒）
+    globalMessageRateLimit;
+    // 滑动窗口大小（毫秒）
+    windowSize = 1000;
+    // 窗口分片数量
+    windowSlots = 10;
+    // 每个分片时长
+    slotDuration;
+    // 连接速率计数
+    connectionWindow;
+    // 全局消息速率计数
+    globalMessageWindow;
+    // 每连接消息速率计数
+    connectionMessageWindows = new Map();
+    // 每连接突发计数（短时间内）
+    connectionBurstCounts = new Map();
+    constructor(configService) {
+        this.configService = configService;
+        // 从配置读取限流参数，使用合理的默认值
+        this.connectionRateLimit = configService.connectionRateLimit ?? 100;
+        this.messageRateLimitPerConnection = configService.messageRateLimitPerConnection ?? 50;
+        this.messageBurstLimit = configService.messageBurstLimit ?? 100;
+        this.globalMessageRateLimit = configService.globalMessageRateLimit ?? 50000;
+        this.slotDuration = this.windowSize / this.windowSlots;
+        // 初始化滑动窗口
+        this.connectionWindow = this.createWindow();
+        this.globalMessageWindow = this.createWindow();
+        // 定期清理过期的连接计数器
+        setInterval(() => this.cleanupExpiredWindows(), 60000);
+        this.logger.log(`RateLimiter initialized: connRate=${this.connectionRateLimit}/s, ` +
+            `msgRate=${this.messageRateLimitPerConnection}/s, ` +
+            `burst=${this.messageBurstLimit}, globalRate=${this.globalMessageRateLimit}/s`);
+    }
+    /**
+     * 创建新的滑动窗口
+     */
+    createWindow() {
+        return {
+            counts: new Array(this.windowSlots).fill(0),
+            windowStartTime: Date.now(),
+        };
+    }
+    /**
+     * 获取当前窗口分片索引
+     */
+    getCurrentSlot(window) {
+        const elapsed = Date.now() - window.windowStartTime;
+        return Math.floor(elapsed / this.slotDuration) % this.windowSlots;
+    }
+    /**
+     * 滑动窗口并返回总计数
+     */
+    slideAndCount(window) {
+        const now = Date.now();
+        const elapsed = now - window.windowStartTime;
+        // 如果超过一个完整窗口，重置
+        if (elapsed >= this.windowSize) {
+            const slotsToReset = Math.min(Math.floor(elapsed / this.slotDuration), this.windowSlots);
+            for (let i = 0; i < slotsToReset; i++) {
+                const slotIndex = (this.getCurrentSlot(window) - i + this.windowSlots) % this.windowSlots;
+                window.counts[slotIndex] = 0;
+            }
+            window.windowStartTime = now - (elapsed % this.windowSize);
+        }
+        return window.counts.reduce((sum, count) => sum + count, 0);
+    }
+    /**
+     * 增加窗口计数
+     */
+    incrementWindow(window) {
+        this.slideAndCount(window); // 先滑动窗口
+        const slot = this.getCurrentSlot(window);
+        window.counts[slot]++;
+    }
+    /**
+     * 检查连接速率限制
+     * @returns true 如果允许连接，false 如果被限流
+     */
+    checkConnectionRate() {
+        const currentRate = this.slideAndCount(this.connectionWindow);
+        if (currentRate >= this.connectionRateLimit) {
+            this.logger.warn(`Connection rate limit exceeded: ${currentRate}/${this.connectionRateLimit}`);
+            return false;
+        }
+        this.incrementWindow(this.connectionWindow);
+        return true;
+    }
+    /**
+     * 检查消息速率限制
+     * @param connectionId 连接ID
+     * @returns { allowed: boolean, retryAfter?: number }
+     */
+    checkMessageRate(connectionId) {
+        const now = Date.now();
+        // 1. 检查突发限制
+        const burst = this.connectionBurstCounts.get(connectionId);
+        if (burst) {
+            if (now < burst.resetTime) {
+                if (burst.count >= this.messageBurstLimit) {
+                    const retryAfter = burst.resetTime - now;
+                    return { allowed: false, retryAfter };
+                }
+                burst.count++;
+            }
+            else {
+                this.connectionBurstCounts.set(connectionId, { count: 1, resetTime: now + 1000 });
+            }
+        }
+        else {
+            this.connectionBurstCounts.set(connectionId, { count: 1, resetTime: now + 1000 });
+        }
+        // 2. 检查连接消息速率
+        let connWindow = this.connectionMessageWindows.get(connectionId);
+        if (!connWindow) {
+            connWindow = this.createWindow();
+            this.connectionMessageWindows.set(connectionId, connWindow);
+        }
+        const connRate = this.slideAndCount(connWindow);
+        if (connRate >= this.messageRateLimitPerConnection) {
+            return { allowed: false, retryAfter: 1000 };
+        }
+        // 3. 检查全局消息速率
+        const globalRate = this.slideAndCount(this.globalMessageWindow);
+        if (globalRate >= this.globalMessageRateLimit) {
+            this.logger.warn(`Global message rate limit exceeded: ${globalRate}/${this.globalMessageRateLimit}`);
+            return { allowed: false, retryAfter: 500 };
+        }
+        // 更新计数
+        this.incrementWindow(connWindow);
+        this.incrementWindow(this.globalMessageWindow);
+        return { allowed: true };
+    }
+    /**
+     * 移除连接的限流计数器
+     */
+    removeConnection(connectionId) {
+        this.connectionMessageWindows.delete(connectionId);
+        this.connectionBurstCounts.delete(connectionId);
+    }
+    /**
+     * 清理过期的窗口数据
+     */
+    cleanupExpiredWindows() {
+        const now = Date.now();
+        const expireThreshold = now - this.windowSize * 2;
+        for (const [connId, window] of this.connectionMessageWindows) {
+            if (window.windowStartTime < expireThreshold) {
+                this.connectionMessageWindows.delete(connId);
+            }
+        }
+        for (const [connId, burst] of this.connectionBurstCounts) {
+            if (burst.resetTime < now) {
+                this.connectionBurstCounts.delete(connId);
+            }
+        }
+    }
+    /**
+     * 获取当前统计信息
+     */
+    getStats() {
+        return {
+            connectionRate: this.slideAndCount(this.connectionWindow),
+            globalMessageRate: this.slideAndCount(this.globalMessageWindow),
+            trackedConnections: this.connectionMessageWindows.size,
+        };
+    }
+};
+exports.RateLimiterService = RateLimiterService;
+exports.RateLimiterService = RateLimiterService = RateLimiterService_1 = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [config_service_1.ConfigService])
+], RateLimiterService);
+//# sourceMappingURL=rate-limiter.service.js.map