@roomkit/worker 1.0.0

package/dist/src/migration/room-migration.service.js

@@ -0,0 +1,297 @@
+"use strict";
+/**
+ * 房间迁移服务
+ *
+ * 功能：
+ * 1. 主动迁移：Worker 主动将房间迁移到另一个 Worker
+ * 2. 被动接管：Worker 故障时，其他 Worker 接管孤儿房间
+ * 3. 迁移协调：确保迁移过程中的数据一致性
+ * 4. 路由更新：通知 Gateway 更新房间路由
+ *
+ * 设计：
+ * - 迁移过程：暂停房间 → 保存快照 → 通知 Gateway → 新 Worker 加载 → 确认完成
+ * - 容错机制：超时重试、回滚、失败通知
+ */
+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 RoomMigrationService_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RoomMigrationService = void 0;
+const common_1 = require("@nestjs/common");
+const redis_service_js_1 = require("../redis/redis.service.js");
+const config_service_js_1 = require("../config/config.service.js");
+const room_persistence_service_js_1 = require("../room/room-persistence.service.js");
+const core_1 = require("@roomkit/core");
+let RoomMigrationService = RoomMigrationService_1 = class RoomMigrationService {
+    redis;
+    config;
+    persistence;
+    logger = new common_1.Logger(RoomMigrationService_1.name);
+    // 正在迁移的房间
+    migratingRooms = new Set();
+    // 迁移超时（默认 30 秒）
+    MIGRATION_TIMEOUT = 30000;
+    constructor(redis, config, persistence) {
+        this.redis = redis;
+        this.config = config;
+        this.persistence = persistence;
+    }
+    /**
+     * 设置 RoomManager（避免循环依赖）
+     */
+    setRoomManager(roomManager) {
+        this.roomManager = roomManager;
+    }
+    roomManager = null;
+    /**
+     * 初始化迁移服务
+     */
+    async initialize() {
+        // 订阅迁移请求频道
+        await this.redis.subscribe(core_1.RedisChannels.migrationRequest(this.config.workerId), (message) => {
+            const request = JSON.parse(message);
+            this.handleMigrationRequest(request);
+        });
+        // 订阅接管请求频道（广播到所有 Worker）
+        await this.redis.subscribe(core_1.RedisChannels.takeoverRequest(), (message) => {
+            const request = JSON.parse(message);
+            this.handleTakeoverRequest(request);
+        });
+        this.logger.log('Migration service initialized');
+    }
+    /**
+     * 主动迁移房间到指定 Worker
+     */
+    async migrateRoom(roomId, targetWorkerId, reason = 'manual') {
+        const startTime = Date.now();
+        if (this.migratingRooms.has(roomId)) {
+            return {
+                success: false,
+                roomId,
+                sourceWorkerId: this.config.workerId,
+                error: 'Migration already in progress',
+                duration: 0,
+            };
+        }
+        this.migratingRooms.add(roomId);
+        this.logger.log(`Starting migration for room ${roomId} to ${targetWorkerId}`);
+        try {
+            // 1. 保存最终快照
+            await this.persistence.saveSnapshot(roomId);
+            await this.persistence.flushEvents(roomId);
+            // 2. 发送迁移请求到目标 Worker
+            const request = {
+                roomId,
+                sourceWorkerId: this.config.workerId,
+                targetWorkerId,
+                reason,
+                timestamp: Date.now(),
+            };
+            await this.redis.publish(core_1.RedisChannels.migrationRequest(targetWorkerId), request);
+            // 3. 等待确认（通过 Redis 状态检查）
+            const confirmed = await this.waitForMigrationConfirmation(roomId, targetWorkerId);
+            if (!confirmed) {
+                throw new Error('Migration confirmation timeout');
+            }
+            // 4. 通知 Gateway 更新路由
+            await this.notifyGatewayRouteChange(roomId, targetWorkerId);
+            const duration = Date.now() - startTime;
+            this.logger.log(`Migration completed for room ${roomId} in ${duration}ms`);
+            return {
+                success: true,
+                roomId,
+                sourceWorkerId: this.config.workerId,
+                targetWorkerId,
+                duration,
+            };
+        }
+        catch (error) {
+            const err = error;
+            this.logger.error(`Migration failed for room ${roomId}: ${err.message}`);
+            return {
+                success: false,
+                roomId,
+                sourceWorkerId: this.config.workerId,
+                targetWorkerId,
+                error: err.message,
+                duration: Date.now() - startTime,
+            };
+        }
+        finally {
+            this.migratingRooms.delete(roomId);
+        }
+    }
+    /**
+     * 处理迁移请求（作为目标 Worker）
+     */
+    async handleMigrationRequest(request) {
+        this.logger.log(`Received migration request for room ${request.roomId} from ${request.sourceWorkerId}`);
+        try {
+            // 1. 从持久化恢复房间
+            const snapshot = await this.persistence.recoverRoom(request.roomId);
+            if (!snapshot) {
+                throw new Error('Failed to recover room from persistence');
+            }
+            // 2. 重建房间（需要 RoomManager 支持，这里先标记）
+            if (!this.roomManager) {
+                throw new Error('RoomManager not set');
+            }
+            await this.roomManager.restoreRoom(snapshot);
+            // 3. 更新 Redis 中的房间所有权
+            await this.redis.client.hset(`gw:room:${request.roomId}:worker`, {
+                workerId: this.config.workerId,
+                migratedFrom: request.sourceWorkerId,
+                migratedAt: Date.now().toString(),
+            });
+            // 4. 发布确认消息
+            await this.redis.publish(core_1.RedisChannels.migrationConfirm(request.roomId), {
+                roomId: request.roomId,
+                targetWorkerId: this.config.workerId,
+                success: true,
+                timestamp: Date.now(),
+            });
+            this.logger.log(`Successfully migrated room ${request.roomId}`);
+        }
+        catch (error) {
+            const err = error;
+            this.logger.error(`Failed to handle migration request for room ${request.roomId}: ${err.message}`);
+            // 发布失败消息
+            await this.redis.publish(core_1.RedisChannels.migrationConfirm(request.roomId), {
+                roomId: request.roomId,
+                targetWorkerId: this.config.workerId,
+                success: false,
+                error: err.message,
+                timestamp: Date.now(),
+            });
+        }
+    }
+    /**
+     * 处理接管请求（孤儿房间）
+     */
+    async handleTakeoverRequest(request) {
+        this.logger.log(`Received takeover request for room ${request.roomId} (original worker: ${request.originalWorkerId})`);
+        // 简单策略：第一个响应的 Worker 接管
+        // 生产环境应该有更复杂的选择算法（负载、地理位置等）
+        try {
+            // 1. 尝试获取接管锁
+            const lockKey = `gw:room:${request.roomId}:takeover-lock`;
+            const acquired = await this.redis.client.set(lockKey, this.config.workerId, 'EX', 60, // 60秒过期
+            'NX');
+            if (!acquired) {
+                this.logger.debug(`Another worker is taking over room ${request.roomId}`);
+                return;
+            }
+            // 2. 从持久化恢复房间
+            const snapshot = await this.persistence.recoverRoom(request.roomId);
+            if (!snapshot) {
+                this.logger.warn(`No snapshot found for room ${request.roomId}, cannot takeover`);
+                await this.redis.client.del(lockKey);
+                return;
+            }
+            // 3. 重建房间
+            if (!this.roomManager) {
+                this.logger.warn('RoomManager not set, cannot takeover');
+                await this.redis.client.del(lockKey);
+                return;
+            }
+            await this.roomManager.restoreRoom(snapshot);
+            // 4. 更新所有权
+            await this.redis.client.hset(`gw:room:${request.roomId}:worker`, {
+                workerId: this.config.workerId,
+                takenOverFrom: request.originalWorkerId,
+                takenOverAt: Date.now().toString(),
+            });
+            // 5. 通知 Gateway
+            await this.notifyGatewayRouteChange(request.roomId, this.config.workerId);
+            this.logger.log(`Successfully took over room ${request.roomId} from ${request.originalWorkerId}`);
+        }
+        catch (error) {
+            const err = error;
+            this.logger.error(`Failed to takeover room ${request.roomId}: ${err.message}`);
+        }
+    }
+    /**
+     * 等待迁移确认
+     */
+    async waitForMigrationConfirmation(roomId, targetWorkerId) {
+        return new Promise((resolve) => {
+            const channel = core_1.RedisChannels.migrationConfirm(roomId);
+            let confirmed = false;
+            const handler = (message) => {
+                if (message.targetWorkerId === targetWorkerId) {
+                    confirmed = message.success;
+                    cleanup();
+                    resolve(confirmed);
+                }
+            };
+            const cleanup = () => {
+                // Redis 订阅没有 unsubscribe 方法，需要手动实现
+                // 这里简单处理，生产环境可以使用更复杂的订阅管理
+                clearTimeout(timeoutHandle);
+            };
+            const timeoutHandle = setTimeout(() => {
+                if (!confirmed) {
+                    cleanup();
+                    resolve(false);
+                }
+            }, this.MIGRATION_TIMEOUT);
+            this.redis.subscribe(channel, handler);
+        });
+    }
+    /**
+     * 通知 Gateway 更新路由
+     */
+    async notifyGatewayRouteChange(roomId, newWorkerId) {
+        await this.redis.publish(core_1.RedisChannels.routeUpdate(), {
+            type: 'room-migrated',
+            roomId,
+            oldWorkerId: this.config.workerId,
+            newWorkerId,
+            timestamp: Date.now(),
+        });
+    }
+    /**
+     * 请求其他 Worker 接管孤儿房间
+     */
+    async requestTakeover(roomId, originalWorkerId, reason = 'worker-failed') {
+        const request = {
+            roomId,
+            originalWorkerId,
+            reason,
+            timestamp: Date.now(),
+        };
+        await this.redis.publish(core_1.RedisChannels.takeoverRequest(), request);
+        this.logger.log(`Requested takeover for room ${roomId}`);
+    }
+    /**
+     * 获取迁移统计
+     */
+    getStats() {
+        return {
+            migratingRooms: this.migratingRooms.size,
+            activeRooms: Array.from(this.migratingRooms),
+        };
+    }
+    /**
+     * 模块销毁时清理
+     */
+    async onModuleDestroy() {
+        this.logger.log('Shutting down migration service...');
+        this.migratingRooms.clear();
+    }
+};
+exports.RoomMigrationService = RoomMigrationService;
+exports.RoomMigrationService = RoomMigrationService = RoomMigrationService_1 = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [redis_service_js_1.RedisService,
+        config_service_js_1.ConfigService,
+        room_persistence_service_js_1.RoomPersistenceService])
+], RoomMigrationService);
+//# sourceMappingURL=room-migration.service.js.map

package/dist/src/migration/room-migration.service.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"room-migration.service.js","sourceRoot":"","sources":["../../../src/migration/room-migration.service.ts"],"names":[],"mappings":";AAAA;;;;;;;;;;;;GAYG;;;;;;;;;;;;;AAEH,2CAAoD;AACpD,gEAAyD;AACzD,mEAA4D;AAC5D,qFAA6E;AAC7E,wCAA8C;AA8BvC,IAAM,oBAAoB,4BAA1B,MAAM,oBAAoB;IAUZ;IACA;IACA;IAXF,MAAM,GAAG,IAAI,eAAM,CAAC,sBAAoB,CAAC,IAAI,CAAC,CAAC;IAEhE,UAAU;IACF,cAAc,GAAG,IAAI,GAAG,EAAU,CAAC;IAE3C,gBAAgB;IACC,iBAAiB,GAAG,KAAK,CAAC;IAE3C,YACmB,KAAmB,EACnB,MAAqB,EACrB,WAAmC;QAFnC,UAAK,GAAL,KAAK,CAAc;QACnB,WAAM,GAAN,MAAM,CAAe;QACrB,gBAAW,GAAX,WAAW,CAAwB;IACnD,CAAC;IAEJ;;OAEG;IACH,cAAc,CAAC,WAA+B;QAC5C,IAAI,CAAC,WAAW,GAAG,WAAW,CAAC;IACjC,CAAC;IAEO,WAAW,GAA8B,IAAI,CAAC;IAEtD;;OAEG;IACH,KAAK,CAAC,UAAU;QACd,WAAW;QACX,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CACxB,oBAAa,CAAC,gBAAgB,CAAC,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,EACpD,CAAC,OAAe,EAAE,EAAE;YAClB,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAqB,CAAC;YACxD,IAAI,CAAC,sBAAsB,CAAC,OAAO,CAAC,CAAC;QACvC,CAAC,CACF,CAAC;QAEF,yBAAyB;QACzB,MAAM,IAAI,CAAC,KAAK,CAAC,SAAS,CACxB,oBAAa,CAAC,eAAe,EAAE,EAC/B,CAAC,OAAe,EAAE,EAAE;YAClB,MAAM,OAAO,GAAG,IAAI,CAAC,KAAK,CAAC,OAAO,CAAoB,CAAC;YACvD,IAAI,CAAC,qBAAqB,CAAC,OAAO,CAAC,CAAC;QACtC,CAAC,CACF,CAAC;QAEF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,+BAA+B,CAAC,CAAC;IACnD,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,WAAW,CACf,MAAc,EACd,cAAsB,EACtB,SAAiD,QAAQ;QAEzD,MAAM,SAAS,GAAG,IAAI,CAAC,GAAG,EAAE,CAAC;QAE7B,IAAI,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,EAAE,CAAC;YACpC,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM;gBACN,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBACpC,KAAK,EAAE,+BAA+B;gBACtC,QAAQ,EAAE,CAAC;aACZ,CAAC;QACJ,CAAC;QAED,IAAI,CAAC,cAAc,CAAC,GAAG,CAAC,MAAM,CAAC,CAAC;QAChC,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,+BAA+B,MAAM,OAAO,cAAc,EAAE,CAAC,CAAC;QAE9E,IAAI,CAAC;YACH,YAAY;YACZ,MAAM,IAAI,CAAC,WAAW,CAAC,YAAY,CAAC,MAAM,CAAC,CAAC;YAC5C,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,MAAM,CAAC,CAAC;YAE3C,sBAAsB;YACtB,MAAM,OAAO,GAAqB;gBAChC,MAAM;gBACN,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBACpC,cAAc;gBACd,MAAM;gBACN,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CAAC;YAEF,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CACtB,oBAAa,CAAC,gBAAgB,CAAC,cAAc,CAAC,EAC9C,OAAO,CACR,CAAC;YAEF,yBAAyB;YACzB,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,4BAA4B,CACvD,MAAM,EACN,cAAc,CACf,CAAC;YAEF,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;YACpD,CAAC;YAED,qBAAqB;YACrB,MAAM,IAAI,CAAC,wBAAwB,CAAC,MAAM,EAAE,cAAc,CAAC,CAAC;YAE5D,MAAM,QAAQ,GAAG,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACxC,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,gCAAgC,MAAM,OAAO,QAAQ,IAAI,CAC1D,CAAC;YAEF,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,MAAM;gBACN,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBACpC,cAAc;gBACd,QAAQ;aACT,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,GAAG,GAAG,KAAc,CAAC;YAC3B,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,6BAA6B,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,CACtD,CAAC;YAEF,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,MAAM;gBACN,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBACpC,cAAc;gBACd,KAAK,EAAE,GAAG,CAAC,OAAO;gBAClB,QAAQ,EAAE,IAAI,CAAC,GAAG,EAAE,GAAG,SAAS;aACjC,CAAC;QACJ,CAAC;gBAAS,CAAC;YACT,IAAI,CAAC,cAAc,CAAC,MAAM,CAAC,MAAM,CAAC,CAAC;QACrC,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,sBAAsB,CAAC,OAAyB;QAC5D,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,uCAAuC,OAAO,CAAC,MAAM,SAAS,OAAO,CAAC,cAAc,EAAE,CACvF,CAAC;QAEF,IAAI,CAAC;YACH,cAAc;YACd,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACpE,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,MAAM,IAAI,KAAK,CAAC,yCAAyC,CAAC,CAAC;YAC7D,CAAC;YAED,mCAAmC;YACnC,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;gBACtB,MAAM,IAAI,KAAK,CAAC,qBAAqB,CAAC,CAAC;YACzC,CAAC;YACD,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YAE7C,sBAAsB;YACtB,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAC1B,WAAW,OAAO,CAAC,MAAM,SAAS,EAClC;gBACE,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC9B,YAAY,EAAE,OAAO,CAAC,cAAc;gBACpC,UAAU,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;aAClC,CACF,CAAC;YAEF,YAAY;YACZ,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CACtB,oBAAa,CAAC,gBAAgB,CAAC,OAAO,CAAC,MAAM,CAAC,EAC9C;gBACE,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBACpC,OAAO,EAAE,IAAI;gBACb,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CACF,CAAC;YAEF,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,8BAA8B,OAAO,CAAC,MAAM,EAAE,CAAC,CAAC;QAClE,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,GAAG,GAAG,KAAc,CAAC;YAC3B,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,+CAA+C,OAAO,CAAC,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,CAChF,CAAC;YAEF,SAAS;YACT,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CACtB,oBAAa,CAAC,gBAAgB,CAAC,OAAO,CAAC,MAAM,CAAC,EAC9C;gBACE,MAAM,EAAE,OAAO,CAAC,MAAM;gBACtB,cAAc,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBACpC,OAAO,EAAE,KAAK;gBACd,KAAK,EAAE,GAAG,CAAC,OAAO;gBAClB,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;aACtB,CACF,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,qBAAqB,CAAC,OAAwB;QAC1D,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,sCAAsC,OAAO,CAAC,MAAM,sBAAsB,OAAO,CAAC,gBAAgB,GAAG,CACtG,CAAC;QAEF,wBAAwB;QACxB,4BAA4B;QAE5B,IAAI,CAAC;YACH,aAAa;YACb,MAAM,OAAO,GAAG,WAAW,OAAO,CAAC,MAAM,gBAAgB,CAAC;YAC1D,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAC1C,OAAO,EACP,IAAI,CAAC,MAAM,CAAC,QAAQ,EACpB,IAAI,EACJ,EAAE,EAAE,QAAQ;YACZ,IAAI,CACL,CAAC;YAEF,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,sCAAsC,OAAO,CAAC,MAAM,EAAE,CACvD,CAAC;gBACF,OAAO;YACT,CAAC;YAED,cAAc;YACd,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YACpE,IAAI,CAAC,QAAQ,EAAE,CAAC;gBACd,IAAI,CAAC,MAAM,CAAC,IAAI,CACd,8BAA8B,OAAO,CAAC,MAAM,mBAAmB,CAChE,CAAC;gBACF,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACrC,OAAO;YACT,CAAC;YAED,UAAU;YACV,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;gBACtB,IAAI,CAAC,MAAM,CAAC,IAAI,CAAC,sCAAsC,CAAC,CAAC;gBACzD,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC;gBACrC,OAAO;YACT,CAAC;YACD,MAAM,IAAI,CAAC,WAAW,CAAC,WAAW,CAAC,QAAQ,CAAC,CAAC;YAE7C,WAAW;YACX,MAAM,IAAI,CAAC,KAAK,CAAC,MAAM,CAAC,IAAI,CAC1B,WAAW,OAAO,CAAC,MAAM,SAAS,EAClC;gBACE,QAAQ,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;gBAC9B,aAAa,EAAE,OAAO,CAAC,gBAAgB;gBACvC,WAAW,EAAE,IAAI,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;aACnC,CACF,CAAC;YAEF,gBAAgB;YAChB,MAAM,IAAI,CAAC,wBAAwB,CAAC,OAAO,CAAC,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ,CAAC,CAAC;YAE1E,IAAI,CAAC,MAAM,CAAC,GAAG,CACb,+BAA+B,OAAO,CAAC,MAAM,SAAS,OAAO,CAAC,gBAAgB,EAAE,CACjF,CAAC;QACJ,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,GAAG,GAAG,KAAc,CAAC;YAC3B,IAAI,CAAC,MAAM,CAAC,KAAK,CACf,2BAA2B,OAAO,CAAC,MAAM,KAAK,GAAG,CAAC,OAAO,EAAE,CAC5D,CAAC;QACJ,CAAC;IACH,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,4BAA4B,CACxC,MAAc,EACd,cAAsB;QAEtB,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,EAAE;YAC7B,MAAM,OAAO,GAAG,oBAAa,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC;YACvD,IAAI,SAAS,GAAG,KAAK,CAAC;YAEtB,MAAM,OAAO,GAAG,CAAC,OAAY,EAAE,EAAE;gBAC/B,IAAI,OAAO,CAAC,cAAc,KAAK,cAAc,EAAE,CAAC;oBAC9C,SAAS,GAAG,OAAO,CAAC,OAAO,CAAC;oBAC5B,OAAO,EAAE,CAAC;oBACV,OAAO,CAAC,SAAS,CAAC,CAAC;gBACrB,CAAC;YACH,CAAC,CAAC;YAEF,MAAM,OAAO,GAAG,GAAG,EAAE;gBACnB,mCAAmC;gBACnC,0BAA0B;gBAC1B,YAAY,CAAC,aAAa,CAAC,CAAC;YAC9B,CAAC,CAAC;YAEF,MAAM,aAAa,GAAG,UAAU,CAAC,GAAG,EAAE;gBACpC,IAAI,CAAC,SAAS,EAAE,CAAC;oBACf,OAAO,EAAE,CAAC;oBACV,OAAO,CAAC,KAAK,CAAC,CAAC;gBACjB,CAAC;YACH,CAAC,EAAE,IAAI,CAAC,iBAAiB,CAAC,CAAC;YAE3B,IAAI,CAAC,KAAK,CAAC,SAAS,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;QACzC,CAAC,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACK,KAAK,CAAC,wBAAwB,CACpC,MAAc,EACd,WAAmB;QAEnB,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,oBAAa,CAAC,WAAW,EAAE,EAAE;YACpD,IAAI,EAAE,eAAe;YACrB,MAAM;YACN,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,QAAQ;YACjC,WAAW;YACX,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC,CAAC;IACL,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe,CACnB,MAAc,EACd,gBAAwB,EACxB,SAAsC,eAAe;QAErD,MAAM,OAAO,GAAoB;YAC/B,MAAM;YACN,gBAAgB;YAChB,MAAM;YACN,SAAS,EAAE,IAAI,CAAC,GAAG,EAAE;SACtB,CAAC;QAEF,MAAM,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,oBAAa,CAAC,eAAe,EAAE,EAAE,OAAO,CAAC,CAAC;QACnE,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,+BAA+B,MAAM,EAAE,CAAC,CAAC;IAC3D,CAAC;IAED;;OAEG;IACH,QAAQ;QAIN,OAAO;YACL,cAAc,EAAE,IAAI,CAAC,cAAc,CAAC,IAAI;YACxC,WAAW,EAAE,KAAK,CAAC,IAAI,CAAC,IAAI,CAAC,cAAc,CAAC;SAC7C,CAAC;IACJ,CAAC;IAED;;OAEG;IACH,KAAK,CAAC,eAAe;QACnB,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,oCAAoC,CAAC,CAAC;QACtD,IAAI,CAAC,cAAc,CAAC,KAAK,EAAE,CAAC;IAC9B,CAAC;CACF,CAAA;AAxWY,oDAAoB;+BAApB,oBAAoB;IADhC,IAAA,mBAAU,GAAE;qCAWe,+BAAY;QACX,iCAAa;QACR,oDAAsB;GAZ3C,oBAAoB,CAwWhC"}

