跳至主要內容

部署 Mastra worker

Mastra worker 作為獨立程序執行,讓你可以分開擴展協調、排程及背景工作,而不受 API 影響。本指南會逐步介紹使用 Docker Compose 或 Kubernetes 進行完全分拆部署的方法。

資訊

本指南涵蓋如何將 worker 分拆至各自的容器。如只需讓 worker 與 API 在同一程序內執行,請參閱 Worker,無需額外設定。

開始之前
開始之前 的直接連結

你需要:

注意

預設的記憶體內 PubSub 無法跨程序傳送事件。將 worker 分拆至不同容器前,必須先設定分散式 PubSub 後端。

設定共用基礎設施
設定共用基礎設施 的直接連結

Mastra 實例連接至分散式 PubSub 後端及共用資料庫。使用環境變數,讓每個容器都可執行相同映像檔。

src/mastra/index.ts
import { Mastra } from '@mastra/core/mastra'
import { RedisStreamsPubSub } from '@mastra/redis-streams'
import { PostgresStore } from '@mastra/pg'

export const mastra = new Mastra({
storage: new PostgresStore({
connectionString: process.env.DATABASE_URL!,
}),
pubsub: new RedisStreamsPubSub({
url: process.env.REDIS_URL!,
}),
})

任何支援的儲存後端都適用。請將儲存配接器換成你偏好的資料庫。

部署
部署 的直接連結

  1. 建置 Mastra 應用程式。輸出會在每個容器中執行。

    mastra build

    這會產生自包含的 .mastra/output/ 目錄。有關建置輸出的詳情,請參閱部署 Mastra 伺服器

  2. 建立 Dockerfile,以複製預先建置的輸出並安裝生產環境依賴套件:

    app/Dockerfile
    FROM node:22-alpine

    WORKDIR /app

    COPY .mastra/output/package.json .mastra/output/.npmrc* ./
    RUN npm install --omit=dev

    COPY .mastra/output/ .

    EXPOSE 4111
    CMD ["node", "index.mjs"]
  3. 定義完全分拆的拓撲。此設定會執行六項服務:資料庫、PubSub 後端、API 伺服器,以及三個 worker。每個 worker 都執行相同映像檔,並透過不同的 MASTRA_WORKERS 值控制啟動哪個 worker。

    API 將 MASTRA_WORKERS: "false" 設定為停用所有事件處理。協調 worker 會設定 MASTRA_STEP_EXECUTION_URL,將步驟執行請求指向 API 的內部 URL。詳情請參閱步驟執行 URL

    所有服務共用一個 MASTRA_WORKER_AUTH_TOKEN。worker 向 API 發出請求時會附上此 token,讓 API 可驗證呼叫者是受信任的內部服務。詳情請參閱 worker 驗證

    docker-compose.yml
    services:
    postgres:
    image: postgres:16-alpine
    environment:
    POSTGRES_USER: mastra
    POSTGRES_PASSWORD: ${POSTGRES_PASSWORD}
    POSTGRES_DB: mastra
    ports:
    - '5432:5432'
    volumes:
    - pgdata:/var/lib/postgresql/data
    healthcheck:
    test: ['CMD-SHELL', 'pg_isready -U mastra']
    interval: 5s
    timeout: 3s
    retries: 5

    redis:
    image: redis:7-alpine
    ports:
    - '6379:6379'
    healthcheck:
    test: ['CMD', 'redis-cli', 'ping']
    interval: 5s
    timeout: 3s
    retries: 5

    api:
    build: ./app
    ports:
    - '4111:4111'
    environment:
    DATABASE_URL: postgres://mastra:${POSTGRES_PASSWORD}@postgres:5432/mastra
    REDIS_URL: redis://redis:6379
    MASTRA_WORKERS: 'false'
    MASTRA_WORKER_AUTH_TOKEN: ${MASTRA_WORKER_AUTH_TOKEN}
    depends_on:
    postgres:
    condition: service_healthy
    redis:
    condition: service_healthy
    healthcheck:
    test: ['CMD', 'wget', '-qO-', 'http://localhost:4111/api/agents']
    interval: 5s
    timeout: 3s
    retries: 5

    orchestration-worker:
    build: ./app
    environment:
    DATABASE_URL: postgres://mastra:${POSTGRES_PASSWORD}@postgres:5432/mastra
    REDIS_URL: redis://redis:6379
    MASTRA_WORKERS: orchestration
    MASTRA_STEP_EXECUTION_URL: http://api:4111/api
    MASTRA_WORKER_AUTH_TOKEN: ${MASTRA_WORKER_AUTH_TOKEN}
    depends_on:
    api:
    condition: service_healthy

    scheduler-worker:
    build: ./app
    environment:
    DATABASE_URL: postgres://mastra:${POSTGRES_PASSWORD}@postgres:5432/mastra
    REDIS_URL: redis://redis:6379
    MASTRA_WORKERS: scheduler
    MASTRA_WORKER_AUTH_TOKEN: ${MASTRA_WORKER_AUTH_TOKEN}
    depends_on:
    api:
    condition: service_healthy

    background-task-worker:
    build: ./app
    environment:
    DATABASE_URL: postgres://mastra:${POSTGRES_PASSWORD}@postgres:5432/mastra
    REDIS_URL: redis://redis:6379
    MASTRA_WORKERS: backgroundTasks
    MASTRA_WORKER_AUTH_TOKEN: ${MASTRA_WORKER_AUTH_TOKEN}
    depends_on:
    api:
    condition: service_healthy

    volumes:
    pgdata:

    docker-compose.yml 旁建立 .env 檔案:

    .env
    POSTGRES_PASSWORD=your-secure-password
    MASTRA_WORKER_AUTH_TOKEN=your-shared-secret-token
    備註

    請記得設定應用程式所需的任何其他環境變數(例如你的模型 Provider API 金鑰)。

  4. 驗證技術棧正在執行,且 API 能夠回應:

    docker compose up -d
    docker compose ps
    curl http://localhost:4111/api/agents

