@gravito/launchpad 1.2.2 → 1.3.2

package/dist/index.js

@@ -20,13 +20,25 @@ var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: tru
 // src/index.ts
 var index_exports = {};
 __export(index_exports, {
+  DEFAULT_POOL_CONFIG: () => DEFAULT_POOL_CONFIG,
+  DEFAULT_REFURBISH_CONFIG: () => DEFAULT_REFURBISH_CONFIG,
   LaunchpadOrbit: () => LaunchpadOrbit,
   Mission: () => Mission,
+  MissionAssigned: () => MissionAssigned,
   MissionControl: () => MissionControl,
+  MissionQueue: () => MissionQueue,
+  MissionQueueTimeout: () => MissionQueueTimeout,
+  MissionQueued: () => MissionQueued,
   PayloadInjector: () => PayloadInjector,
+  PoolExhausted: () => PoolExhausted,
+  PoolExhaustedException: () => PoolExhaustedException,
   PoolManager: () => PoolManager,
+  QueueTimeoutException: () => QueueTimeoutException,
   RefurbishUnit: () => RefurbishUnit,
+  RefurbishmentCompleted: () => RefurbishmentCompleted,
   Rocket: () => Rocket,
+  RocketIgnited: () => RocketIgnited,
+  RocketSplashedDown: () => RocketSplashedDown,
   RocketStatus: () => RocketStatus,
   bootstrapLaunchpad: () => bootstrapLaunchpad
 });
@@ -143,6 +155,36 @@ registry = "${fallbackRegistry}"
   }
 };
 
+// src/Domain/Interfaces.ts
+var DEFAULT_POOL_CONFIG = {
+  maxRockets: 20,
+  warmupCount: 3,
+  maxQueueSize: 50,
+  queueTimeoutMs: 3e4,
+  exhaustionStrategy: "queue",
+  dynamicLimit: 5
+};
+var DEFAULT_REFURBISH_CONFIG = {
+  strategy: "deep-clean",
+  cleanupCommands: [
+    // 1. 殺掉所有使用者進程
+    "pkill -9 -u bun || true",
+    'pkill -9 -u root -f "bun|node" || true',
+    // 2. 清理應用目錄
+    "rm -rf /app/* /app/.* 2>/dev/null || true",
+    // 3. 清理暫存檔案
+    "rm -rf /tmp/* /var/tmp/* 2>/dev/null || true",
+    // 4. 清理 bun 暫存
+    "rm -rf /root/.bun/install/cache/tmp/* 2>/dev/null || true",
+    "rm -rf /home/bun/.bun/install/cache/tmp/* 2>/dev/null || true",
+    // 5. 清理日誌
+    "rm -rf /var/log/*.log 2>/dev/null || true"
+  ],
+  cleanupTimeoutMs: 3e4,
+  failureAction: "decommission",
+  maxRetries: 2
+};
+
 // src/Domain/Rocket.ts
 var import_enterprise2 = require("@gravito/enterprise");
 
@@ -173,6 +215,28 @@ var RefurbishmentCompleted = class extends import_enterprise.DomainEvent {
     this.rocketId = rocketId;
   }
 };