package/dist/src/redis/redis.module.d.ts

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

package/dist/src/redis/redis.module.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"redis.module.d.ts","sourceRoot":"","sources":["../../../src/redis/redis.module.ts"],"names":[],"mappings":"AAGA,qBAKa,WAAW;CAAG"}

package/dist/src/redis/redis.module.js

@@ -0,0 +1,22 @@
+"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.RedisModule = void 0;
+const common_1 = require("@nestjs/common");
+const redis_service_1 = require("./redis.service");
+let RedisModule = class RedisModule {
+};
+exports.RedisModule = RedisModule;
+exports.RedisModule = RedisModule = __decorate([
+    (0, common_1.Global)(),
+    (0, common_1.Module)({
+        providers: [redis_service_1.RedisService],
+        exports: [redis_service_1.RedisService],
+    })
+], RedisModule);
+//# sourceMappingURL=redis.module.js.map

package/dist/src/redis/redis.module.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"redis.module.js","sourceRoot":"","sources":["../../../src/redis/redis.module.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAAgD;AAChD,mDAA+C;AAOxC,IAAM,WAAW,GAAjB,MAAM,WAAW;CAAG,CAAA;AAAd,kCAAW;sBAAX,WAAW;IALvB,IAAA,eAAM,GAAE;IACR,IAAA,eAAM,EAAC;QACN,SAAS,EAAE,CAAC,4BAAY,CAAC;QACzB,OAAO,EAAE,CAAC,4BAAY,CAAC;KACxB,CAAC;GACW,WAAW,CAAG"}

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