步驟執行 URL
步驟執行 URL 的直接連結

在完全分拆的部署中,協調 worker 會在 API 以外的另一個容器中執行。處理 Workflow 事件時,它會透過 HTTP 將步驟執行工作交由 API 處理。

MASTRA_STEP_EXECUTION_URL 設為 API 的內部 URL,並包括 /api 前綴:

MASTRA_STEP_EXECUTION_URL=http://api:4111/api

協調 worker 會就每個步驟向 ${MASTRA_STEP_EXECUTION_URL}/workflows/:workflowId/runs/:runId/steps/execute 發送 POST 請求。API 會解析 Workflow,並在本機執行步驟。

如沒有此變數,協調 worker 會嘗試在程序內執行步驟。當 worker 與 API 一同執行時,此方式有效;但在分拆部署中,worker 無法存取完整的 Mastra runtime,因此會執行失敗。

擴展
擴展 的直接連結

協調 worker 及背景工作 worker 可以安全地水平擴展。PubSub consumer group 會在各實例之間分配事件,因此每個事件只會處理一次:

docker compose up -d --scale orchestration-worker=3
docker compose up -d --scale background-task-worker=2

API 亦可在負載平衡器後方水平擴展。

**請勿擴展排程 worker。**只能執行一個實例。如有多個排程器輪詢相同儲存空間,便會為同一排程觸發重複事件。

當機復原
當機復原 的直接連結

由於分散式 PubSub 後端會保留尚未確認的事件,worker 可在當機後復原:

  • 協調 worker:待處理事件會保留在 PubSub 後端。worker 重新啟動後,會從中斷之處繼續處理。
  • 排程 worker:不會永久遺漏任何事件。重新啟動時,排程器會根據當前時間計算下一次觸發時間,而不是從中斷之處繼續。
  • 步驟執行期間的 API:協調 worker 的 HTTP 請求會失敗。事件會被 nacked,並在下一次嘗試時重新傳送。
注意

如 API 在步驟執行期間當機(例如 sleep 途中),該步驟的工作便會遺失。Workflow 執行可能一直停留在 running 狀態。Mastra 目前尚未針對此情況提供以逾時為基礎的自動復原機制。