docker-compose.yml

name: aimeter

# AI-Meter — Option C (event-driven) stack. Everything runs in containers.
#   db     : TimescaleDB (Postgres + time-series) — the columnar/time-series store + rollups
#   redis  : Redis Streams — the event queue between ingest and enrichment
#   api    : FastAPI — ingest endpoint (-> stream) + read endpoints (<- db)
#   worker : async consumer — reads the stream, computes impact, writes the db
#
# The stream is the ONLY path into the database: api publishes, worker persists.

services:
  db:
    build: ./db
    command: ["postgres", "-c", "shared_preload_libraries=timescaledb"]
    restart: unless-stopped
    environment:
      POSTGRES_USER: ${AIMETER_DB_USER:-aimeter}
      POSTGRES_PASSWORD: ${AIMETER_DB_PASSWORD:-aimeter}
      POSTGRES_DB: ${AIMETER_DB_NAME:-aimeter}
    volumes:
      - dbdata:/var/lib/postgresql/data
      - ./db/init:/docker-entrypoint-initdb.d:ro
    healthcheck:
      test: ["CMD-SHELL", "pg_isready -U $${POSTGRES_USER} -d $${POSTGRES_DB}"]
      interval: 5s
      timeout: 3s
      retries: 20

  redis:
    image: public.ecr.aws/docker/library/redis:7.4.9-alpine
    command: ["redis-server", "--appendonly", "yes", "--appendfsync", "everysec", "--save", "60", "1000"]
    restart: unless-stopped
    volumes:
      - redisdata:/data
    healthcheck:
      test: ["CMD", "redis-cli", "ping"]
      interval: 5s
      timeout: 3s
      retries: 20

  api:
    build: .
    image: aimeter-app:local
    command: uvicorn aimeter.api.main:app --host 0.0.0.0 --port 8000
    restart: unless-stopped
    volumes:
      # live-mount the static dir so monitor.html edits show on refresh (no rebuild)
      - ./aimeter/api/static:/app/aimeter/api/static:ro
    environment:
      AIMETER_DATABASE_URL: postgresql://${AIMETER_DB_USER:-aimeter}:${AIMETER_DB_PASSWORD:-aimeter}@db:5432/${AIMETER_DB_NAME:-aimeter}
      AIMETER_REDIS_URL: redis://redis:6379/0
      AIMETER_GRID_PROVIDER: ${AIMETER_GRID_PROVIDER:-bundled}
      AIMETER_DEFAULT_REGION: ${AIMETER_DEFAULT_REGION:-WORLD}
      AIMETER_API_TOKEN: ${AIMETER_API_TOKEN:-}
      AIMETER_CLAIM_IDLE_MS: ${AIMETER_CLAIM_IDLE_MS:-30000}
      AIMETER_MAX_DELIVERY_ATTEMPTS: ${AIMETER_MAX_DELIVERY_ATTEMPTS:-5}
    ports:
      - "127.0.0.1:${AIMETER_API_PORT:-8000}:8000"
    depends_on:
      db:
        condition: service_healthy
      redis:
        condition: service_healthy
    healthcheck:
      test: ["CMD", "python", "-c", "import urllib.request; urllib.request.urlopen('http://localhost:8000/healthz', timeout=2)"]
      interval: 10s
      timeout: 3s
      retries: 10

  worker:
    image: aimeter-app:local
    command: python -m aimeter.worker
    restart: unless-stopped
    environment:
      AIMETER_DATABASE_URL: postgresql://${AIMETER_DB_USER:-aimeter}:${AIMETER_DB_PASSWORD:-aimeter}@db:5432/${AIMETER_DB_NAME:-aimeter}
      AIMETER_REDIS_URL: redis://redis:6379/0
      AIMETER_GRID_PROVIDER: ${AIMETER_GRID_PROVIDER:-bundled}
      AIMETER_DEFAULT_REGION: ${AIMETER_DEFAULT_REGION:-WORLD}
      AIMETER_API_TOKEN: ${AIMETER_API_TOKEN:-}
      AIMETER_CLAIM_IDLE_MS: ${AIMETER_CLAIM_IDLE_MS:-30000}
      AIMETER_MAX_DELIVERY_ATTEMPTS: ${AIMETER_MAX_DELIVERY_ATTEMPTS:-5}
    depends_on:
      db:
        condition: service_healthy
      redis:
        condition: service_healthy
    healthcheck:
      test: ["CMD", "python", "-c", "import redis,os; r=redis.Redis.from_url(os.environ['AIMETER_REDIS_URL']); assert r.get('aimeter:worker:heartbeat')"]
      interval: 10s
      timeout: 3s
      retries: 10

volumes:
  dbdata:
  redisdata: