96 lines
3.5 KiB
YAML
96 lines
3.5 KiB
YAML
# Wild Dragon MAM — Worker Node Stack
|
|
# ─────────────────────────────────────
|
|
# Deploy on any machine you want to join the cluster as a worker.
|
|
# The primary stack (mam-api, db, redis) continues running on TrueNAS.
|
|
#
|
|
# Required env vars (set in .env.worker or export before running):
|
|
# MAM_API_URL URL of the primary MAM API e.g. http://10.0.0.25:47432
|
|
# NODE_TOKEN Bearer token from the primary's Tokens page
|
|
#
|
|
# Optional hardware overrides (if Docker can't see /dev directly):
|
|
# GPU_COUNT Number of NVIDIA GPUs on this node (default: auto-detect from /dev/nvidia*)
|
|
# BMD_COUNT Number of Blackmagic DeckLink cards (default: auto-detect from /dev/blackmagic/)
|
|
#
|
|
# Optional env vars (needed only if starting the worker or capture profiles):
|
|
# REDIS_URL, DATABASE_URL, S3_ENDPOINT, S3_BUCKET, S3_ACCESS_KEY, S3_SECRET_KEY
|
|
# BMD_DEVICE_0 DeckLink device path (default: /dev/blackmagic/dv0)
|
|
# BMD_DEVICE_1 DeckLink device path (default: /dev/blackmagic/dv1)
|
|
#
|
|
# Profiles:
|
|
# (default) node-agent only — cluster visibility + hardware heartbeat
|
|
# --profile worker + CPU job worker (proxy generation, transcoding)
|
|
# --profile capture + SDI capture service (requires Blackmagic DeckLink card)
|
|
#
|
|
# To enable GPU transcoding, also apply docker-compose.gpu.yml:
|
|
# docker compose -f docker-compose.worker.yml -f docker-compose.gpu.yml --profile worker up -d
|
|
#
|
|
# Examples:
|
|
# # Agent only (register node in cluster, report hardware):
|
|
# docker compose -f docker-compose.worker.yml up -d
|
|
#
|
|
# # SDI capture node (1 DeckLink, no GPU worker):
|
|
# docker compose -f docker-compose.worker.yml --profile capture up -d
|
|
#
|
|
# # Full node (GPU worker + SDI capture):
|
|
# docker compose -f docker-compose.worker.yml -f docker-compose.gpu.yml \
|
|
# --profile worker --profile capture up -d
|
|
|
|
services:
|
|
|
|
node-agent:
|
|
build: ./services/node-agent
|
|
restart: unless-stopped
|
|
environment:
|
|
MAM_API_URL: ${MAM_API_URL}
|
|
NODE_TOKEN: ${NODE_TOKEN:-}
|
|
NODE_ROLE: ${NODE_ROLE:-worker}
|
|
AGENT_PORT: 3002
|
|
HEARTBEAT_MS: ${HEARTBEAT_MS:-30000}
|
|
GPU_COUNT: ${GPU_COUNT:--1}
|
|
BMD_COUNT: ${BMD_COUNT:--1}
|
|
ports:
|
|
- "${AGENT_PORT:-7436}:3002"
|
|
networks:
|
|
- wild-dragon-worker
|
|
|
|
worker:
|
|
build: ./services/worker
|
|
profiles: [worker]
|
|
restart: unless-stopped
|
|
environment:
|
|
REDIS_URL: ${REDIS_URL}
|
|
DATABASE_URL: ${DATABASE_URL}
|
|
S3_ENDPOINT: ${S3_ENDPOINT}
|
|
S3_BUCKET: ${S3_BUCKET}
|
|
S3_ACCESS_KEY: ${S3_ACCESS_KEY}
|
|
S3_SECRET_KEY: ${S3_SECRET_KEY}
|
|
S3_REGION: ${S3_REGION:-us-east-1}
|
|
NVENC_ENABLED: ${NVENC_ENABLED:-false}
|
|
networks:
|
|
- wild-dragon-worker
|
|
|
|
# SDI capture service — only start on nodes with Blackmagic DeckLink cards
|
|
# Set BMD_DEVICE_0 in .env.worker to the actual device path, e.g. /dev/blackmagic/dv0
|
|
capture:
|
|
build: ./services/capture
|
|
profiles: [capture]
|
|
restart: unless-stopped
|
|
environment:
|
|
REDIS_URL: ${REDIS_URL}
|
|
DATABASE_URL: ${DATABASE_URL}
|
|
S3_ENDPOINT: ${S3_ENDPOINT}
|
|
S3_BUCKET: ${S3_BUCKET}
|
|
S3_ACCESS_KEY: ${S3_ACCESS_KEY}
|
|
S3_SECRET_KEY: ${S3_SECRET_KEY}
|
|
CAPTURE_PORT: 3001
|
|
devices:
|
|
- ${BMD_DEVICE_0:-/dev/blackmagic/dv0}:/dev/blackmagic/dv0
|
|
- ${BMD_DEVICE_1:-/dev/blackmagic/dv1}:/dev/blackmagic/dv1
|
|
ports:
|
|
- "${CAPTURE_PORT:-7437}:3001"
|
|
networks:
|
|
- wild-dragon-worker
|
|
|
|
networks:
|
|
wild-dragon-worker:
|
|
driver: bridge
|