+var PoolExhausted = class extends import_enterprise.DomainEvent {
+  constructor(totalRockets, maxRockets, queueLength) {
+    super();
+    this.totalRockets = totalRockets;
+    this.maxRockets = maxRockets;
+    this.queueLength = queueLength;
+  }
+};
+var MissionQueued = class extends import_enterprise.DomainEvent {
+  constructor(missionId, queuePosition) {
+    super();
+    this.missionId = missionId;
+    this.queuePosition = queuePosition;
+  }
+};
+var MissionQueueTimeout = class extends import_enterprise.DomainEvent {
+  constructor(missionId, waitTimeMs) {
+    super();
+    this.missionId = missionId;
+    this.waitTimeMs = waitTimeMs;
+  }
+};
 
 // src/Domain/RocketStatus.ts
 var RocketStatus = /* @__PURE__ */ ((RocketStatus2) => {
@@ -260,6 +324,23 @@ var Rocket = class _Rocket extends import_enterprise2.AggregateRoot {
   decommission() {
     this._status = "DECOMMISSIONED" /* DECOMMISSIONED */;
   }
+  /**
+   * 替換底層容器（僅在 REFURBISHING 狀態時允許）
+   * 用於 destroy-recreate 策略
+   *
+   * @param newContainerId - 新容器 ID
+   * @throws {Error} 當火箭不在 REFURBISHING 狀態時
+   *
+   * @public
+   * @since 1.3.0
+   */
+  replaceContainer(newContainerId) {
+    if (this._status !== "REFURBISHING" /* REFURBISHING */) {
+      throw new Error(`\u7121\u6CD5\u66FF\u63DB\u5BB9\u5668\uFF1A\u706B\u7BAD ${this.id} \u4E0D\u5728 REFURBISHING \u72C0\u614B`);
+    }
+    this._containerId = newContainerId;
+    this._assignedDomain = null;
+  }
   toJSON() {
     return {
       id: this.id,
@@ -278,44 +359,218 @@ var Rocket = class _Rocket extends import_enterprise2.AggregateRoot {
   }
 };
 
+// src/Application/MissionQueue.ts
+var MissionQueue = class {
+  queue = [];
+  maxSize;
+  timeoutMs;
+  constructor(maxSize = 50, timeoutMs = 3e4) {
+    this.maxSize = maxSize;
+    this.timeoutMs = timeoutMs;
+  }
+  get length() {
+    return this.queue.length;
+  }
+  get isFull() {
+    return this.queue.length >= this.maxSize;
+  }
+  /**
+   * 將任務加入隊列，返回 Promise 等待分配
+   */
+  enqueue(mission) {
+    if (this.isFull) {
+      return Promise.reject(
+        new PoolExhaustedException(`\u4EFB\u52D9\u968A\u5217\u5DF2\u6EFF\uFF08${this.maxSize}\uFF09\uFF0C\u7121\u6CD5\u63A5\u53D7\u65B0\u4EFB\u52D9`)
+      );
+    }
+    return new Promise((resolve, reject) => {
+      const timeoutId = setTimeout(() => {
+        this.removeFromQueue(mission.id);
+        reject(new QueueTimeoutException(`\u4EFB\u52D9 ${mission.id} \u7B49\u5F85\u8D85\u6642`));
+      }, this.timeoutMs);
+      this.queue.push({
+        mission,
+        resolve,
+        reject,
+        enqueuedAt: Date.now(),
+        timeoutId
+      });
+      console.log(
+        `[MissionQueue] \u4EFB\u52D9 ${mission.id} \u5DF2\u52A0\u5165\u968A\u5217\uFF0C\u7576\u524D\u968A\u5217\u9577\u5EA6: ${this.queue.length}`
+      );
+    });
+  }
+  /**
+   * 取出下一個等待的任務
+   */
+  dequeue() {
+    const item = this.queue.shift();
+    if (item) {
+      clearTimeout(item.timeoutId);
+    }
+    return item ?? null;
+  }
+  /**
+   * 從隊列中移除指定任務
+   */
+  removeFromQueue(missionId) {
+    const index = this.queue.findIndex((q) => q.mission.id === missionId);
+    if (index >= 0) {
+      const [removed] = this.queue.splice(index, 1);
+      clearTimeout(removed.timeoutId);
+    }
+  }
+  /**
+   * 清空隊列並拒絕所有等待的任務
+   */
+  clear(reason) {
+    for (const item of this.queue) {
+      clearTimeout(item.timeoutId);
+      item.reject(new Error(reason));
+    }
+    this.queue = [];
+  }
+  /**
+   * 取得隊列統計資訊
+   */
+  getStats() {
+    const oldestWaitMs = this.queue.length > 0 ? Date.now() - this.queue[0].enqueuedAt : null;
+    return {
+      length: this.queue.length,
+      maxSize: this.maxSize,
+      oldestWaitMs
+    };
+  }
+};
+var PoolExhaustedException = class extends Error {
+  constructor(message) {
+    super(message);
+    this.name = "PoolExhaustedException";
+  }
+};
+var QueueTimeoutException = class extends Error {
+  constructor(message) {
+    super(message);
+    this.name = "QueueTimeoutException";
+  }
+};
+
 // src/Application/PoolManager.ts
 var PoolManager = class {
-  constructor(dockerAdapter, rocketRepository, refurbishUnit, router) {
+  constructor(dockerAdapter, rocketRepository, refurbishUnit, router, config) {
     this.dockerAdapter = dockerAdapter;
     this.rocketRepository = rocketRepository;
     this.refurbishUnit = refurbishUnit;
     this.router = router;
+    this.config = { ...DEFAULT_POOL_CONFIG, ...config };
+    this.missionQueue = new MissionQueue(this.config.maxQueueSize, this.config.queueTimeoutMs);
+  }
+  config;
+  missionQueue;
+  dynamicRocketCount = 0;
+  /**
+   * 取得當前 Pool 狀態
+   */
+  async getPoolStatus() {
+    const rockets = await this.rocketRepository.findAll();
+    const statusCounts = {
+      total: rockets.length,
+      idle: 0,
+      orbiting: 0,
+      refurbishing: 0,
+      decommissioned: 0
+    };
+    for (const rocket of rockets) {
+      switch (rocket.status) {
+        case "IDLE" /* IDLE */:
+          statusCounts.idle++;
+          break;
+        case "PREPARING" /* PREPARING */:
+        case "ORBITING" /* ORBITING */:
+          statusCounts.orbiting++;
+          break;
+        case "REFURBISHING" /* REFURBISHING */:
+          statusCounts.refurbishing++;
+          break;
+        case "DECOMMISSIONED" /* DECOMMISSIONED */:
+          statusCounts.decommissioned++;
+          break;
+      }
+    }
+    return {
+      ...statusCounts,
+      dynamicCount: this.dynamicRocketCount,
+      maxRockets: this.config.maxRockets,
+      queueLength: this.missionQueue.length
+    };
   }
   /**
    * 初始化發射場：預先準備指定數量的火箭
    */
   async warmup(count) {
+    const targetCount = count ?? this.config.warmupCount;
     const currentRockets = await this.rocketRepository.findAll();
-    const needed = count - currentRockets.length;
+    const activeRockets = currentRockets.filter((r) => r.status !== "DECOMMISSIONED" /* DECOMMISSIONED */);
+    const needed = Math.min(
+      targetCount - activeRockets.length,
+      this.config.maxRockets - activeRockets.length
+    );
     if (needed <= 0) {
+      console.log(`[PoolManager] \u7121\u9700\u71B1\u6A5F\uFF0C\u7576\u524D\u5DF2\u6709 ${activeRockets.length} \u67B6\u706B\u7BAD`);
       return;
     }
-    console.log(`[LaunchPad] \u6B63\u5728\u71B1\u6A5F\uFF0C\u6E96\u5099\u767C\u5C04 ${needed} \u67B6\u65B0\u706B\u7BAD...`);
+    console.log(`[PoolManager] \u6B63\u5728\u71B1\u6A5F\uFF0C\u6E96\u5099\u767C\u5C04 ${needed} \u67B6\u65B0\u706B\u7BAD...`);
     for (let i = 0; i < needed; i++) {
       const containerId = await this.dockerAdapter.createBaseContainer();
       const rocketId = `rocket-${crypto.randomUUID()}`;
       const rocket = new Rocket(rocketId, containerId);
       await this.rocketRepository.save(rocket);
     }
+    console.log(`[PoolManager] \u71B1\u6A5F\u5B8C\u6210\uFF0CPool \u7576\u524D\u5171 ${activeRockets.length + needed} \u67B6\u706B\u7BAD`);
   }
   /**
    * 獲取一架可用的火箭並分配任務
+   *
+   * @throws {PoolExhaustedException} 當 Pool 耗盡且無法處理請求時
    */
   async assignMission(mission) {
     let rocket = await this.rocketRepository.findIdle();
-    if (!rocket) {
-      console.log(`[LaunchPad] \u8CC7\u6E90\u5403\u7DCA\uFF0C\u6B63\u5728\u7DCA\u6025\u547C\u53EB\u5F8C\u63F4\u706B\u7BAD...`);
-      const containerId = await this.dockerAdapter.createBaseContainer();
-      rocket = new Rocket(`rocket-dynamic-${Date.now()}`, containerId);
+    if (rocket) {
+      rocket.assignMission(mission);
+      await this.rocketRepository.save(rocket);
+      return rocket;
+    }
+    const poolStatus = await this.getPoolStatus();
+    const activeCount = poolStatus.total - poolStatus.decommissioned;
+    console.log(
+      `[PoolManager] \u7121\u53EF\u7528\u706B\u7BAD\uFF0C\u7576\u524D\u72C0\u614B: \u7E3D\u6578=${activeCount}/${this.config.maxRockets}, \u968A\u5217=${this.missionQueue.length}/${this.config.maxQueueSize}`
+    );
+    switch (this.config.exhaustionStrategy) {
+      case "reject":
+        throw new PoolExhaustedException(`Rocket Pool \u5DF2\u8017\u76E1\uFF0C\u7121\u6CD5\u8655\u7406\u4EFB\u52D9 ${mission.id}`);
+      case "dynamic": {
+        if (activeCount < this.config.maxRockets && this.dynamicRocketCount < (this.config.dynamicLimit ?? 5)) {
+          console.log(`[PoolManager] \u52D5\u614B\u5EFA\u7ACB\u5F8C\u63F4\u706B\u7BAD...`);
+          const containerId = await this.dockerAdapter.createBaseContainer();
+          rocket = new Rocket(`rocket-dynamic-${Date.now()}`, containerId);
+          this.dynamicRocketCount++;
+          rocket.assignMission(mission);
+          await this.rocketRepository.save(rocket);
+          return rocket;
+        }
+        if (this.missionQueue.isFull) {
+          throw new PoolExhaustedException(`Rocket Pool \u8207\u968A\u5217\u5747\u5DF2\u6EFF\uFF0C\u7121\u6CD5\u8655\u7406\u4EFB\u52D9 ${mission.id}`);
+        }
+        console.log(`[PoolManager] \u4EFB\u52D9 ${mission.id} \u9032\u5165\u7B49\u5F85\u968A\u5217`);
+        return this.missionQueue.enqueue(mission);
+      }
+      default:
+        if (this.missionQueue.isFull) {
+          throw new PoolExhaustedException(`Rocket Pool \u8207\u968A\u5217\u5747\u5DF2\u6EFF\uFF0C\u7121\u6CD5\u8655\u7406\u4EFB\u52D9 ${mission.id}`);
+        }
+        console.log(`[PoolManager] \u4EFB\u52D9 ${mission.id} \u9032\u5165\u7B49\u5F85\u968A\u5217`);
+        return this.missionQueue.enqueue(mission);
     }
-    rocket.assignMission(mission);
-    await this.rocketRepository.save(rocket);
-    return rocket;
   }
   /**
    * 回收指定任務的火箭
@@ -324,7 +579,7 @@ var PoolManager = class {
     const allRockets = await this.rocketRepository.findAll();
     const rocket = allRockets.find((r) => r.currentMission?.id === missionId);
     if (!rocket) {
-      console.warn(`[LaunchPad] \u627E\u4E0D\u5230\u5C6C\u65BC\u4EFB\u52D9 ${missionId} \u7684\u706B\u7BAD`);
+      console.warn(`[PoolManager] \u627E\u4E0D\u5230\u5C6C\u65BC\u4EFB\u52D9 ${missionId} \u7684\u706B\u7BAD`);
       return;
     }
     if (this.router && rocket.assignedDomain) {
@@ -337,30 +592,67 @@ var PoolManager = class {
       rocket.finishRefurbishment();
     }
     await this.rocketRepository.save(rocket);
+    await this.processQueue();
+  }
+  /**
+   * 處理等待隊列中的任務
+   */
+  async processQueue() {
+    const queuedItem = this.missionQueue.dequeue();
+    if (!queuedItem) {
+      return;
+    }
+    try {
+      const rocket = await this.rocketRepository.findIdle();
+      if (rocket) {
+        rocket.assignMission(queuedItem.mission);
+        await this.rocketRepository.save(rocket);
+        queuedItem.resolve(rocket);
+        console.log(`[PoolManager] \u968A\u5217\u4EFB\u52D9 ${queuedItem.mission.id} \u5DF2\u5206\u914D\u706B\u7BAD ${rocket.id}`);
+      } else {
+        console.warn(`[PoolManager] \u8655\u7406\u968A\u5217\u6642\u4ECD\u7121\u53EF\u7528\u706B\u7BAD\uFF0C\u4EFB\u52D9\u91CD\u65B0\u5165\u968A`);
+        this.missionQueue.enqueue(queuedItem.mission).then(queuedItem.resolve).catch(queuedItem.reject);
+      }
+    } catch (error) {
+      queuedItem.reject(error);
+    }
+  }
+  /**
+   * 取得隊列統計資訊
+   */
+  getQueueStats() {
+    return this.missionQueue.getStats();
   }
 };
 
 // src/Application/RefurbishUnit.ts
 var RefurbishUnit = class {
-  constructor(docker) {
+  constructor(docker, config) {
     this.docker = docker;
+    this.config = { ...DEFAULT_REFURBISH_CONFIG, ...config };
   }
+  config;
   /**
    * 執行火箭翻新邏輯
    */
   async refurbish(rocket) {
-    console.log(`[RefurbishUnit] \u6B63\u5728\u7FFB\u65B0\u706B\u7BAD: ${rocket.id} (\u5BB9\u5668: ${rocket.containerId})`);
+    console.log(
+      `[RefurbishUnit] \u6B63\u5728\u7FFB\u65B0\u706B\u7BAD: ${rocket.id} (\u7B56\u7565: ${this.config.strategy}, \u5BB9\u5668: ${rocket.containerId})`
+    );
     rocket.splashDown();
     try {
-      const cleanupCommands = ["sh", "-c", "rm -rf /app/* && pkill -f bun || true && rm -rf /tmp/*"];
-      const result = await this.docker.executeCommand(rocket.containerId, cleanupCommands);
+      const commands = this.config.cleanupCommands ?? [];
+      const fullCommand = commands.join(" && ");
+      const result = await this.docker.executeCommand(rocket.containerId, ["sh", "-c", fullCommand]);
       if (result.exitCode !== 0) {
         console.error(`[RefurbishUnit] \u6E05\u7406\u5931\u6557: ${result.stderr}`);
-        rocket.decommission();
-        return;
+        if (this.config.failureAction === "decommission") {
+          rocket.decommission();
+          return;
+        }
       }
       rocket.finishRefurbishment();
-      console.log(`[RefurbishUnit] \u706B\u7BAD ${rocket.id} \u7FFB\u65B0\u5B8C\u6210\uFF0C\u5DF2\u9032\u5165 IDLE \u72C0\u614B\u3002`);
+      console.log(`[RefurbishUnit] \u706B\u7BAD ${rocket.id} \u7FFB\u65B0\u5B8C\u6210\uFF0C\u5DF2\u9032\u5165 IDLE \u72C0\u614B`);
     } catch (error) {
       console.error(`[RefurbishUnit] \u56DE\u6536\u904E\u7A0B\u767C\u751F\u7570\u5E38:`, error);
       rocket.decommission();
@@ -393,8 +685,15 @@ var import_core = require("@gravito/core");
 var DockerAdapter = class {
   baseImage = "oven/bun:1.0-slim";
   runtime = (0, import_core.getRuntimeAdapter)();
+  // 快取目錄配置
+  cacheConfig = {
+    hostCachePath: process.env.BUN_CACHE_PATH || `${process.env.HOME}/.bun/install/cache`,
+    containerCachePathRoot: "/root/.bun/install/cache",
+    containerCachePathBun: "/home/bun/.bun/install/cache"
+  };
   async createBaseContainer() {
     const rocketId = `rocket-${crypto.randomUUID()}`;
+    await this.ensureCacheDirectory();
     const proc = this.runtime.spawn([
       "docker",
       "run",
@@ -406,10 +705,39 @@ var DockerAdapter = class {
       "-p",
       "3000",
       // 讓 Docker 分配隨機宿主機埠
+      // === 快取掛載（關鍵） ===
       "-v",
-      `${process.env.HOME}/.bun/install/cache:/root/.bun/install/cache`,
+      `${this.cacheConfig.hostCachePath}:${this.cacheConfig.containerCachePathRoot}:rw`,
       "-v",
-      `${process.env.HOME}/.bun/install/cache:/home/bun/.bun/install/cache`,
+      `${this.cacheConfig.hostCachePath}:${this.cacheConfig.containerCachePathBun}:rw`,
+      // === 環境變數配置（確保 bun 使用快取） ===
+      "-e",
+      `BUN_INSTALL_CACHE_DIR=${this.cacheConfig.containerCachePathBun}`,
+      "-e",
+      "BUN_INSTALL_CACHE=shared",
+      "-e",
+      "BUN_INSTALL_PREFER_OFFLINE=true",
+      // === 效能相關環境變數 ===
+      "-e",
+      "NODE_ENV=development",
+      // === 資源限制（防止單一容器占用過多資源） ===
+      "--memory",
+      "1g",
+      "--memory-swap",
+      "1g",
+      "--cpus",
+      "1.0",
+      // === 安全設定 ===
+      "--security-opt",
+      "no-new-privileges:true",
+      "--cap-drop",
+      "ALL",
+      "--cap-add",
+      "CHOWN",
+      "--cap-add",
+      "SETUID",
+      "--cap-add",
+      "SETGID",
       this.baseImage,
       "tail",
       "-f",
@@ -419,6 +747,7 @@ var DockerAdapter = class {
     const containerId = stdout.trim();
     const exitCode = await proc.exited;
     if (containerId.length === 64 && /^[0-9a-f]+$/.test(containerId)) {
+      await this.verifyCacheMount(containerId);
       return containerId;
     }
     if (exitCode !== 0) {
@@ -427,6 +756,37 @@ var DockerAdapter = class {
     }
     return containerId;
   }
+  /**
+   * 確保快取目錄存在
+   *
+   * @private
+   * @since 1.3.0
+   */
+  async ensureCacheDirectory() {
+    const cachePath = this.cacheConfig.hostCachePath;
+    const proc = this.runtime.spawn(["mkdir", "-p", cachePath]);
+    await proc.exited;
+    const chmodProc = this.runtime.spawn(["chmod", "777", cachePath]);
+    await chmodProc.exited;
+  }
+  /**
+   * 驗證快取是否正確掛載
+   *
+   * @private
+   * @since 1.3.0
+   */
+  async verifyCacheMount(containerId) {
+    const result = await this.executeCommand(containerId, [
+      "sh",
+      "-c",
+      `ls -la ${this.cacheConfig.containerCachePathBun} 2>/dev/null || echo 'NOT_MOUNTED'`
+    ]);
+    if (result.stdout.includes("NOT_MOUNTED")) {
+      console.warn(`[DockerAdapter] \u8B66\u544A: \u5BB9\u5668 ${containerId} \u7684\u5FEB\u53D6\u76EE\u9304\u53EF\u80FD\u672A\u6B63\u78BA\u639B\u8F09`);
+    } else {
+      console.log(`[DockerAdapter] \u5FEB\u53D6\u76EE\u9304\u5DF2\u78BA\u8A8D\u639B\u8F09: ${containerId}`);
+    }
+  }
   async getExposedPort(containerId, containerPort = 3e3) {
     const proc = this.runtime.spawn(["docker", "port", containerId, containerPort.toString()]);
     const stdout = await new Response(proc.stdout ?? null).text();
@@ -807,13 +1167,25 @@ async function bootstrapLaunchpad() {
 }
 // Annotate the CommonJS export names for ESM import in node:
 0 && (module.exports = {
+  DEFAULT_POOL_CONFIG,
+  DEFAULT_REFURBISH_CONFIG,
   LaunchpadOrbit,
   Mission,
+  MissionAssigned,
   MissionControl,
+  MissionQueue,
+  MissionQueueTimeout,
+  MissionQueued,
   PayloadInjector,
+  PoolExhausted,
+  PoolExhaustedException,
   PoolManager,
+  QueueTimeoutException,
   RefurbishUnit,
+  RefurbishmentCompleted,
   Rocket,
+  RocketIgnited,
+  RocketSplashedDown,
   RocketStatus,
   bootstrapLaunchpad
 });