rl-rockcli 0.0.1 → 0.0.2

package/commands/sandbox/replay/worker.js

@@ -0,0 +1,166 @@
+const path = require('path');
+const { SandboxClient, SandboxConfig } = require('../../../utils/sandbox-client');
+const { ReplayExecutor } = require('./executor');
+const { ReplayRequest, ReplayPlan } = require('./request');
+
+/**
+ * 单个沙箱回放工作者
+ * 封装独立的 SandboxClient 和 ReplayExecutor
+ */
+class ReplayWorker {
+  /**
+   * @param {number} workerId - 工作者 ID（1-based）
+   * @param {SandboxConfig} config - 沙箱配置
+   * @param {ReplayPlan} plan - 回放计划（会被深拷贝）
+   * @param {Object} options - 回放选项
+   * @param {string} options.timing - 时间控制
+   * @param {number} options.interval - 固定间隔
+   * @param {string} options.logFile - 日志文件基础路径
+   */
+  constructor(workerId, config, plan, options = {}) {
+    this.workerId = workerId;
+    this.config = config;
+    this.plan = this.deepClonePlan(plan);
+    this.options = options;
+
+    // 独立的沙箱客户端和执行器
+    this.client = null;
+    this.executor = null;
+
+    // 执行状态
+    this.status = 'pending';  // pending | running | completed | failed
+    this.error = null;
+    this.results = null;
+    this.startTime = null;
+    this.endTime = null;
+  }
+
+  /**
+   * 深拷贝回放计划
+   * 确保每个 worker 有独立的请求对象（requestBody 可能被修改）
+   * @param {ReplayPlan} plan
+   * @returns {ReplayPlan}
+   */
+  deepClonePlan(plan) {
+    const cloned = new ReplayPlan();
+    cloned.originalCount = plan.originalCount;
+    cloned.stats = { ...plan.stats };
+    cloned.pidAssociations = new Map(plan.pidAssociations);
+
+    // 深拷贝请求列表
+    cloned.requests = plan.requests.map(req => this.cloneRequest(req));
+    cloned.skippedRequests = plan.skippedRequests.map(req => this.cloneRequest(req));
+    cloned.mergedRequests = plan.mergedRequests.map(req => this.cloneRequest(req));
+
+    return cloned;
+  }
+
+  /**
+   * 深拷贝单个请求
+   * @param {ReplayRequest} req
+   * @returns {ReplayRequest}
+   */
+  cloneRequest(req) {
+    const cloned = new ReplayRequest({
+      method: req.method,
+      uri: req.uri,
+      requestBody: JSON.parse(JSON.stringify(req.requestBody || {})),
+      headers: JSON.parse(JSON.stringify(req.headers || {})),
+      responseTime: req.responseTime
+    }, req.index);
+
+    cloned.category = req.category;
+    cloned.action = req.action;
+    cloned.originalPid = req.originalPid;
+    cloned.skipReason = req.skipReason;
+
+    return cloned;
+  }
+
+  /**
+   * 获取 worker 专属的日志文件名
+   * @returns {string}
+   */
+  getLogFileName() {
+    const baseLogFile = this.options.logFile || 'replay.log';
+    const ext = path.extname(baseLogFile);
+    const base = path.basename(baseLogFile, ext);
+    const dir = path.dirname(baseLogFile);
+    return path.join(dir, `${base}-worker${this.workerId}${ext}`);
+  }
+
+  /**
+   * 执行回放
+   * @returns {Promise<WorkerResult>}
+   */
+  async execute() {
+    this.status = 'running';
+    this.startTime = Date.now();
+
+    try {
+      // 创建独立的 SandboxClient
+      this.client = new SandboxClient(this.config);
+
+      // 创建独立的 ReplayExecutor
+      this.executor = new ReplayExecutor(this.client, {
+        timing: this.options.timing,
+        interval: this.options.interval,
+        quiet: true,  // 并发模式下强制静默，由 orchestrator 统一输出
+        logFile: this.getLogFileName(),
+        plan: this.plan,
+        workerId: this.workerId
+      });
+
+      // 执行回放
+      this.results = await this.executor.execute(this.plan);
+      this.status = 'completed';
+
+    } catch (error) {
+      this.status = 'failed';
+      this.error = error;
+      this.results = {
+        total: this.plan.requests.length,
+        success: 0,
+        failed: this.plan.requests.length,
+        skipped: 0
+      };
+    } finally {
+      this.endTime = Date.now();
+    }
+
+    return this.getWorkerResult();
+  }
+
+  /**
+   * 获取工作者执行结果
+   * @returns {WorkerResult}
+   */
+  getWorkerResult() {
+    return {
+      workerId: this.workerId,
+      status: this.status,
+      sandboxId: this.executor?.currentSandboxId || null,
+      results: this.results,
+      error: this.error?.message || null,
+      duration: this.endTime - this.startTime,
+      logFile: this.getLogFileName()
+    };
+  }
+
+  /**
+   * 获取当前进度
+   * @returns {Object}
+   */
+  getProgress() {
+    return {
+      workerId: this.workerId,
+      status: this.status,
+      sandboxId: this.executor?.currentSandboxId?.substring(0, 8) || null,
+      success: this.executor?.results?.success || 0,
+      failed: this.executor?.results?.failed || 0,
+      total: this.plan.requests.length
+    };
+  }
+}
+
+module.exports = { ReplayWorker };

package/commands/sandbox/replay.js

@@ -0,0 +1,2 @@
+// Core replay - re-export from new location
+module.exports = require('./core/replay');

package/commands/sandbox/session.js

@@ -0,0 +1,2 @@
+// Compatibility wrapper - delegates to core implementation
+module.exports = require('./core/session');