@@ -0,0 +1,68 @@
+import { OnModuleInit, OnModuleDestroy } from '@nestjs/common';
+import Redis from 'ioredis';
+import { ConfigService } from '../config/config.service';
+export declare class RedisService implements OnModuleInit, OnModuleDestroy {
+    private readonly configService;
+    private readonly logger;
+    readonly client: Redis;
+    readonly subscriber: Redis;
+    private messageHandlers;
+    private registeredGameTypes;
+    private heartbeatTimer?;
+    private cleanupTimer?;
+    private readonly HEARTBEAT_INTERVAL;
+    private readonly HEARTBEAT_TIMEOUT;
+    private readonly CLEANUP_INTERVAL;
+    constructor(configService: ConfigService);
+    onModuleInit(): Promise<void>;
+    onModuleDestroy(): Promise<void>;
+    /**
+     * 注册Worker（由 RoomRegistryService 调用）
+     */
+    registerWorker(gameTypes: string[]): Promise<void>;
+    /**
+     * 注销Worker
+     */
+    private unregisterWorker;
+    /**
+     * 订阅通道
+     */
+    private subscribeToChannels;
+    /**
+     * 订阅频道并注册处理器
+     */
+    subscribe(channel: string, handler: (message: string) => void): Promise<void>;
+    /**
+     * 注册消息处理器
+     */
+    onMessage(channel: string, handler: (message: string) => void): void;
+    /**
+     * 发布消息
+     */
+    publish(channel: string, message: string | object): Promise<void>;
+    /**
+     * 更新Worker统计信息
+     */
+    updateStats(roomCount: number, playerCount: number): Promise<void>;
+    /**
+     * 启动心跳定时器
+     */
+    private startHeartbeat;
+    /**
+     * 发送心跳
+     */
+    private sendHeartbeat;
+    /**
+     * 启动清理定时器
+     */
+    private startCleanup;
+    /**
+     * 清理过期的Worker节点
+     */
+    private cleanupDeadWorkers;
+    /**
+     * 移除Worker（从所有列表中）
+     */
+    private removeWorker;
+}
+//# sourceMappingURL=redis.service.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"redis.service.d.ts","sourceRoot":"","sources":["../../../src/redis/redis.service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAc,YAAY,EAAE,eAAe,EAAU,MAAM,gBAAgB,CAAC;AACnF,OAAO,KAAK,MAAM,SAAS,CAAC;AAC5B,OAAO,EAAE,aAAa,EAAE,MAAM,0BAA0B,CAAC;AAGzD,qBACa,YAAa,YAAW,YAAY,EAAE,eAAe;IAkBpD,OAAO,CAAC,QAAQ,CAAC,aAAa;IAjB1C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAiC;IAExD,QAAQ,CAAC,MAAM,EAAE,KAAK,CAAC;IACvB,QAAQ,CAAC,UAAU,EAAE,KAAK,CAAC;IAE3B,OAAO,CAAC,eAAe,CAAqD;IAC5E,OAAO,CAAC,mBAAmB,CAAgB;IAG3C,OAAO,CAAC,cAAc,CAAC,CAAiB;IACxC,OAAO,CAAC,YAAY,CAAC,CAAiB;IAGtC,OAAO,CAAC,QAAQ,CAAC,kBAAkB,CAAQ;IAC3C,OAAO,CAAC,QAAQ,CAAC,iBAAiB,CAAS;IAC3C,OAAO,CAAC,QAAQ,CAAC,gBAAgB,CAAS;gBAEb,aAAa,EAAE,aAAa;IAanD,YAAY;IA4BZ,eAAe;IAgBrB;;OAEG;IACG,cAAc,CAAC,SAAS,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC,IAAI,CAAC;IA6BxD;;OAEG;YACW,gBAAgB;IAkB9B;;OAEG;YACW,mBAAmB;IAMjC;;OAEG;IACG,SAAS,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI,GAAG,OAAO,CAAC,IAAI,CAAC;IAMnF;;OAEG;IACH,SAAS,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,CAAC,OAAO,EAAE,MAAM,KAAK,IAAI;IAI7D;;OAEG;IACG,OAAO,CAAC,OAAO,EAAE,MAAM,EAAE,OAAO,EAAE,MAAM,GAAG,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAKvE;;OAEG;IACG,WAAW,CAAC,SAAS,EAAE,MAAM,EAAE,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC,IAAI,CAAC;IAYxE;;OAEG;IACH,OAAO,CAAC,cAAc;IAYtB;;OAEG;YACW,aAAa;IAK3B;;OAEG;IACH,OAAO,CAAC,YAAY;IAYpB;;OAEG;YACW,kBAAkB;IAgChC;;OAEG;YACW,YAAY;CAsB3B"}

