@gravito/launchpad 1.2.2 → 1.3.2

package/src/Infrastructure/Docker/DockerAdapter.ts

@@ -14,9 +14,19 @@ export class DockerAdapter implements IDockerAdapter {
   private baseImage = 'oven/bun:1.0-slim'
   private runtime = getRuntimeAdapter()
 
+  // 快取目錄配置
+  private readonly 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(): Promise<string> {
     const rocketId = `rocket-${crypto.randomUUID()}`
 
+    // 確保宿主機快取目錄存在
+    await this.ensureCacheDirectory()
+
     const proc = this.runtime.spawn([
       'docker',
       'run',
@@ -27,10 +37,39 @@ export class DockerAdapter implements IDockerAdapter {
       'gravito-origin=launchpad',
       '-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',
@@ -42,6 +81,8 @@ export class DockerAdapter implements IDockerAdapter {
     const exitCode = await proc.exited
 
     if (containerId.length === 64 && /^[0-9a-f]+$/.test(containerId)) {
+      // 驗證快取是否正確掛載
+      await this.verifyCacheMount(containerId)
       return containerId
     }
 
@@ -53,6 +94,42 @@ export class DockerAdapter implements IDockerAdapter {
     return containerId
   }
 
+  /**
+   * 確保快取目錄存在
+   *
+   * @private
+   * @since 1.3.0
+   */
+  private async ensureCacheDirectory(): Promise<void> {
+    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
+   */
+  private async verifyCacheMount(containerId: string): Promise<void> {
+    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] 警告: 容器 ${containerId} 的快取目錄可能未正確掛載`)
+    } else {
+      console.log(`[DockerAdapter] 快取目錄已確認掛載: ${containerId}`)
+    }
+  }
+
   async getExposedPort(containerId: string, containerPort = 3000): Promise<number> {
     const proc = this.runtime.spawn(['docker', 'port', containerId, containerPort.toString()])
     const stdout = await new Response(proc.stdout ?? null).text()

package/src/index.ts

@@ -13,9 +13,12 @@ import { CachedRocketRepository } from './Infrastructure/Persistence/CachedRocke
 import { BunProxyAdapter } from './Infrastructure/Router/BunProxyAdapter'
 
 export * from './Application/MissionControl'
+export * from './Application/MissionQueue'
 export * from './Application/PayloadInjector'
 export * from './Application/PoolManager'
 export * from './Application/RefurbishUnit'
+export * from './Domain/Events'
+export * from './Domain/Interfaces'
 export * from './Domain/Mission'
 export * from './Domain/Rocket'
 export * from './Domain/RocketStatus'

package/tests/Deployment.test.ts

@@ -23,7 +23,9 @@ describe('Payload Injection Flow', () => {
     findAll: mock(() => Promise.resolve([])),
   }
 
-  const pool = new PoolManager(mockDocker, mockRepo)
+  const pool = new PoolManager(mockDocker, mockRepo, undefined, undefined, {
+    exhaustionStrategy: 'dynamic',
+  })
   const injector = new PayloadInjector(mockDocker, mockGit)
 
   it('應該能從指派任務到成功點火', async () => {

package/tests/pool-manager.test.ts

@@ -39,7 +39,9 @@ describe('PoolManager', () => {
   test('assigns missions to idle rockets or creates new ones', async () => {
     const docker = new FakeDocker()
     const repo = new InMemoryRocketRepository()
-    const pool = new PoolManager(docker as any, repo)
+    const pool = new PoolManager(docker as any, repo, undefined, undefined, {
+      exhaustionStrategy: 'dynamic',
+    })
 
     const mission = Mission.create({
       id: 'mission-2',

package/tests/resource-limits.test.ts

@@ -0,0 +1,241 @@
+import { beforeEach, describe, expect, mock, test } from 'bun:test'
+import { MissionQueue, PoolExhaustedException } from '../src/Application/MissionQueue'
+import { PoolManager } from '../src/Application/PoolManager'
+import { Mission } from '../src/Domain/Mission'
+import type { Rocket } from '../src/Domain/Rocket'
+import { RocketStatus } from '../src/Domain/RocketStatus'
+
+describe('MissionQueue', () => {
+  let queue: MissionQueue
+
+  beforeEach(() => {
+    queue = new MissionQueue(3, 1000)
+  })
+
+  test('should enqueue missions and track length', () => {
+    const mission = Mission.create({
+      id: 'pr-1',
+      repoUrl: 'https://github.com/test/repo',
+      branch: 'main',
+      commitSha: 'abc123',
+    })
+
+    const promise = queue.enqueue(mission)
+    expect(queue.length).toBe(1)
+    expect(promise).toBeInstanceOf(Promise)
+
+    // 清理：dequeue 或 catch 錯誤以避免 unhandled rejection
+    promise.catch(() => {})
+  })
+
+  test('should reject when queue is full', async () => {
+    const promises = []
+    for (let i = 0; i < 3; i++) {
+      const promise = queue.enqueue(
+        Mission.create({
+          id: `pr-${i}`,
+          repoUrl: 'https://github.com/test/repo',
+          branch: 'main',
+          commitSha: 'abc123',
+        })
+      )
+      // 捕獲超時錯誤以避免 unhandled rejection
+      promise.catch(() => {})
+      promises.push(promise)
+    }
+
+    const mission = Mission.create({
+      id: 'pr-4',
+      repoUrl: 'https://github.com/test/repo',
+      branch: 'main',
+      commitSha: 'abc123',
+    })
+
+    await expect(queue.enqueue(mission)).rejects.toThrow(PoolExhaustedException)
+  })
+
+  test('should dequeue in FIFO order', () => {
+    const m1 = Mission.create({
+      id: 'pr-1',
+      repoUrl: 'https://github.com/test/repo',
+      branch: 'main',
+      commitSha: 'abc123',
+    })
+    const m2 = Mission.create({
+      id: 'pr-2',
+      repoUrl: 'https://github.com/test/repo',
+      branch: 'main',
+      commitSha: 'abc123',
+    })
+
+    const p1 = queue.enqueue(m1)
+    const p2 = queue.enqueue(m2)
+
+    const first = queue.dequeue()
+    expect(first?.mission.id).toBe('pr-1')
+
+    const second = queue.dequeue()
+    expect(second?.mission.id).toBe('pr-2')
+
+    const empty = queue.dequeue()
+    expect(empty).toBeNull()
+
+    // 清理未使用的 promises
+    p1.catch(() => {})
+    p2.catch(() => {})
+  })
+
+  test('should provide stats', async () => {
+    const promise = queue.enqueue(
+      Mission.create({
+        id: 'pr-1',
+        repoUrl: 'https://github.com/test/repo',
+        branch: 'main',
+        commitSha: 'abc123',
+      })
+    )
+
+    // 等待一點時間讓時間戳有差異
+    await new Promise((resolve) => setTimeout(resolve, 10))
+
+    const stats = queue.getStats()
+    expect(stats.length).toBe(1)
+    expect(stats.maxSize).toBe(3)
+    expect(stats.oldestWaitMs).toBeGreaterThanOrEqual(0)
+
+    // 清理
+    promise.catch(() => {})
+  })
+})
+
+describe('PoolManager Resource Limits', () => {
+  let mockDocker: any
+  let mockRepo: any
+  let rockets: Rocket[]
+
+  beforeEach(() => {
+    rockets = []
+    mockDocker = {
+      createBaseContainer: mock(() => Promise.resolve(`container-${Date.now()}`)),
+      removeContainer: mock(() => Promise.resolve()),
+      executeCommand: mock(() => Promise.resolve({ exitCode: 0, stdout: '', stderr: '' })),
+      getExposedPort: mock(() => Promise.resolve(3000)),
+      removeContainerByLabel: mock(() => Promise.resolve()),
+      streamLogs: mock(() => {}),
+      getStats: mock(() => Promise.resolve({ cpu: '0%', memory: '0MB' })),
+    }
+    mockRepo = {
+      save: mock((r: Rocket) => {
+        const index = rockets.findIndex((rocket) => rocket.id === r.id)
+        if (index >= 0) {
+          rockets[index] = r
+        } else {
+          rockets.push(r)
+        }
+        return Promise.resolve()
+      }),
+      findAll: mock(() => Promise.resolve(rockets)),
+      findIdle: mock(() =>
+        Promise.resolve(rockets.find((r) => r.status === RocketStatus.IDLE) || null)
+      ),
+      findById: mock((id: string) => Promise.resolve(rockets.find((r) => r.id === id) || null)),
+      delete: mock(() => Promise.resolve()),
+    }
+  })
+
+  test('should reject when maxRockets reached with reject strategy', async () => {
+    const manager = new PoolManager(mockDocker, mockRepo, undefined, undefined, {
+      maxRockets: 2,
+      exhaustionStrategy: 'reject',
+      warmupCount: 2,
+    })
+
+    await manager.warmup()
+    expect(rockets.length).toBe(2)
+
+    // 使用全部火箭
+    const m1 = Mission.create({
+      id: 'pr-1',
+      repoUrl: 'https://github.com/test/repo',
+      branch: 'main',
+      commitSha: 'abc1',
+    })
+    const m2 = Mission.create({
+      id: 'pr-2',
+      repoUrl: 'https://github.com/test/repo',
+      branch: 'main',
+      commitSha: 'abc2',
+    })
+
+    await manager.assignMission(m1)
+    await manager.assignMission(m2)
+
+    // 第三個請求應該被拒絕
+    const m3 = Mission.create({
+      id: 'pr-3',
+      repoUrl: 'https://github.com/test/repo',
+      branch: 'main',
+      commitSha: 'abc3',
+    })
+
+    await expect(manager.assignMission(m3)).rejects.toThrow(PoolExhaustedException)
+  })
+
+  test('should respect maxRockets limit', async () => {
+    const manager = new PoolManager(mockDocker, mockRepo, undefined, undefined, {
+      maxRockets: 5,
+      warmupCount: 3,
+    })
+
+    await manager.warmup()
+
+    const status = await manager.getPoolStatus()
+    expect(status.total).toBe(3)
+    expect(status.idle).toBe(3)
+    expect(status.maxRockets).toBe(5)
+  })
+
+  test('should queue missions when pool is exhausted', async () => {
+    const manager = new PoolManager(mockDocker, mockRepo, undefined, undefined, {
+      maxRockets: 1,
+      exhaustionStrategy: 'queue',
+      queueTimeoutMs: 5000,
+      warmupCount: 1,
+    })
+
+    await manager.warmup()
+
+    const m1 = Mission.create({
+      id: 'pr-1',
+      repoUrl: 'https://github.com/test/repo',
+      branch: 'main',
+      commitSha: 'abc1',
+    })
+
+    const rocket1 = await manager.assignMission(m1)
+    expect(rocket1.status).toBe(RocketStatus.PREPARING)
+
+    // 第二個任務應該進入隊列
+    const m2 = Mission.create({
+      id: 'pr-2',
+      repoUrl: 'https://github.com/test/repo',
+      branch: 'main',
+      commitSha: 'abc2',
+    })
+
+    const promise = manager.assignMission(m2)
+
+    // 等待 promise 開始處理
+    await new Promise((resolve) => setTimeout(resolve, 10))
+
+    const stats = manager.getQueueStats()
+    expect(stats.length).toBe(1)
+
+    // 模擬回收
+    rocket1.ignite()
+    await manager.recycle('pr-1')
+
+    const rocket2 = await promise
+    expect(rocket2.currentMission?.id).toBe('pr-2')
+  })
+})

package/tsconfig.json

@@ -1,27 +1,15 @@
 {
-    "extends": "../../tsconfig.json",
-    "compilerOptions": {
-        "outDir": "./dist",
-        "skipLibCheck": true,
-        "types": [
-            "bun-types"
-        ],
-        "baseUrl": ".",
-        "paths": {
-            "@gravito/core": [
-                "../../packages/core/src/index.ts"
-            ],
-            "@gravito/*": [
-                "../../packages/*/src/index.ts"
-            ]
-        }
-    },
-    "include": [
-        "src/**/*"
-    ],
-    "exclude": [
-        "node_modules",
-        "dist",
-        "**/*.test.ts"
-    ]
-}
+  "extends": "../../tsconfig.json",
+  "compilerOptions": {
+    "outDir": "./dist",
+    "skipLibCheck": true,
+    "types": ["bun-types"],
+    "baseUrl": ".",
+    "paths": {
+      "@gravito/core": ["../../packages/core/src/index.ts"],
+      "@gravito/*": ["../../packages/*/src/index.ts"]
+    }
+  },
+  "include": ["src/**/*"],
+  "exclude": ["node_modules", "dist", "**/*.test.ts"]
+}