package/dist/src/redis/redis.service.js

@@ -0,0 +1,260 @@
+"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 __importDefault = (this && this.__importDefault) || function (mod) {
+    return (mod && mod.__esModule) ? mod : { "default": mod };
+};
+var RedisService_1;
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.RedisService = void 0;
+const common_1 = require("@nestjs/common");
+const ioredis_1 = __importDefault(require("ioredis"));
+const config_service_1 = require("../config/config.service");
+const core_1 = require("@roomkit/core");
+let RedisService = RedisService_1 = class RedisService {
+    configService;
+    logger = new common_1.Logger(RedisService_1.name);
+    client;
+    subscriber;
+    messageHandlers = new Map();
+    registeredGameTypes = [];
+    // 心跳和清理定时器
+    heartbeatTimer;
+    cleanupTimer;
+    // 常量配置
+    HEARTBEAT_INTERVAL = 5000; // 5秒心跳
+    HEARTBEAT_TIMEOUT = 30000; // 30秒超时
+    CLEANUP_INTERVAL = 60000; // 60秒清理一次
+    constructor(configService) {
+        this.configService = configService;
+        const redisConfig = {
+            host: this.configService.redisHost,
+            port: this.configService.redisPort,
+            password: this.configService.redisPassword,
+            retryStrategy: (times) => Math.min(times * 50, 2000),
+            maxRetriesPerRequest: 3,
+        };
+        this.client = new ioredis_1.default(redisConfig);
+        this.subscriber = new ioredis_1.default(redisConfig);
+    }
+    async onModuleInit() {
+        this.client.on('connect', () => this.logger.log('Redis client connected'));
+        this.client.on('error', (err) => this.logger.error('Redis client error', err));
+        this.subscriber.on('connect', () => this.logger.log('Redis subscriber connected'));
+        this.subscriber.on('error', (err) => this.logger.error('Redis subscriber error', err));
+        // 设置消息处理
+        this.subscriber.on('message', (channel, message) => {
+            this.logger.debug(`[Redis] Received message on channel: ${channel}, length: ${message.length}`);
+            const handler = this.messageHandlers.get(channel);
+            if (handler) {
+                handler(message);
+            }
+            else {
+                this.logger.warn(`[Redis] No handler registered for channel: ${channel}`);
+            }
+        });
+        // 订阅通道
+        await this.subscribeToChannels();
+        // 启动心跳定时器
+        this.startHeartbeat();
+        // 启动清理定时器
+        this.startCleanup();
+    }
+    async onModuleDestroy() {
+        // 停止定时器
+        if (this.heartbeatTimer) {
+            clearInterval(this.heartbeatTimer);
+        }
+        if (this.cleanupTimer) {
+            clearInterval(this.cleanupTimer);
+        }
+        // 注销Worker
+        await this.unregisterWorker();
+        await this.client.quit();
+        await this.subscriber.quit();
+    }
+    /**
+     * 注册Worker（由 RoomRegistryService 调用）
+     */
+    async registerWorker(gameTypes) {
+        const workerId = this.configService.workerId;
+        this.registeredGameTypes = gameTypes;
+        // 注册Worker信息
+        await this.client.hset(core_1.RedisKeys.worker(workerId), {
+            workerId,
+            gameTypes: JSON.stringify(gameTypes),
+            status: 'active', // 状态: active | draining | offline
+            roomCount: 0,
+            playerCount: 0,
+            lastHeartbeat: Date.now(),
+            startedAt: Date.now(),
+        });
+        // 设置TTL为30秒（需要心跳续期）
+        await this.client.expire(core_1.RedisKeys.worker(workerId), 30);
+        // 添加到Worker列表
+        await this.client.sadd(core_1.RedisKeys.workersAll(), workerId);
+        // 添加到各游戏类型的Worker列表
+        for (const gameType of gameTypes) {
+            await this.client.sadd(core_1.RedisKeys.workersByType(gameType), workerId);
+        }
+        this.logger.log(`Worker registered: ${workerId}, gameTypes: ${gameTypes.join(', ')}, status: active`);
+    }
+    /**
+     * 注销Worker
+     */
+    async unregisterWorker() {
+        const workerId = this.configService.workerId;
+        const gameTypes = this.registeredGameTypes;
+        // 更新状态为离线
+        await this.client.hset(core_1.RedisKeys.worker(workerId), 'status', 'offline');
+        // 从Worker列表移除
+        await this.client.srem(core_1.RedisKeys.workersAll(), workerId);
+        // 从各游戏类型的Worker列表移除
+        for (const gameType of gameTypes) {
+            await this.client.srem(core_1.RedisKeys.workersByType(gameType), workerId);
+        }
+        this.logger.log(`Worker unregistered: ${workerId}`);
+    }
+    /**
+     * 订阅通道
+     */
+    async subscribeToChannels() {
+        // Worker 频道由 MessageHandlerService 订阅，这里不需要提前订阅
+        // 避免订阅时 handler 还未注册的问题
+        this.logger.log('Redis subscription ready');
+    }
+    /**
+     * 订阅频道并注册处理器
+     */
+    async subscribe(channel, handler) {
+        this.messageHandlers.set(channel, handler);
+        await this.subscriber.subscribe(channel);
+        this.logger.log(`Subscribed to channel: ${channel}`);
+    }
+    /**
+     * 注册消息处理器
+     */
+    onMessage(channel, handler) {
+        this.messageHandlers.set(channel, handler);
+    }
+    /**
+     * 发布消息
+     */
+    async publish(channel, message) {
+        const data = typeof message === 'string' ? message : JSON.stringify(message);
+        await this.client.publish(channel, data);
+    }
+    /**
+     * 更新Worker统计信息
+     */
+    async updateStats(roomCount, playerCount) {
+        const workerId = this.configService.workerId;
+        await this.client.hset(core_1.RedisKeys.worker(workerId), 'roomCount', roomCount.toString());
+        await this.client.hset(core_1.RedisKeys.worker(workerId), 'playerCount', playerCount.toString());
+        await this.client.hset(core_1.RedisKeys.worker(workerId), 'lastHeartbeat', Date.now().toString());
+        // 续期TTL（根据状态决定时长）
+        const status = await this.client.hget(core_1.RedisKeys.worker(workerId), 'status');
+        const ttl = status === 'draining' ? 120 : 30;
+        await this.client.expire(core_1.RedisKeys.worker(workerId), ttl);
+    }
+    /**
+     * 启动心跳定时器
+     */
+    startHeartbeat() {
+        this.heartbeatTimer = setInterval(async () => {
+            try {
+                await this.sendHeartbeat();
+            }
+            catch (error) {
+                this.logger.error('Failed to send heartbeat', error);
+            }
+        }, this.HEARTBEAT_INTERVAL);
+        this.logger.log(`Heartbeat started: interval=${this.HEARTBEAT_INTERVAL}ms`);
+    }
+    /**
+     * 发送心跳
+     */
+    async sendHeartbeat() {
+        const workerId = this.configService.workerId;
+        await this.client.hset(core_1.RedisKeys.worker(workerId), 'lastHeartbeat', Date.now().toString());
+    }
+    /**
+     * 启动清理定时器
+     */
+    startCleanup() {
+        this.cleanupTimer = setInterval(async () => {
+            try {
+                await this.cleanupDeadWorkers();
+            }
+            catch (error) {
+                this.logger.error('Failed to cleanup dead workers', error);
+            }
+        }, this.CLEANUP_INTERVAL);
+        this.logger.log(`Cleanup started: interval=${this.CLEANUP_INTERVAL}ms`);
+    }
+    /**
+     * 清理过期的Worker节点
+     */
+    async cleanupDeadWorkers() {
+        const now = Date.now();
+        const timeout = this.HEARTBEAT_TIMEOUT;
+        // 获取所有Worker
+        const workerIds = await this.client.smembers(core_1.RedisKeys.workersAll());
+        let cleaned = 0;
+        for (const workerId of workerIds) {
+            const workerData = await this.client.hgetall(core_1.RedisKeys.worker(workerId));
+            if (!workerData || !workerData.lastHeartbeat) {
+                // 没有心跳信息，清理掉
+                await this.removeWorker(workerId);
+                cleaned++;
+                continue;
+            }
+            const lastHeartbeat = parseInt(workerData.lastHeartbeat);
+            if (now - lastHeartbeat > timeout) {
+                // 心跳超时，清理掉
+                this.logger.warn(`Removing dead worker: ${workerId} (last heartbeat: ${new Date(lastHeartbeat).toISOString()})`);
+                await this.removeWorker(workerId);
+                cleaned++;
+            }
+        }
+        if (cleaned > 0) {
+            this.logger.log(`Cleaned up ${cleaned} dead worker(s)`);
+        }
+    }
+    /**
+     * 移除Worker（从所有列表中）
+     */
+    async removeWorker(workerId) {
+        // 获取Worker信息
+        const workerData = await this.client.hgetall(core_1.RedisKeys.worker(workerId));
+        // 从总列表移除
+        await this.client.srem(core_1.RedisKeys.workersAll(), workerId);
+        // 从游戏类型列表移除
+        if (workerData.gameTypes) {
+            try {
+                const gameTypes = JSON.parse(workerData.gameTypes);
+                for (const gameType of gameTypes) {
+                    await this.client.srem(core_1.RedisKeys.workersByType(gameType), workerId);
+                }
+            }
+            catch (e) {
+                // 忽略JSON解析错误
+            }
+        }
+        // 删除Worker数据
+        await this.client.del(core_1.RedisKeys.worker(workerId));
+    }
+};
+exports.RedisService = RedisService;
+exports.RedisService = RedisService = RedisService_1 = __decorate([
+    (0, common_1.Injectable)(),
+    __metadata("design:paramtypes", [config_service_1.ConfigService])
+], RedisService);
+//# sourceMappingURL=redis.service.js.map